Prior-Aware Cross Pseudo Supervision for Semi-supervised Tooth Segmentation
摘要
Automatic and precise tooth segmentation is crucial in computer aided dentistry, serving a pivotal role in various applications likes diagnosis and treatment planning. While prominent methods can attain satisfactory segmentation results, disproportion in the proportion of annotated images have an impact on model performance. In this paper, we present a cascade semi-supervised method named C-CPS for tooth segmentation, which designed for imbalanced datasets. C-CPS is built upon the idea that the model can acquire more essential information about uncertainty. It consists of two subtly different decoders and utilizes two distinct strategies for generating pseudo-labels. Particularly, one strategy aims at minimizing entropy and enhancing posterior probabilities, while the other constrains predictions by incorporating prior information. To boost the model training, we integrate the two generation strategies into a cycle supervision module. We evaluate C-CPS in MICCAI 2023 Challenge and respectively achieve 76.70% of Dice, 81.46% of IOU, and 16.42% normalized HD.