针对超声甲状腺结节检测难题,提出新型注意力机制提升小结节识别能力。
Nodule-DETR: A Novel DETR Architecture with Frequency-Channel Attention for Ultrasound Thyroid Nodule Detection
- 引入频域通道注意力与多尺度可变形注意力,增强低对比度结节特征
- 在真实临床数据集上实现[email protected]:0.95提升0.149,达当前最优
- 适合医疗影像辅助诊断场景,尤其对边界模糊的小结节检测有优势
甲状腺癌是全球最常见的内分泌恶性肿瘤,发病率持续上升。超声虽为甲状腺结节检测的首选影像方式,但受限于图像对比度低、结节边界模糊等问题,诊断准确率常受影响。为此,本文提出Nodule-DETR,一种专用于超声图像中甲状腺结节检测的新型检测变压器(DETR)架构。该模型引入三项创新:多光谱频域通道注意力(MSFCA)模块,利用频域分析强化低对比度结节特征;分层特征融合(HFF)模块,实现高效多尺度信息整合;以及多尺度可变形注意力(MSDA),灵活捕捉小尺寸及不规则形状结节。我们在包含真实世界甲状腺超声图像的临床数据集上进行了大量实验,结果表明,Nodule-DETR性能达到当前最优水平,在[email protected]:0.95指标上相比基线模型提升0.149。该模型优异的准确性凸显其在计算机辅助甲状腺诊断中的重要应用潜力。代码已开源:https://github.com/wjj1wjj/Nodule-DETR。
原文摘要 · Abstract (English)
Thyroid cancer is the most common endocrine malignancy, and its incidence is rising globally. While ultrasound is the preferred imaging modality for detecting thyroid nodules, its diagnostic accuracy is often limited by challenges such as low image contrast and blurred nodule boundaries. To address these issues, we propose Nodule-DETR, a novel detection transformer (DETR) architecture designed for robust thyroid nodule detection in ultrasound images. Nodule-DETR introduces three key innovations: a Multi-Spectral Frequency-domain Channel Attention (MSFCA) module that leverages frequency analysis to enhance features of low-contrast nodules; a Hierarchical Feature Fusion (HFF) module for efficient multi-scale integration; and Multi-Scale Deformable Attention (MSDA) to flexibly capture small and irregularly shaped nodules. We conducted extensive experiments on a clinical dataset of real-world thyroid ultrasound images. The results demonstrate that Nodule-DETR achieves state-of-the-art performance, outperforming the baseline model by a significant margin of 0.149 in [email protected]:0.95. The superior accuracy of Nodule-DETR highlights its significant potential for clinical application as an effective tool in computer-aided thyroid diagnosis. The code of work is available at https://github.com/wjj1wjj/Nodule-DETR.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。