LUCMT: Learnable under-sampling and reconstructed network with cross multi-head attention transformer for accelerating MR image reconstruction
Ziqi Yang,Mingfeng Jiang,Dongshen Ruan,Yang Li,Tao Tan,Sumei Huang,Feng Liu
DOI: https://doi.org/10.1016/j.cmpb.2024.108359
Abstract:Background and objective: As a widely used technique for Magnetic Resonance Image (MRI) acceleration, compressed sensing MRI involves two main issues: designing an effective sampling strategy and reconstructing the image from significantly under-sampled K-space data. In this paper, an innovative approach is proposed to address these two challenges simultaneously. Methods: A novel MRI reconstruction method, termed as LUCMT, is implemented by integrating a learnable under-sampling strategy with a reconstruction network based on the Cross Multi-head Attention Transformer. In contrast to conventional static sampling methods, the proposed adaptive sampling scheme is processed optimally by learning the optimal sampling technique, which involves binarizing the sampling pattern by a sigmoid function and computing gradients by backpropagation. And the reconstruction network is designed by using CS-MRI depth unfolding network that incorporates a Cross Multi-head Attention (CMA) module with inertial and gradient descent terms. Results: T1 brain MR images from the FastMRI dataset are used to validate the performance of the proposed method. A series of experiments are conducted to validate the superior performance of our proposed network in terms of quantitative metrics and visual quality. Compared with other state-of-the-art reconstruction methods, LUCMT achieves better reconstruction performances with more accurate details. Specifically, LUCMT achieves PSNR and SSIM results of 41.87/0.9749, 46.64/0.9868, 50.41/0.9924, and 53.51/0.9955 at sampling rates of 10 %, 20 %, 30 %, and 40 %, respectively. Conclusions: The proposed LUCMT method can provide a promising way for generating optimal under-sampling mask and accelerating MRI reconstruction accurately.