[Automatic micro-CT pulp cavity image segmentation based on few-shot transfer learning].
Authors
Affiliations (1)
Affiliations (1)
- State Key Laboratory of Oral & Maxillofacial Reconstruction and Regeneration, Key Laboratory of Oral Biomedicine Ministry of Education, Hubei Key Laboratory of Stomatology, School & Hospital of Stomatology, Wuhan University, Wuhan 430079, China.
Abstract
<b>Objective:</b> To develop a micro-CT pulp cavity image segmentation model based on few-shot transfer learning, enabling efficient and accurate segmentation of the pulp cavity with limited training samples, thereby supporting three-dimensional(3D) anatomical research of the pulp cavity and digital root canal treatment. <b>Methods:</b> Extracted teeth (<i>n</i>=110) due to pathological reasons were collected from the Department of Oral and Maxillofacial Surgery, Hospital of Stomatology, Wuhan University, between January 2025 and September 2025. These teeth were scanned using micro-CT. The acquired images were randomly divided (simple random sampling) into a training set (10 teeth), a test set (90 teeth), and an independent test set (10 teeth) at a ratio of 1∶9∶1. 3D pulp cavity images were meticulously annotated by two oral clinicians (attending physicians) to establish the ground truth for segmentation. By introducing a cross-domain adaptation strategy, the natural image segmentation foundation model, segment anything model (SAM), was transferred to the task of pulp cavity image segmentation under the guidance of a limited number of training samples, thereby constructing an automatic micro-CT pulp cavity image segmentation model (PulpSAM). Multi-dimensional quantitative and qualitative analyses were performed to compare the segmentation performance on the pulp cavity (entire pulp cavity, apical 3 mm, and lateral accessory canals) achieved by deep learning models obtained with different numbers of training samples (0, 1, 3, and 10 teeth), i.e., SAM, PulpSAM-1, PulpSAM-3, and PulpSAM-10. These results were also compared with those of the U-Net and nnU-Net models. Additionally, the time required for three methods-manual annotation, fully automated model segmentation, and model segmentation followed by manual refinement-was compared. <b>Results:</b> With 1, 3, and 10 training teeth, the PulpSAM model achieved high segmentation performance, with median segmentation accuracy (Intersection over Union, Dice coefficient, precision, and accuracy) all≥92.3%, median 95% Hausdorff distance≤0.04 mm, and median average symmetric surface distance (ASSD) of 0.02 mm. These metrics were significantly superior to those of the SAM model (<i>P</i><0.008 3). The performance of PulpSAM-1 was significantly lower than that of PulpSAM-3 (<i>P</i><0.008 3). Compared with PulpSAM-10, PulpSAM-3 showed significantly higher recall and ASSD (both P<0.008 3), and significantly lower precision and relative volume difference (both <i>P</i><0.008 3). No statistically significant differences were observed between PulpSAM-3 and PulpSAM-10 in Intersection over Union, Dice coefficient, accuracy, or 95% Hausdorff distance (all <i>P</i>>0.008 3). With the exception of volume, the other eight segmentation accuracy metrics of the PulpSAM model were significantly superior to those of the corresponding U-Net and nnU-Net models (<i>P</i><0.05). The segmentation times required for manual annotation, PulpSAM-3 automatic segmentation, and PulpSAM-3 segmentation followed by manual refinement were 3 354.6 (852) s, 190 (27.9) s, and 646.4 (171.5) s, respectively, with statistically significant differences among the three methods (<i>P</i><0.001). <b>Conclusions:</b> This study developed the PulpSAM model based on a few-shot transfer learning strategy, achieving efficient and accurate automatic segmentation of the pulp cavity in micro-CT images.