用先验知识增强DETR,提升超声结节检测精度
Prior-Guided DETR for Ultrasound Nodule Detection
- 分阶段注入几何与结构先验,改善不规则结节特征提取
- 在4个数据集上超越18种方法,复杂形态结节检测效果更优
- 适合医学影像检测研究者与超声诊断系统开发者
准确检测超声结节对甲状腺和乳腺癌的早期诊断与治疗至关重要。但不规则形状、边界模糊、尺度差异大及斑点噪声干扰导致检测困难。为此,我们提出一种面向超声结节检测的先验引导DETR框架。该框架在多阶段逐步引入不同先验知识:首先,在CNN主干中嵌入带先验正则化的空间自适应可变形FFN(SDFPR),将几何先验注入可变形采样,稳定对不规则模糊结节的特征提取;其次,设计多尺度空间-频率特征混合器(MSFFM),空间域处理强调轮廓连续性与边界线索,频域建模捕捉全局形态并抑制斑点噪声;此外,密集特征交互(DFI)机制在所有编码层传播与利用这些先验调制的特征,使解码器在一致的几何与结构引导下优化查询。在两个临床收集的甲状腺超声数据集(Thyroid I, Thyroid II)和两个公开基准(TN3K, BUSI)上的实验表明,所提方法在18种对比方法中表现最优,尤其在复杂形态结节检测上优势显著。代码已开源。
原文摘要 · Abstract (English)
Accurate detection of ultrasound nodules is essential for the early diagnosis and treatment of thyroid and breast cancers. However, this task remains challenging due to irregular nodule shapes, indistinct boundaries, substantial scale variations, and the presence of speckle noise that degrades structural visibility. To address these challenges, we propose a prior-guided DETR framework specifically designed for ultrasound nodule detection. Instead of relying on purely data-driven feature learning, the proposed framework progressively incorporates different prior knowledge at multiple stages of the network. First, a Spatially-adaptive Deformable FFN with Prior Regularization (SDFPR) is embedded into the CNN backbone to inject geometric priors into deformable sampling, stabilizing feature extraction for irregular and blurred nodules. Second, a Multi-scale Spatial-Frequency Feature Mixer (MSFFM) is designed to extract multi-scale structural priors, where spatial-domain processing emphasizes contour continuity and boundary cues, while frequency-domain modeling captures global morphology and suppresses speckle noise. Furthermore, a Dense Feature Interaction (DFI) mechanism propagates and exploits these prior-modulated features across all encoder layers, enabling the decoder to enhance query refinement under consistent geometric and structural guidance. Experiments conducted on two clinically collected thyroid ultrasound datasets (Thyroid I and Thyroid II) and two public benchmarks (TN3K and BUSI) for thyroid and breast nodules demonstrate that the proposed method achieves superior accuracy compared with 18 detection methods, particularly in detecting morphologically complex nodules.The source code is publicly available at https://github.com/wjj1wjj/Ultrasound-DETR.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。