Back to Search Start Over

High fidelity deep learning‐based MRI reconstruction with instance‐wise discriminative feature matching loss.

Authors :
Wang, Ke
Tamir, Jonathan I.
De Goyeneche, Alfredo
Wollner, Uri
Brada, Rafi
Yu, Stella X.
Lustig, Michael
Source :
Magnetic Resonance in Medicine; Jul2022, Vol. 88 Issue 1, p476-491, 16p
Publication Year :
2022

Abstract

Purpose: To improve reconstruction fidelity of fine structures and textures in deep learning‐ (DL) based reconstructions. Methods: A novel patch‐based Unsupervised Feature Loss (UFLoss) is proposed and incorporated into the training of DL‐based reconstruction frameworks in order to preserve perceptual similarity and high‐order statistics. The UFLoss provides instance‐level discrimination by mapping similar instances to similar low‐dimensional feature vectors and is trained without any human annotation. By adding an additional loss function on the low‐dimensional feature space during training, the reconstruction frameworks from under‐sampled or corrupted data can reproduce more realistic images that are closer to the original with finer textures, sharper edges, and improved overall image quality. The performance of the proposed UFLoss is demonstrated on unrolled networks for accelerated two‐ (2D) and three‐dimensional (3D) knee MRI reconstruction with retrospective under‐sampling. Quantitative metrics including normalized root mean squared error (NRMSE), structural similarity index (SSIM), and our proposed UFLoss were used to evaluate the performance of the proposed method and compare it with others. Results: In vivo experiments indicate that adding the UFLoss encourages sharper edges and more faithful contrasts compared to traditional and learning‐based methods with pure ℓ2$$ {\ell}_2 $$ loss. More detailed textures can be seen in both 2D and 3D knee MR images. Quantitative results indicate that reconstruction with UFLoss can provide comparable NRMSE and a higher SSIM while achieving a much lower UFLoss value. Conclusion: We present UFLoss, a patch‐based unsupervised learned feature loss, which allows the training of DL‐based reconstruction to obtain more detailed texture, finer features, and sharper edges with higher overall image quality under DL‐based reconstruction frameworks. (Code available at: https://github.com/mikgroup/UFLoss) [ABSTRACT FROM AUTHOR]

Details

Language :
English
ISSN :
07403194
Volume :
88
Issue :
1
Database :
Complementary Index
Journal :
Magnetic Resonance in Medicine
Publication Type :
Academic Journal
Accession number :
156556669
Full Text :
https://doi.org/10.1002/mrm.29227