中文

Caries-DETR:面向牙齿结构感知的先验与牙龈感知式动态损失优化的基于 DETR 的龋齿检测

计算机视觉与模式识别 2026-04-28 v1

摘要

牙齿龋 disease appears 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

关键词

引用

@article{arxiv.2604.23718,
  title  = {Caries DETR: Tooth Structure-aware Prior and Lesion-aware Dynamic Loss Refinement for DETR Based Caries Detection},
  author = {Xuefen Liu and Xinquan Yang and Mianjie Zheng and Kun Tang and Xuguang Li and Xiaoqi Guo and Linlin Shen and He Meng},
  journal= {arXiv preprint arXiv:2604.23718},
  year   = {2026}
}