SSAD: Self-supervised Auxiliary Detection Framework for Panoramic X-ray based Dental Disease Diagnosis

Zijian Cai,Xinquan Yang,Xuguang Li,Xiaoling Luo,Xuechen Li,Linlin Shen,He Meng,Yongqiang Deng
2024-06-20
Abstract:Panoramic X-ray is a simple and effective tool for diagnosing dental diseases in clinical practice. When deep learning models are developed to assist dentist in interpreting panoramic X-rays, most of their performance suffers from the limited annotated data, which requires dentist's expertise and a lot of time cost. Although self-supervised learning (SSL) has been proposed to address this challenge, the two-stage process of pretraining and fine-tuning requires even more training time and computational resources. In this paper, we present a self-supervised auxiliary detection (SSAD) framework, which is plug-and-play and compatible with any detectors. It consists of a reconstruction branch and a detection branch. Both branches are trained simultaneously, sharing the same encoder, without the need for finetuning. The reconstruction branch learns to restore the tooth texture of healthy or diseased teeth, while the detection branch utilizes these learned features for diagnosis. To enhance the encoder's ability to capture fine-grained features, we incorporate the image encoder of SAM to construct a texture consistency (TC) loss, which extracts image embedding from the input and output of reconstruction branch, and then enforces both embedding into the same feature space. Extensive experiments on the public DENTEX dataset through three detection tasks demonstrate that the proposed SSAD framework achieves state-of-the-art performance compared to mainstream object detection methods and SSL methods. The code is available at <a class="link-external link-https" href="https://github.com/Dylonsword/SSAD" rel="external noopener nofollow">this https URL</a>
Computer Vision and Pattern Recognition
What problem does this paper attempt to address?
The paper attempts to address the issue of limited annotated data constraining the performance of deep learning models in the automatic detection of dental diseases using panoramic X-ray images in dental diagnosis. Specifically, the paper proposes a Self-Supervised Assisted Detection (SSAD) framework, which combines an image reconstruction branch and a disease detection branch, aiming to improve the accuracy of dental disease detection by learning dental texture features. Additionally, the paper designs a Texture Consistency Loss to enhance the encoder's ability to capture fine-grained features, thereby further improving detection performance. Experimental results show that this framework outperforms existing mainstream object detection methods and self-supervised learning methods on the public DENTEX dataset.