arXiv:2512.03598cs.CV2025-12

用记忆原型辅助补全缺损牙科点云,提升重建精度与细节

Memory-Guided Point Cloud Completion for Dental Reconstruction

  • 引入可学习记忆库,检索近似牙形原型并融合到特征中
  • 在 Teeth3DS 基准上实现更优的 Chamfer Distance,细节更清晰
  • 无需位置标签,可无缝接入主流重建模型,适合临床应用

部分牙科点云常因遮挡和扫描视角有限导致大面积缺失,干扰编码器全局特征,迫使解码器虚构结构。本文提出一种检索增强型补全框架,在标准编码-解码结构中引入可学习原型记忆库。编码器将部分输入转化为全局描述符后,从记忆库中检索最接近的流形原型,并通过置信度加权融合至查询特征再进行解码。记忆库端到端优化并自组织为可复用的牙形原型,无需牙位标签,提供结构先验以稳定缺失区域推断,释放解码器容量用于细节恢复。该模块即插即用,兼容常见补全主干网络,且保持相同训练损失。在自构建的 Teeth3DS 基准上的实验表明,该方法在 Chamfer Distance 上持续提升,可视化结果展现出更锐利的牙尖、嵴线及邻接面过渡。

原文摘要 · Abstract (English)

Partial dental point clouds often suffer from large missing regions caused by occlusion and limited scanning views, which bias encoder-only global features and force decoders to hallucinate structures. We propose a retrieval-augmented framework for tooth completion that integrates a prototype memory into standard encoder--decoder pipelines. After encoding a partial input into a global descriptor, the model retrieves the nearest manifold prototype from a learnable memory and fuses it with the query feature through confidence-gated weighting before decoding. The memory is optimized end-to-end and self-organizes into reusable tooth-shape prototypes without requiring tooth-position labels, thereby providing structural priors that stabilize missing-region inference and free decoder capacity for detail recovery. The module is plug-and-play and compatible with common completion backbones, while keeping the same training losses. Experiments on a self-processed Teeth3DS benchmark demonstrate consistent improvements in Chamfer Distance, with visualizations showing sharper cusps, ridges, and interproximal transitions. Our approach provides a simple yet effective way to exploit cross-sample regularities for more accurate and faithful dental point-cloud completion.

点云补全牙科重建记忆机制生成模型

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