Segmentation Network of Dental Mesh Model Based on Attention Mechanism and Neighbor Enhancement

To address the issue of existing deep learning-based segmentation methods for dental models, which overly rely on coordinate features and overlook the differences and complementary nature of geometric and coordinate features, a segmentation network based on attention mechanism and neighbor enhancement is proposed. First, an attention-enhanced module is employed to enhance the neighborhood of input features through different encoding schemes. It designs a differential attention mechanism and a self-attention mechanism to extract comprehensive contextual features of the mesh by considering the properties of geometric and coordinate features separately. Next, a complementary pooling function is constructed to fuse maximum pooling, attention pooling, and average pooling, which promotes the complementary interaction among different features and enhances the robustness of the segmentation network. Finally, an adaptive attention module is embedded to balance the numerical differences among multi-scale features for effective mesh segmentation. Compared with other methods in 200 real patients dental model dataset, the segmenta tion accuracy of dental model is 96.18%. The experimental results show that the proposed method is superior to the compared deep learning networks and has better performance in the segmentation of dental models.