返回
Geo-Net: Geometry-Guided Pretraining for Tooth Point Cloud Segmentation
DOI:10.1177/00220345241292566.png)
摘要
En 中文
Accurately delineating individual teeth in 3-dimensional tooth point clouds is an important orthodontic application. Learning-based segmentation methods rely on labeled datasets, which are typically limited in scale due to the labor-intensive process of annotating each tooth. In this article, we propose a self-supervised pretraining framework, named Geo-Net, to boost segmentation performance by leveraging large-scale unlabeled data. The framework is based on the scalable masked autoencoders, and 2 geometry-guided designs, curvature-aware patching algorithm (CPA) and scale-aware reconstruction (SCR), are proposed to enhance the masked pretraining for tooth point cloud segmentation. In particular, CPA is designed to assemble informative patches as the reconstruction unit, guided by the estimated pointwise curvatures. Aimed at equipping the pretrained encoder with scale-aware modeling capacity, we also propose SCR to perform multiple reconstructions across shallow and deep layers. In vitro experiments reveal that after pretraining with large-scale unlabeled data, the proposed Geo-Net can outperform the supervised counterparts in mean Intersection of Union (mIoU) with the same amount of annotated labeled data. The code and data are available at https://github.com/yifliu3/Geo-Net.
Keyword:
artificial intelligence
deep learning/machine learning
computer vision/convolutional neural networks
dental informatics/bioinformatics
electronic dental records
dental anatomy
期刊
IF:
5.9
论文数:
6.0K
被引数:
2.5W
机构
引用论文
GRAB-Net: Graph-Based Boundary-Aware Network for Medical Point Cloud SegmentationGRAB-Net: 基于图的边界感知网络用于医学点云分割
TSegNet: An efficient and accurate tooth segmentation network on 3D dental modelTSegNet: 一种高效准确的三维牙模型牙齿分割网络
MEDICAL IMAGE ANALYSIS
IF11.8

