Topology-Constrained Quantized nnUNet for Efficient and Anatomically Accurate 3D Tooth Segmentation
Not on the map. Why?
Abstract (arXiv)
We propose a topology-constrained quantized nnUNet framework for efficient and anatomically accurate 3D tooth segmentation, addressing the challenges of spatial distortion introduced by quantization in deep learning models. The proposed method integrates a novel tooth-specific topological loss into quantization-aware training, preserving critical anatomical structures such as tooth count, adjacency relationships, and cavity integrity while maintaining computational efficiency. The system employs an 8-bit quantized nnUNet backbone, where weights and activations are dynamically calibrated to minimize precision loss during inference. Furthermore, the topological loss combines connected-component analysis, adjacency consistency, and hole detection penalties, ensuring anatomical fidelity without modifying the underlying network architecture. The joint optimization objective harmonizes cross-entropy loss, quantization regularization, and topological constraints, enabling end-to-end training with gradient approximations for persistent homology terms. Experiments demonstrate that our approach significantly reduces topological errors compared to conventional quantized models, achieving clinically plausible segmentations on dental CBCT scans. The method retains the hardware efficiency of integer-only inference, making it suitable for deployment in resource-constrained clinical environments. This work bridges the gap between computational efficiency and anatomical precision in medical image segmentation, offering a practical solution for real-world dental applications.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 9 of its 15 references that are in the map. First posted on arXiv 5 May 2026.
Builds on
9 of 15 references.
References · 9
- 2017Quantization and Training of Neural Networks for Efficient Integer-Arithmetic-Only Inferencecited by 1,548
- 2021DiNTS: Differentiable Neural Network Topology Search for 3D Medical Image Segmentationcited by 47
- 2021A survey on shape-constraint deep learning for medical image segmentationcited by 15
- 20233DTeethSeg'22: 3D Teeth Scan Segmentation and Labeling Challengecited by 14
- 2023Topology-Aware Focal Loss for 3D Image Segmentationcited by 12
- 20253D Dental Model Segmentation with Geometrical Boundary Preservingcited by 7
- 2025PanoGAN A Deep Generative Model for Panoramic Dental Radiographscited by 3
- 2025Impact of Labeling Inaccuracy and Image Noise on Tooth Segmentation in Panoramic Radiographs using Federated, Centralized and Local Learningcited by 3
- 2025PanoDiff-SR: Synthesizing Dental Panoramic Radiographs using Diffusion and Super-resolutioncited by 2
Built on by
Not cited yet.