TC-DiffRecon: Texture coordination MRI reconstruction method based on diffusion model and modified MF-UNet method

Fuente: arXiv
Enregistré dans:
Détails bibliographiques
Auteurs principaux: Zhang, Chenyan, Chen, Yifei, Fan, Zhenxiong, Huang, Yiyu, Weng, Wenchao, Ge, Ruiquan, Zeng, Dong, Wang, Changmiao
Format: Preprint
Publié: 2024
Sujets:
Accès en ligne:
Tags: Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
_version_ 1866914683347271680
author Zhang, Chenyan
Chen, Yifei
Fan, Zhenxiong
Huang, Yiyu
Weng, Wenchao
Ge, Ruiquan
Zeng, Dong
Wang, Changmiao
author_facet Zhang, Chenyan
Chen, Yifei
Fan, Zhenxiong
Huang, Yiyu
Weng, Wenchao
Ge, Ruiquan
Zeng, Dong
Wang, Changmiao
contents Recently, diffusion models have gained significant attention as a novel set of deep learning-based generative methods. These models attempt to sample data from a Gaussian distribution that adheres to a target distribution, and have been successfully adapted to the reconstruction of MRI data. However, as an unconditional generative model, the diffusion model typically disrupts image coordination because of the consistent projection of data introduced by conditional bootstrap. This often results in image fragmentation and incoherence. Furthermore, the inherent limitations of the diffusion model often lead to excessive smoothing of the generated images. In the same vein, some deep learning-based models often suffer from poor generalization performance, meaning their effectiveness is greatly affected by different acceleration factors. To address these challenges, we propose a novel diffusion model-based MRI reconstruction method, named TC-DiffRecon, which does not rely on a specific acceleration factor for training. We also suggest the incorporation of the MF-UNet module, designed to enhance the quality of MRI images generated by the model while mitigating the over-smoothing issue to a certain extent. During the image generation sampling process, we employ a novel TCKG module and a Coarse-to-Fine sampling scheme. These additions aim to harmonize image texture, expedite the sampling process, while achieving data consistency. Our source code is available at https://github.com/JustlfC03/TC-DiffRecon.
format Preprint
id arxiv_https___arxiv_org_abs_2402_11274
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle TC-DiffRecon: Texture coordination MRI reconstruction method based on diffusion model and modified MF-UNet method
Zhang, Chenyan
Chen, Yifei
Fan, Zhenxiong
Huang, Yiyu
Weng, Wenchao
Ge, Ruiquan
Zeng, Dong
Wang, Changmiao
Image and Video Processing
Computer Vision and Pattern Recognition
Machine Learning
Recently, diffusion models have gained significant attention as a novel set of deep learning-based generative methods. These models attempt to sample data from a Gaussian distribution that adheres to a target distribution, and have been successfully adapted to the reconstruction of MRI data. However, as an unconditional generative model, the diffusion model typically disrupts image coordination because of the consistent projection of data introduced by conditional bootstrap. This often results in image fragmentation and incoherence. Furthermore, the inherent limitations of the diffusion model often lead to excessive smoothing of the generated images. In the same vein, some deep learning-based models often suffer from poor generalization performance, meaning their effectiveness is greatly affected by different acceleration factors. To address these challenges, we propose a novel diffusion model-based MRI reconstruction method, named TC-DiffRecon, which does not rely on a specific acceleration factor for training. We also suggest the incorporation of the MF-UNet module, designed to enhance the quality of MRI images generated by the model while mitigating the over-smoothing issue to a certain extent. During the image generation sampling process, we employ a novel TCKG module and a Coarse-to-Fine sampling scheme. These additions aim to harmonize image texture, expedite the sampling process, while achieving data consistency. Our source code is available at https://github.com/JustlfC03/TC-DiffRecon.
title TC-DiffRecon: Texture coordination MRI reconstruction method based on diffusion model and modified MF-UNet method
topic Image and Video Processing
Computer Vision and Pattern Recognition
Machine Learning
url https://arxiv.org/abs/2402.11274