Dr. Vivek Yedavalli Profile

Dr. Vivek Yedavalli

SPIE Involvement:

Author

Publications (1)

This will count as one of your downloads.

You will have access to both the presentation and article (if available).

DOWNLOAD NOW

This content is available for download via your institution's subscription. To access this item, please sign in to your personal account.

Email or Username Forgot your username?

Password Forgot your password?

Show

Keep me signed in

No SPIE account? Create an account

SPIE Journal Paper | 8 June 2022

Direct quantification of epistemic and aleatoric uncertainty in 3D U-net segmentation

Craig Jones, Guoqing Wang, Vivek Yedavalli, Haris Sair

JMI, Vol. 9, Issue 03, 034002, (June 2022) https://doi.org/10.1117/12.10.1117/1.JMI.9.3.034002

KEYWORDS: Image segmentation, Tissues, Neural networks, Tumors, Data modeling, 3D modeling, Brain, Magnetic resonance imaging, Brain mapping, 3D image processing

Read Abstract +

Purpose: To derive a multinomial probability function and quantitative measures of the data and epistemic uncertainty as direct output of a 3D U-Net segmentation network. Approach: A set of T1 brain MRI images were downloaded from the Connectome Project and segmented using FMRIB’s FAST algorithm to be used as ground truth. A 3D U-Net neural network was trained with sample sizes of 200, 500, and 898 T1 brain images using a loss function defined as the negative logarithm of the likelihood based on a derivation of the definition of the multinomial probability function. From this definition, the epistemic and aleatoric uncertainty equations were derived and used to quantify maps of the uncertainty along with tissue segmentations. Results: Maps of the tissue segmentation along with the epistemic and aleatoric uncertainty, per voxel, are presented. The uncertainty decreased based on the increasing number of training data used to train the neural network. The neural network trained with 898 volumes resulted in uncertainty maps that were high primarily in the tissue boundary regions. The epistemic and aleatoric uncertainty were averaged over all test data (connectome and tumor separately), and the epistemic uncertainty showed a decreasing trend, as expected, with increasing numbers of data used to train the model. The aleatoric uncertainty showed a similar trend which was also expected as the aleatoric uncertainty is not expected to be as dependent on the number of training data. Conclusion: The derived uncertainty equations from a multinomial probability distribution were able to quantify the aleatoric and epistemic uncertainty per voxel and are applicable for all two-dimensional and three-dimensional neural networks.

My Library

You currently do not have any folders to save your paper to! Create a new folder below.

Folder Name

Folder Description

View contact details

UPDATE YOUR PROFILE

Is this your profile? Update it now.

Sign into your SPIE.org account

Don’t have a profile and want one?

Create an account on SPIE.org

Access to the requested content is limited to institutions that have purchased or subscribe to SPIE eBooks. You are receiving this notice because your organization may not have SPIE eBooks access.*

*Shibboleth/Open Athens users─please sign in to access your institution's subscriptions.

To obtain this item, you may purchase the complete book in print or electronic format on SPIE.org.

ORGANIZATIONAL
Sign in with credentials provided by your organization.

Organizational Username

Organizational Password

Show/Hide Password

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE

Includes PDF, HTML & Video, when available

Members:

Non-members: ADD TO CART

Keywords/Phrases

Search In:

Publication Years