PWD: Prior-Guided and Wavelet-Enhanced Diffusion Model for Limited-Angle CT

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Liu, Yi, Wen, Yiyang, Zhou, Zekun, Ma, Junqi, Wang, Linghang, Yao, Yucheng, Shi, Liu, Liu, Qiegen
Natura: Preprint
Pubblicazione: 2025
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866913935709437952
author Liu, Yi
Wen, Yiyang
Zhou, Zekun
Ma, Junqi
Wang, Linghang
Yao, Yucheng
Shi, Liu
Liu, Qiegen
author_facet Liu, Yi
Wen, Yiyang
Zhou, Zekun
Ma, Junqi
Wang, Linghang
Yao, Yucheng
Shi, Liu
Liu, Qiegen
contents Generative diffusion models have received increasing attention in medical imaging, particularly in limited-angle computed tomography (LACT). Standard diffusion models achieve high-quality image reconstruction but require a large number of sampling steps during inference, resulting in substantial computational overhead. Although skip-sampling strategies have been proposed to improve efficiency, they often lead to loss of fine structural details. To address this issue, we propose a prior information embedding and wavelet feature fusion fast sampling diffusion model for LACT reconstruction. The PWD enables efficient sampling while preserving reconstruction fidelity in LACT, and effectively mitigates the degradation typically introduced by skip-sampling. Specifically, during the training phase, PWD maps the distribution of LACT images to that of fully sampled target images, enabling the model to learn structural correspondences between them. During inference, the LACT image serves as an explicit prior to guide the sampling trajectory, allowing for high-quality reconstruction with significantly fewer steps. In addition, PWD performs multi-scale feature fusion in the wavelet domain, effectively enhancing the reconstruction of fine details by leveraging both low-frequency and high-frequency information. Quantitative and qualitative evaluations on clinical dental arch CBCT and periapical datasets demonstrate that PWD outperforms existing methods under the same sampling condition. Using only 50 sampling steps, PWD achieves at least 1.7 dB improvement in PSNR and 10% gain in SSIM.
format Preprint
id arxiv_https___arxiv_org_abs_2507_05317
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle PWD: Prior-Guided and Wavelet-Enhanced Diffusion Model for Limited-Angle CT
Liu, Yi
Wen, Yiyang
Zhou, Zekun
Ma, Junqi
Wang, Linghang
Yao, Yucheng
Shi, Liu
Liu, Qiegen
Image and Video Processing
Artificial Intelligence
Computer Vision and Pattern Recognition
Generative diffusion models have received increasing attention in medical imaging, particularly in limited-angle computed tomography (LACT). Standard diffusion models achieve high-quality image reconstruction but require a large number of sampling steps during inference, resulting in substantial computational overhead. Although skip-sampling strategies have been proposed to improve efficiency, they often lead to loss of fine structural details. To address this issue, we propose a prior information embedding and wavelet feature fusion fast sampling diffusion model for LACT reconstruction. The PWD enables efficient sampling while preserving reconstruction fidelity in LACT, and effectively mitigates the degradation typically introduced by skip-sampling. Specifically, during the training phase, PWD maps the distribution of LACT images to that of fully sampled target images, enabling the model to learn structural correspondences between them. During inference, the LACT image serves as an explicit prior to guide the sampling trajectory, allowing for high-quality reconstruction with significantly fewer steps. In addition, PWD performs multi-scale feature fusion in the wavelet domain, effectively enhancing the reconstruction of fine details by leveraging both low-frequency and high-frequency information. Quantitative and qualitative evaluations on clinical dental arch CBCT and periapical datasets demonstrate that PWD outperforms existing methods under the same sampling condition. Using only 50 sampling steps, PWD achieves at least 1.7 dB improvement in PSNR and 10% gain in SSIM.
title PWD: Prior-Guided and Wavelet-Enhanced Diffusion Model for Limited-Angle CT
topic Image and Video Processing
Artificial Intelligence
Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2507.05317