TSRNet: A Dual-stream Network for Refining 3D Tooth Segmentation

Hairong Jin,Yuefan Shen,Jianwen Lou,Kun Zhou,Youyi Zheng
DOI: https://doi.org/10.1109/tvcg.2024.3413345
IF: 5.2
2024-01-01
IEEE Transactions on Visualization and Computer Graphics
Abstract:The field of 3D tooth segmentation has made considerable advances thanks to deep learning, but challenges remain with coarse segmentation boundaries and prediction errors. In this paper, we introduce a novel learnable method to refine coarse results obtained from existing 3D tooth segmentation algorithms. The refinement framework features a dual-stream network called TSRNet (Tooth Segmentation Refinement Network) to rectify defective boundary and distance maps extracted from the coarse segmentation. The boundary map provides explicit boundary information, while the distance map provides gradient information in the form of the shortest geodesic distance between the vertex and the segmentation boundary. Following well-designed rules, the two refined maps are utilized to move the coarse tooth boundaries toward their correct positions through an iterative refinement process. The two-stage refinement method is validated on both 3D tooth and segmentation benchmark datasets. Extensive experiments demonstrate that our method significantly improves upon the coarse results from baseline methods and achieves state-of-the-art performance. (Code will be publicly available at https://github.com/bibi547/TSRNet.git.)
What problem does this paper attempt to address?