Quantifying the uncertainty of neural networks using Monte Carlo dropout for deep learning based quantitative MRI
Avci, Mehmet Yigit, Li, Ziyu, Fan, Qiuyun, Huang, Susie, Bilgic, Berkin, Tian, Qiyuan
–arXiv.org Artificial Intelligence
Dropout is conventionally used during the training phase as regularization method and for quantifying uncertainty in deep learning. We propose to use dropout during training as well as inference steps, and average multiple predictions to improve the accuracy, while reducing and quantifying the uncertainty. The results are evaluated for fractional anisotropy (FA) and mean diffusivity (MD) maps which are obtained from only 3 direction scans. With our method, accuracy can be improved significantly compared to network outputs without dropout, especially when the training dataset is small. Moreover, confidence maps are generated which may aid in diagnosis of unseen pathology or artifacts.
arXiv.org Artificial Intelligence
Dec-2-2021
- Country:
- North America > United States
- Massachusetts (0.05)
- Europe
- United Kingdom > England
- Oxfordshire > Oxford (0.14)
- Middle East > Republic of Türkiye
- Istanbul Province > Istanbul (0.05)
- United Kingdom > England
- Asia
- Middle East > Republic of Türkiye
- Istanbul Province > Istanbul (0.05)
- China > Tianjin Province
- Tianjin (0.05)
- Middle East > Republic of Türkiye
- North America > United States
- Genre:
- Research Report (0.50)
- Industry:
- Health & Medicine
- Diagnostic Medicine > Imaging (1.00)
- Health Care Technology (0.95)
- Health & Medicine
- Technology: