针对牙科龋齿检测难题,提出基于结构先验与动态损失优化的Transformer模型。
Caries DETR: Tooth Structure-aware Prior and Lesion-aware Dynamic Loss Refinement for DETR Based Caries Detection

- 利用牙齿结构先验初始化查询,聚焦解剖关键区域
- 通过病变大小和质量自适应重加权损失,提升微小病灶检测能力
- 在两个公开数据集上达到当前最优,适合临床早期诊断应用
由于龋齿在口内影像中表现为细微、低对比度的病灶,现有深度学习模型在早期检测上面临巨大挑战。尽管近期基于Transformer的检测器在自然图像中表现优异,但往往无法捕捉对牙科龋齿检测至关重要的领域特定解剖先验。本文提出Caries-DETR,一种专用于口内影像龋齿检测的Transformer框架。设计了牙齿结构感知查询初始化(TSQI),结合大规模口内照片预训练与结构感知分支(SPB),融合高频结构先验,引导模型关注解剖学意义显著的病灶区域。此外,提出病变感知动态损失精炼(LDLR),通过基于病变大小、解剖相关性及预测质量的自适应损失重加权实现质量驱动的困难样本挖掘,优化对细微病灶的检测。在两个公开数据集(AlphaDent和DentalAI)上的大量实验表明,Caries-DETR相较现有方法取得最先进的性能,并展现出良好的泛化与鲁棒性。代码与数据见:https://github.com/XuefenLiu-SZU/Caries-DETR
原文摘要 · Abstract (English)
As dental caries appear as subtle, low-contrast lesions in intraoral imaging, existing deep learning models face significant challenges in the early detection of caries. While recent Transformer-based detectors have shown promising results in natural images, they often fail to capture the domain-specific anatomical priors crucial for dental caries detection. In this paper, we propose Caries-DETR, a specialized Transformer framework for caries detection in intraoral images. A Tooth Structure-aware Query Initialization (TSQI) is designed, leveraging large-scale intraoral photograph pre-training and a structure perception branch (SPB) to integrate high-frequency structural priors, guiding the model to focus on anatomically significant lesion areas. Furthermore, we design a Lesion-aware Dynamic Loss Refinement (LDLR) to implement quality-driven hard mining through adaptive loss reweighting based on lesion size, anatomical relevance, and prediction quality, optimizing detection for subtle lesions. Extensive experiments on two public datasets (i.e., AlphaDent and DentalAI) demonstrate that Caries-DETR achieves a state-of-the-art performance compared to existing methods and exhibits good generalization and robustness. Code and data at https://github.com/XuefenLiu-SZU/Caries-DETR}{https://github.com/XuefenLiu-SZU/Caries-DETR.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。