SNRAware:基于信噪比单位训练和g因子图增强的深度学习MRI去噪方法改进

SNRAware: Improved Deep Learning MRI Denoising with SNR Unit Training and G-factor Map Augmentation

摘要 Abstract

开发并评估一种新的利用MRI重建过程中定量噪声分布信息的深度学习MR去噪方法,以提高去噪性能和泛化能力。本回顾性研究在来自96,605例心脏回波门控电影序列(共计2,885,236张图像)的大规模数据集上,采用两种主干架构训练了14种不同的Transformer和卷积模型。该方法称为SNRAware,通过模拟大规模高质量且多样化的合成数据集,并向模型提供噪声分布的定量信息,利用MRI重建过程中的知识提升去噪性能。在分布内测试中,对3,000个样本的保留数据集进行测试,使用PSNR和SSIM衡量性能,并对比无噪声增强的消融实验。分布外测试则在1.5T采集的心脏实时电影、首过心脏灌注以及神经和脊柱MRI上进行,用于检验模型在不同成像序列、动态变化对比度、不同解剖结构和磁场强度下的泛化能力。分布内测试中表现最佳的模型在分布外样本上表现出色,分别使实时电影和灌注成像的CNR提高了6.5倍和2.9倍。此外,仅使用100%心脏电影数据训练的模型在1.5T采集的T1 MPRAGE神经三维扫描和T2 TSE脊柱MRI上也具有良好的泛化能力。

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.