Efficient CBCT Segmentation via nnU-Net with Structure-Aware Post-processing and Interactive Refinement
摘要
Accurate segmentation of anatomical structures from cone-beam computed tomography (CBCT) is essential for clinical applications in dentistry, maxillofacial surgery, and orthodontics. The ToothFairy3 Challenge has a comprehensive 77-class segmentation task, emphasizing both accuracy and computational efficiency. In this work, we present a method based on the nnU-Net framework, enhanced with a Structure Aware Post-processing (SAP) strategy. nnU-Net serves as a backbone for multi-class segmentation, while SAP refines predictions by introducing individualized thresholds for each anatomical structure, thereby mitigating noise and preserving clinically important fine structures. To further improve efficiency, we disabled mirroring augmentation during training and employed inference acceleration strategies, including the removal of test-time augmentation and optimized interpolation on floating-point tensors. Experimental results validate the effectiveness of our approach in balancing segmentation accuracy with computational efficiency. To further ensure robustness in challenging clinical scenarios, we also utilize an interactive refinement module based on nnInteractive. This strategy allows clinicians to correct local segmentation errors with minimal user guidance, providing a safety net for complex anatomical variations.