TY - GEN
T1 - Bayesian deep learning for accelerated mr image reconstruction
AU - Schlemper, Jo
AU - Castro, Daniel C.
AU - Bai, Wenjia
AU - Qin, Chen
AU - Oktay, Ozan
AU - Duan, Jinming
AU - Price, Anthony N.
AU - Hajnal, Jo
AU - Rueckert, Daniel
N1 - Publisher Copyright:
© 2018, Springer Nature Switzerland AG.
PY - 2018
Y1 - 2018
N2 - Recently, many deep learning (DL) based MR image reconstruction methods have been proposed with promising results. However, only a handful of work has been focussing on characterising the behaviour of deep networks, such as investigating when the networks may fail to reconstruct. In this work, we explore the applicability of Bayesian DL techniques to model the uncertainty associated with DL-based reconstructions. In particular, we apply MC-dropout and heteroscedastic loss to the reconstruction networks to model epistemic and aleatoric uncertainty. We show that the proposed Bayesian methods achieve competitive performance when the test images are relatively far from the training data distribution and outperforms when the baseline method is over-parametrised. In addition, we qualitatively show that there seems to be a correlation between the magnitude of the produced uncertainty maps and the error maps, demonstrating the potential utility of the Bayesian DL methods for assessing the reliability of the reconstructed images.
AB - Recently, many deep learning (DL) based MR image reconstruction methods have been proposed with promising results. However, only a handful of work has been focussing on characterising the behaviour of deep networks, such as investigating when the networks may fail to reconstruct. In this work, we explore the applicability of Bayesian DL techniques to model the uncertainty associated with DL-based reconstructions. In particular, we apply MC-dropout and heteroscedastic loss to the reconstruction networks to model epistemic and aleatoric uncertainty. We show that the proposed Bayesian methods achieve competitive performance when the test images are relatively far from the training data distribution and outperforms when the baseline method is over-parametrised. In addition, we qualitatively show that there seems to be a correlation between the magnitude of the produced uncertainty maps and the error maps, demonstrating the potential utility of the Bayesian DL methods for assessing the reliability of the reconstructed images.
UR - http://www.scopus.com/inward/record.url?scp=85053887042&partnerID=8YFLogxK
U2 - 10.1007/978-3-030-00129-2_8
DO - 10.1007/978-3-030-00129-2_8
M3 - Conference contribution
AN - SCOPUS:85053887042
SN - 9783030001285
T3 - Lecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics)
SP - 64
EP - 71
BT - Machine Learning for Medical Image Reconstruction - First International Workshop, MLMIR 2018, Held in Conjunction with MICCAI 2018, Proceedings
A2 - Knoll, Florian
A2 - Maier, Andreas
A2 - Rueckert, Daniel
PB - Springer Verlag
T2 - 1st Workshop on Machine Learning for Medical Image Reconstruction, MLMIR 2018 Held in Conjunction with 21st Medical Image Computing and Computer Assisted Intervention, MICCAI 2018
Y2 - 16 September 2018 through 16 September 2018
ER -