Deep Learning (DL) has shown potential in accelerating Magnetic Resonance Image acquisition and reconstruction. Nevertheless, there is a dearth of tailored methods to guarantee that the reconstruction of small features is achieved with high fidelity. In this work, we employ adversarial attacks to generate small synthetic perturbations, which are difficult to reconstruct for a trained DL reconstruction network. Then, we use robust training to increase the network’s sensitivity to these small features and encourage their reconstruction. Next, we investigate the generalization of said approach to real world features. For this, a musculoskeletal radiologist annotated a set of cartilage and meniscal lesions from the knee Fast-MRI dataset, and a classification network was devised to assess the reconstruction of the features. Experimental results show that by introducing robust training to a reconstruction network, the rate of false negative features (4.8%) in image reconstruction can be reduced. These results are encouraging, and highlight the necessity for attention to this problem by the image reconstruction community, as a milestone for the introduction of DL reconstruction in clinical practice. To support further research, we make our annotations and code publicly available at https://github.com/fcaliva/fastMRI_BB_abnormalities_annotation.
mri reconstruction · adversarial attack · robust training · abnormality detection · fast mri
@article{melba:2021:007:calivá,
title = "Adversarial Robust Training of Deep Learning MRI Reconstruction Models",
author = "Calivá, Francesco and Cheng, Kaiyang and Shah, Rutwik and Pedoia, Valentina",
journal = "Machine Learning for Biomedical Imaging",
volume = "1",
issue = "MIDL 2020 special issue",
year = "2021",
pages = "1--32",
issn = "2766-905X",
doi = "https://doi.org/10.59275/j.melba.2021-df47",
url = "https://melba-journal.org/2021:007"
}
TY - JOUR
AU - Calivá, Francesco
AU - Cheng, Kaiyang
AU - Shah, Rutwik
AU - Pedoia, Valentina
PY - 2021
TI - Adversarial Robust Training of Deep Learning MRI Reconstruction Models
T2 - Machine Learning for Biomedical Imaging
VL - 1
IS - MIDL 2020 special issue
SP - 1
EP - 32
SN - 2766-905X
DO - https://doi.org/10.59275/j.melba.2021-df47
UR - https://melba-journal.org/2021:007
ER -