HDMamba:基于状态空间模型的低剂量CT去噪方法OA
HDMamba:State-space Model-based Denoising Method for Low-dose CT
在低剂量CT去噪任务中,针对基于自注意力机制的模型因二次计算复杂度高而导致显存占用过大,以及多尺度特征融合缺乏自适应性引发的高频细节丢失问题,提出一种新的去噪模型HDMamba.该模型以编码器-解码器为基础架构,将一个基于Haar小波变换的特征增强模块(HWFE)部署于整个架构的前端,为后续的处理过程提供更丰富的特征表达.在每层编码器-解码器中引入一个多尺度特征感知状态空间模块(MFSSM),以空间SSM替代传统自注意力机制,实现线性复杂度下的长程依赖捕捉,并在多尺度前馈网络的空洞卷积分支后引入可学习的参数α和β动态加权不同尺度特征.在AAPM数据集上的实验结果表明,本文模型在保持低参数量(31.13 M)的同时,其余各项指标显著优于一些现有的去噪方法.
In low-dose CT denoising tasks,to address the issues of excessive GPU memory consumption caused by the high com-putational complexity of the quadratic calculation in self-attention mechanism-based models and loss of high-frequency details due to lack of adaptability in multi-scale feature fusion,this paper proposes a novel denoising model called HDMamba.The model adopts an encoder-decoder architecture with a Haar wavelet-based feature enhancement(HWFE)module deployed at the front end to provide richer feature representations for subsequent processing.Within each encoder-decoder layer,a multi-scale feature-aware state space module(MFSSM)is introduced to replace traditional self-attention mechanisms,employing spatial SSMs to achieve linear-complexity long-range dependency capture.Additionally,learnable parameters α and β are incorporated after the dilated convolution branch in the multi-scale feed-forward network to dynamically weight features at different scales.Experimental results on the AAPM dataset demonstrate that the proposed model maintains a low parameter count(31.13 M)while significantly outperforming existing denoising methods in all other evaluation metrics.
张智轩;强彦;李青;闫心怡;段惠中
中北大学软件学院,山西 太原 030051中北大学软件学院,山西 太原 030051中北大学软件学院,山西 太原 030051中北大学软件学院,山西 太原 030051中北大学软件学院,山西 太原 030051
信息技术与安全科学
低剂量CT去噪状态空间模型空洞卷积小波变换多尺度特征融合
low-dose CT denoisingstate space modeldilated convolutionwavelet transformmulti-scale feature fusion
《计算机与现代化》 2026 (5)
9-16,24,9
中国博士后科学基金面上项目(2025M772904)国家资助博士后研究人员计划(GZC20241586)山西省基础研究计划资助项目(202403021212184)山西省高等学校科技创新项目(2024L181)
评论