论文标题
神经架构搜索压缩感测磁共振图像重建
Neural Architecture Search for Compressed Sensing Magnetic Resonance Image Reconstruction
论文作者
论文摘要
最近的工作表明,基于深度学习(DL)的压缩感测(CS)实现可以通过从子采样的K-Space数据中重建MR图像来加速磁共振(MR)成像。但是,以前方法中采用的网络体系结构都是由手工制作设计的。神经体系结构搜索(NAS)算法可以自动构建神经网络体系结构,在几个视觉任务中表现优于人类设计的神经网络体系结构。受此启发,我们在这里提出了一个新颖有效的网络,用于通过NAS而不是手动尝试,以解决MR图像重建问题。尤其是,以模型驱动的MR重建管道集成到模型驱动的MR重建管道中的特定细胞结构会以可区分的方式自动从灵活的预定义搜索空间中搜索。实验结果表明,与以前的最先进方法相比,在PSNR和SSIM方面,我们的搜索网络可以产生更好的重建结果,其计算资源少4-6倍。进行了广泛的实验,以分析超参数如何影响重建性能和搜索结构。还在不同的器官MR数据集上评估了搜索架构的普遍性。我们提出的方法可以在MR重建问题的计算成本和重建性能之间进行更好的权衡,并具有良好的推广性,并为其他医学图像应用设计神经网络提供见解。评估代码将在https://github.com/yjump/nas-for-csmri上找到。
Recent works have demonstrated that deep learning (DL) based compressed sensing (CS) implementation can accelerate Magnetic Resonance (MR) Imaging by reconstructing MR images from sub-sampled k-space data. However, network architectures adopted in previous methods are all designed by handcraft. Neural Architecture Search (NAS) algorithms can automatically build neural network architectures which have outperformed human designed ones in several vision tasks. Inspired by this, here we proposed a novel and efficient network for the MR image reconstruction problem via NAS instead of manual attempts. Particularly, a specific cell structure, which was integrated into the model-driven MR reconstruction pipeline, was automatically searched from a flexible pre-defined operation search space in a differentiable manner. Experimental results show that our searched network can produce better reconstruction results compared to previous state-of-the-art methods in terms of PSNR and SSIM with 4-6 times fewer computation resources. Extensive experiments were conducted to analyze how hyper-parameters affect reconstruction performance and the searched structures. The generalizability of the searched architecture was also evaluated on different organ MR datasets. Our proposed method can reach a better trade-off between computation cost and reconstruction performance for MR reconstruction problem with good generalizability and offer insights to design neural networks for other medical image applications. The evaluation code will be available at https://github.com/yjump/NAS-for-CSMRI.