arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2201.03230eess.IVcs.AIcs.CVcs.LG

Swin Transformer for Fast MRI

Imperial College London(帝国理工学院);Royal Brompton Hospital(皇家布朗普顿医院);Sun Yat-sen University(中山大学);Beihang University(北京航空航天大学);University of the Basque Country UPV/EHU(巴斯克大学(UPV/EHU));TECNALIA(泰克纳利亚研究中心);Shenzhen Second People’s Hospital(深圳市第二人民医院);The First Affiliated Hospital of Shenzhen University Health Science Center(深圳大学医学部第一附属医院)

Jiahao Huang, Yingying Fang, Yinzhe Wu, Huanjun Wu, Zhifan Gao, Yang Li, Javier Del Ser, Jun Xia, Guang Yang

英文摘要:

Magnetic resonance imaging (MRI) is an important non-invasive clinical tool that can produce high-resolution and reproducible images. However, a long scanning time is required for high-quality MR images, which leads to exhaustion and discomfort of patients, inducing more artefacts due to voluntary movements of the patients and involuntary physiological movements. To accelerate the scanning process, methods by k-space undersampling and deep learning based reconstruction have been popularised. This work introduced SwinMR, a novel Swin transformer based method for fast MRI reconstruction. The whole network consisted of an input module (IM), a feature extraction module (FEM) and an output module (OM). The IM and OM were 2D convolutional layers and the FEM was composed of a cascaded of residual Swin transformer blocks (RSTBs) and 2D convolutional layers. The RSTB consisted of a series of Swin transformer layers (STLs). The shifted windows multi-head self-attention (W-MSA/SW-MSA) of STL was performed in shifted windows rather than the multi-head self-attention (MSA) of the original transformer in the whole image space. A novel multi-channel loss was proposed by using the sensitivity maps, which was proved to reserve more textures and details. We performed a series of comparative studies and ablation studies in the Calgary-Campinas public brain MR dataset and conducted a downstream segmentation experiment in the Multi-modal Brain Tumour Segmentation Challenge 2017 dataset. The results demonstrate our SwinMR achieved high-quality reconstruction compared with other benchmark methods, and it shows great robustness with different undersampling masks, under noise interruption and on different datasets. The code is publicly available at https://github.com/ayanglab/SwinMR.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑