论文标题
一个密集的互连网络,用于深度学习加速MRI
A Densely Interconnected Network for Deep Learning Accelerated MRI
论文作者
论文摘要
目的:通过密集连接的深度学习重建框架来改善加速的MRI重建。 材料和方法:通过应用三个架构修改来修改级联的深度学习重建框架(基线模型):级联输入和输出之间的输入级级密集的连接,改进的深度学习子网络和随后深度学习网络之间的远程SKIP连接。进行了一项消融研究,其中在NYU FastMRI Neuro数据集上训练了五个模型配置,并在四倍和八倍的加速度上结合了端到端方案。通过比较其各自的结构相似性指数度量(SSIM),归一化平方误差(NMSE)和峰信号与噪声比(PSNR)来评估训练的模型。 结果:提出的密集互连的残留级联网络(DIRCN)利用了所有三种建议的修改,分别在四倍和八倍的加速度方面提高了8%和11%的SSIM。对于八倍的加速度,与基线模型相比,该模型的NMSE下降了23%。在一项消融研究中,单个体系结构的修饰都通过分别减少SSIM和NMSE的分别为4倍加速度的SSIM和NMSE,从而有助于这一改进。 结论:拟议的体系结构修改允许对已经存在的级联框架进行简单调整,以进一步改善所得的重建。
Objective: To improve accelerated MRI reconstruction through a densely connected cascading deep learning reconstruction framework. Materials and Methods: A cascading deep learning reconstruction framework (baseline model) was modified by applying three architectural modifications: Input-level dense connections between cascade inputs and outputs, an improved deep learning sub-network, and long-range skip-connections between subsequent deep learning networks. An ablation study was performed, where five model configurations were trained on the NYU fastMRI neuro dataset with an end-to-end scheme conjunct on four- and eight-fold acceleration. The trained models were evaluated by comparing their respective structural similarity index measure (SSIM), normalized mean square error (NMSE) and peak signal to noise ratio (PSNR). Results: The proposed densely interconnected residual cascading network (DIRCN), utilizing all three suggested modifications, achieved a SSIM improvement of 8% and 11% for four- and eight-fold acceleration, respectively. For eight-fold acceleration, the model achieved a 23% decrease in the NMSE when compared to the baseline model. In an ablation study, the individual architectural modifications all contributed to this improvement, by reducing the SSIM and NMSE with approximately 3% and 5% for four-fold acceleration, respectively. Conclusion: The proposed architectural modifications allow for simple adjustments on an already existing cascading framework to further improve the resulting reconstructions.