通过信噪比信息训练,提升MRI去噪模型在不同场景下的性能与泛化能力。
SNRAware: Improved Deep Learning MRI Denoising with SNR Unit Training and G-factor Map Augmentation
- 利用重建过程中的噪声分布信息生成合成数据,指导模型学习更精准的去噪机制。
- 在3T心脏电影影像上训练的模型,在1.5T实时心动图和灌注成像中实现6.5倍和2.9倍信噪比提升。
- 仅用心脏电影数据训练的模型也能有效泛化到神经和脊柱MRI,适合跨模态应用。
为开发并评估一种新型深度学习MRI去噪方法,该方法利用重建过程中定量的噪声分布信息以提升去噪性能与泛化能力。本回顾性研究在96,605例3T心脏延迟触发电影序列的2,885,236张图像上,训练了14种基于Transformer和卷积结构的模型。提出的训练方案SNRAware通过模拟高质量、多样化的合成数据集,并向模型提供噪声分布的定量信息,改进去噪效果。在3000样本的保留测试集上使用PSNR和SSIM评估性能,并与无噪声增强的对比实验进行消融分析。跨分布测试涵盖心脏实时电影、首次通过灌注、神经及脊柱MRI,均在1.5T下采集,用于检验模型在不同成像序列、动态对比度、解剖结构和场强下的泛化能力。最优模型在分布内测试中表现优异,并成功泛化至分布外样本,使实时电影和灌注成像的CNR分别提升6.5倍和2.9倍。此外,仅使用100%心脏电影数据训练的模型,也良好适配于T1 MPRAGE神经3D扫描和T2 TSE脊柱MRI。
原文摘要 · Abstract (English)
To develop and evaluate a new deep learning MR denoising method that leverages quantitative noise distribution information from the reconstruction process to improve denoising performance and generalization. This retrospective study trained 14 different transformer and convolutional models with two backbone architectures on a large dataset of 2,885,236 images from 96,605 cardiac retro-gated cine complex series acquired at 3T. The proposed training scheme, termed SNRAware, leverages knowledge of the MRI reconstruction process to improve denoising performance by simulating large, high quality, and diverse synthetic datasets, and providing quantitative information about the noise distribution to the model. In-distribution testing was performed on a hold-out dataset of 3000 samples with performance measured using PSNR and SSIM, with ablation comparison without the noise augmentation. Out-of-distribution tests were conducted on cardiac real-time cine, first-pass cardiac perfusion, and neuro and spine MRI, all acquired at 1.5T, to test model generalization across imaging sequences, dynamically changing contrast, different anatomies, and field strengths. The best model found in the in-distribution test generalized well to out-of-distribution samples, delivering 6.5x and 2.9x CNR improvement for real-time cine and perfusion imaging, respectively. Further, a model trained with 100% cardiac cine data generalized well to a T1 MPRAGE neuro 3D scan and T2 TSE spine MRI.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。