Using Diffusion Model to Generate High-Resolution MRI

Chih-Wei Chang,Shaoyan Pan,Junbo Peng,Elahheh Salari,Justin Roper,Richard Qiu,Yuan Gao,Tian Liu,Hui-Kuo Shu,Hui Mao,Xiaofeng Yang
DOI: https://doi.org/10.1117/12.3006586
2024-01-01
Abstract:This study aims to enhance the resolution of Magnetic Resonance Imaging (MRI) using a cutting-edge diffusion probabilistic Deep Learning (DL) technique, addressing the challenges posed by long image acquisition times and limited scanning dimensions. In this research, we propose a novel approach utilizing a probabilistic DL model to synthesize High-Resolution MRI (HR-MRI) images from Low-Resolution (LR) inputs. The proposed model consists of two main steps. In the forward process, Gaussian noise is systematically introduced to LR images through a Markov chain. In the reverse process, a U-Net model is trained using a loss function based on Kullback-Leibler divergence, which maximizes the likelihood of producing ground truth images. We assess the effectiveness of our method on T2-FLAIR images from 120 brain patients in the public BraTS2020 T2-FLAIR database. To gauge performance, we compare our approach with a clinical bicubic model (referred to as Bicubic) and Conditional Generative Adversarial Networks (CGAN). On the BraTS2020 dataset, our framework enhances the Peak Signal-to-Noise Ratio (PSNR) of LR images by 7%, whereas CGAN results in a 3% reduction. The corresponding Multi-scale Structural similarity (MSSIM) values for the proposed method and CGAN are 0.972±0.017 and 0.966±0.024. In this study, we have examined the potential of a diffusion probabilistic DL framework to elevate MRI image resolution. Our proposed method demonstrates the capability to generate high-quality HR images while avoiding issues such as mode collapse or learning multimodal distributions, which are commonly observed in CGAN-based approaches. This framework has the potential to significantly reduce MRI acquisition times for HR imaging, thereby mitigating the risk of motion artifacts and crosstalk.
What problem does this paper attempt to address?