WDBDM: Wavelet-based dual-branch diffusion model for low-dose CT and PET denoising.
Authors
Affiliations (10)
Affiliations (10)
- School of Information Science and Engineering, Lanzhou University, Lanzhou 730000, China; Gansu Provincial Key Laboratory of Wearable Computing, Lanzhou University, Lanzhou 730000, China. Electronic address: [email protected].
- School of Information Science and Engineering, Lanzhou University, Lanzhou 730000, China; Gansu Provincial Key Laboratory of Wearable Computing, Lanzhou University, Lanzhou 730000, China. Electronic address: [email protected].
- School of Information Science and Engineering, Lanzhou University, Lanzhou 730000, China; Gansu Provincial Key Laboratory of Wearable Computing, Lanzhou University, Lanzhou 730000, China. Electronic address: [email protected].
- School of Computing and Data Science, The University of Hong Kong, Hong Kong, China. Electronic address: [email protected].
- Faculty of Medical Imaging Technology, College of Health Science and Technology, Shanghai Jiao Tong University School of Medicine, Shanghai 200240, China. Electronic address: [email protected].
- School of Computing and Data Science, The University of Hong Kong, Hong Kong, China. Electronic address: [email protected].
- Department of Nuclear Medicine, University of Bern, Bern 3012, Switzerland. Electronic address: [email protected].
- School of Information Science and Engineering, Lanzhou University, Lanzhou 730000, China; Gansu Provincial Key Laboratory of Wearable Computing, Lanzhou University, Lanzhou 730000, China. Electronic address: [email protected].
- School of Information Science and Engineering, Lanzhou University, Lanzhou 730000, China; Gansu Provincial Key Laboratory of Wearable Computing, Lanzhou University, Lanzhou 730000, China. Electronic address: [email protected].
- Gansu Provincial Key Laboratory of Wearable Computing, Lanzhou University, Lanzhou 730000, China; School of Medical Technology, Beijing Institute of Technology, Beijing 100081, China. Electronic address: [email protected].
Abstract
X-ray computed tomography (CT) and positron emission tomography (PET) are imaging modalities for assessing various diseases. While normal-dose CT (NDCT) and normal-dose PET (NDPET) imaging ensure high image quality, they often raise concerns regarding potential health risks from radiation exposure. This contradiction between reducing radiation dose and preserving diagnostic performance can be effectively addressed by reconstructing low-dose CT and low-dose PET images into high-quality images comparable to their normal-dose counterparts. In this study, we present WDBDM, a wavelet-based dual-branch diffusion framework for denoising low-dose data to generate normal-dose quality images. WDBDM consists of four main parts: Discrete Wavelet Transform (DWT), Low-Frequency Diffusion Branch (LFDB), High-Frequency Diffusion Branch (HFDB) and Fusion Module. Furthermore, we investigate the effectiveness of the Fusion Spatial-Frequency Convolution Module (FSFCM) in the diffusion branch, which can jointly extract spatial and frequency domain information, thereby significantly enhancing the feature representation capability of the model. Moreover, to prevent error propagation from imperfect recovery operators and to enable bidirectional guidance across high- and low-frequency components during sampling, we integrate HLF-MEMNet, a novel recovery network, into the WDBDM framework. This network leverages contextual information and high-low frequency mutual guidance constraints during the sampling process, preventing structural distortion and ensuring better alignment with the input at the next time step. Experiments on four public datasets and two imaging modalities demonstrate that WDBDM outperforms existing methods in denoising performance and generalization ability.