arXiv:2605.04201cs.CV2026-05

8-bit量化nnUNet+拓扑损失,让牙齿分割又快又准

Topology-Constrained Quantized nnUNet for Efficient and Anatomically Accurate 3D Tooth Segmentation

  • 在量化训练中加入牙齿特异性拓扑损失,保持解剖结构
  • 拓扑错误降低显著,8位量化下仍保持临床可用精度
  • 适合资源受限的牙科临床部署,无需修改网络结构

我们提出一种拓扑约束的量化nnUNet框架,用于高效且解剖准确的3D牙齿分割,解决深度学习模型量化带来的空间失真问题。该方法在量化感知训练中引入新型牙齿特异性拓扑损失,保留牙齿数量、邻接关系和髓腔完整性等关键解剖结构,同时保持计算效率。系统采用8位量化nnUNet骨干网络,权重和激活动态校准以最小化推理时的精度损失。拓扑损失结合连通域分析、邻接一致性与孔洞检测惩罚项,确保解剖保真度而不改变底层网络架构。联合优化目标融合交叉熵损失、量化正则化与拓扑约束,通过梯度近似实现持久同调项的端到端训练。实验表明,本方法相比传统量化模型显著减少拓扑错误,在牙科CBCT扫描上实现临床可接受的分割结果。方法保留整数仅推理的硬件效率,适用于资源受限的临床环境。本工作弥合了计算效率与解剖精确性之间的差距,为实际牙科应用提供可行解决方案。

原文摘要 · Abstract (English)

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.

3D分割量化牙齿分割拓扑约束

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。