太原市中心医院(太原 030009)
孙海刚,副主任医师。E-mail:sunhaigang2025@163.com
纸质出版:2026-04-20
移动端阅览
孙海刚,张叶飞,刘彦斌.面向尿路结石CT图像的DAST-UNet双重注意力自动分割模型[J].北京生物医学工程,2026,45(2):127-136.
SUN Haigang,ZHANG Yefei,LIU Yanbin.Automatic segmentation of urinary stones in CT images using DAST-UNet with dual attention mechanism[J].Beijing Biomedical Engineering,2026,45(2):127-136.
孙海刚,张叶飞,刘彦斌.面向尿路结石CT图像的DAST-UNet双重注意力自动分割模型[J].北京生物医学工程,2026,45(2):127-136. DOI: 10.3969/j.issn.1002-3208.2026.02.003.
SUN Haigang,ZHANG Yefei,LIU Yanbin.Automatic segmentation of urinary stones in CT images using DAST-UNet with dual attention mechanism[J].Beijing Biomedical Engineering,2026,45(2):127-136. DOI: 10.3969/j.issn.1002-3208.2026.02.003.
目的
2
腹部CT图像中尿路结石存在形态多样、边界模糊及与周围组织对比度低等问题,给自动分割任务带来了显著挑战,尤其在小目标检测与复杂背景干扰下,传统方法往往难以兼顾精度与鲁棒性。为此,本文提出一种融合双重注意力机制的DAST-UNet模型,用于实现尿路结石的高精度自动分割。
方法
2
所提模型以Swin-Unet为主干结构,设计了CASI(channel and spatial interactive)模块与TDA(token dependency attention)模块,分别从通道-空间维度和局部-全局语义关系两个层面增强模型特征表达能力。整体采用对称的编码器-解码器架构,通过多尺度特征融合与逐步上采样策略提升分割性能。其中,CASI模块增强模型对目标区域的关注并有效抑制背景噪声,TDA模块则通过加强Token之间的依赖关系,有效捕捉细粒度结构特征,优化上下文信息建模。
结果
2
在构建的尿路结石CT图像数据集上,DAST-UNet在多个关键性能指标上均优于Res Unet、SegNet与Swin-Unet等经典模型,最终在测试集上获得Dice系数80.34%、灵敏性80.21%、准确率89.91%和IoU 67.14%。训练过程中模型损失函数快速收敛,验证集loss与训练集趋势一致,表明网络训练过程稳定,未出现明显过拟合。消融实验进一步验证了CASI和TDA模块对性能提升的具体贡献,Swin-Unet+CASI模型在Dice、灵敏性等指标上有明显改进,继续引入TDA后性能进一步提升,最终DAST-UNet在所有组合中表现最佳。
结论
2
本文提出的DAST-UNet模型在结构上实现了对局部细节与全局上下文的协同建模,有效提升了CT图像中尿路结石的自动分割精度。实验结果充分验证了双重注意力机制对小目标识别与复杂背景处理的积极作用,为相关临床应用提供了一种可靠且具有推广价值的自动化解决方案。
Objective
2
Urinary tract stones in abdominal CT images present significant challenges for automatic segmentation, particularly due to their diverse morphologies, blurred boundaries, and low contrast with surrounding tissues. Traditional methods often struggle to achieve a balance of accuracy and robustness, particularly when detecting small objects and encountering complex background interference. Therefore, a DAST-UNet model is proposed that incorporates a dual attention mechanism to achieve high-precision automatic segmentation of urinary tract stones.
Methods
2
The proposed model utilizes the Swin-Unet backbone architecture and incorporates a CASI (channel and spatial interactive) module and a TDA (Token dependency attention) module to enhance the model’s feature representation capabilities from the channel-spatial dimension and the local-global semantic relationship level, respectively. A symmetric encoder-decoder architecture is employed to improve segmentation performance through multi-scale feature fusion and a progressive upsampling strategy. The CASI module enhances the model’s focus on the target region and effectively suppresses background noise, while the TDA module effectively captures fine-grained structural features and optimizes contextual information modeling by strengthening dependencies between Tokens.
Results
2
On a constructed urinary tract calculi CT image dataset, DAST-UNet outperforms classic models such as Res Unet, SegNet, and Swin-Unet across multiple key performance metrics, ultimately achieving a Dice coefficient of 80.34%, sensitivity of 80.21%, accuracy of 89.91%, and intersection over union (IoU) of 67.14% on the test set. The model loss function converges rapidly during training, with the validation set loss trend consistent with the training set, indicating that the network training process is stable and free of significant overfitting. Ablation experiments further validate the specific contributions of the CASI and TDA modules to performance improvement. The Swin-Unet + CASI model shows significant improvements in metrics such as Dice and sensitivity. The introduction of TDA further improves performance, resulting in DAST-UNet performing the best among all combinations.
Conclusions
2
The proposed DAST-UNet model achieves a coordinated modeling of local details and global context, effectively improving the accuracy of automatic segmentation of urinary tract calculi in CT images. The experimental results fully verify the positive effect of the dual attention mechanism on small target recognition and complex background processing, and provide a reliable and scalable automated solution for related clinical applications.
Yildirim K, Bozdag PG, Talo M, et al. Deep learning model for automated kidney stone detection using coronal CT images[J]. Computers in Biology and Medicine, 2021, 135: 104569.
Asif S, Zheng X, Zhu Y. An optimized fusion of deep learning models for kidney stone detection from CT images[J]. Journal of King Saud University-Computer and Information Sciences, 2024, 36(7): 102130.
El Beze J, Mazeaud C, Daul C, et al. Evaluation and understanding of automated urinary stone recognition methods[J]. BJU International, 2022, 130(6): 786-798.
Babajide R, Lembrikova K, Ziemba J, et al. Automated machine learning segmentation and measurement of urinary stones on CT scan[J]. Urology, 2022, 169: 41-46.
Huang ZH, Liu YY, Wu WJ, et al. Design and validation of a deep learning model for renal stone detection and segmentation on kidney-ureter-bladder images[J]. Bioengineering, 2023, 10(8): 970.
Kilic U, Karabey Aksakalli I, Tumuklu Ozyer G, et al. Exploring the effect of image enhancement techniques with deep neural networks on direct urinary system (DUSX) images for automated kidney stone detection[J]. International Journal of Intelligent Systems, 2023, 2023: 3801485.
Shetty M, Shetty SB, Sequeria GP, et al. Kidney stone detection using CNN[C]//2024 Second International Conference on Data Science and Information System (ICDSIS). Hassan, India: IEEE Press, 2024: 1-6.
Preedanan W, Suzuki K, Kondo T, et al. Improvement of urinary stone segmentation using GAN-based urinary stones inpainting augmentation[J]. IEEE Access, 2022, 10: 115131-115142.
Li D, Xiao C, Liu Y, et al. Deep segmentation networks for segmenting kidneys and detecting kidney stones in unenhanced abdominal CT images[J]. Diagnostics, 2022, 12(8): 1788.
Choi HS, Kim JS, Whangbo TK, et al. Improved detection of urolithiasis using high-resolution computed tomography images by a vision transformer model[J]. International Neurourology Journal, 2023, 27(Suppl 2): S99-S103.
Islam MN, Hasan M, Hossain MK, et al. Vision transformer and explainable transfer learning models for auto detection of kidney cyst, stone and tumor from CT-radiography[J]. Scientific Reports, 2022, 12(1): 11440.
Yao J, Jin S. Multi-category segmentation of Sentinel-2 images based on the Swin UNet method[J]. Remote Sensing, 2022, 14: 3382.
Du J, Guan K, Zhou Y, et al. Parameter-free similarity-aware attention module for medical image classification and segmentation[J]. IEEE Transactions on Emerging Topics in Computational Intelligence, 2023, 7(3): 845-857.
0
浏览量
6
下载量
0
CSCD
关联资源
相关文章
相关作者
相关机构
京公网安备11010802024621