Heterogeneous data fusion and loss function design for tooth point cloud segmentation

被引:12
作者
Liu, Dongsheng [1 ]
Tian, Yan [1 ,2 ]
Zhang, Yujie [1 ,2 ]
Gelernter, Judith [3 ]
Wang, Xun [1 ]
机构
[1] Zhejiang Gongshang Univ, Sch Comp Sci & Informat Engn, Hangzhou 310018, Peoples R China
[2] Shining3D Tech Co Ltd, Hangzhou 310018, Peoples R China
[3] Rutgers State Univ, Informat Sci Dept, New Brunswick, NJ 08901 USA
基金
中国国家自然科学基金;
关键词
Computer vision; Information fusion; Loss function; Point cloud; NETWORK;
D O I
10.1007/s00521-022-07379-y
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
Tooth point cloud segmentation plays an important role in the digital dentistry, and has received much attention in the past decade. Recently, methods based on the graph neural network have made significant progress. However, the development has been hindered by two challenges: (1) the heterogeneous geometry data are analyzed separately or combined linearly which leads to a semantic gap in different streams; (2) there is mis-alignment between the loss function and evaluation metrics in the segmentation task. In this paper, a novel interacted graph network is presented that combines cues from heterogeneous geometry data by extending the graph attention architecture to propagate information among the different graphs. Moreover, in this paper, an approach is designed to search the segmentation loss function based on the computation graphs according to the evaluation metrics, and the evolution algorithm is revised to avoid potential loss and equivalent loss functions. Our method and other methods use the Shining3D Tooth Segmentation dataset, with experimental results compared in terms of accuracy.
引用
收藏
页码:17371 / 17380
页数:10
相关论文
共 46 条
  • [1] [Anonymous], 2018, P INT C LEARN REPR V
  • [2] The Lovasz-Softmax loss: A tractable surrogate for the optimization of the intersection-over-union measure in neural networks
    Berman, Maxim
    Triki, Amal Rannen
    Blaschko, Matthew B.
    [J]. 2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2018, : 4413 - 4421
  • [3] Caliva F, 2019, ARXIV190803679
  • [4] Geometric attentional dynamic graph convolutional neural networks for point cloud analysis
    Cui, Yiming
    Liu, Xin
    Liu, Hongmin
    Zhang, Jiyong
    Zare, Alina
    Fan, Bin
    [J]. NEUROCOMPUTING, 2021, 432 : 300 - 310
  • [5] TSegNet: An efficient and accurate tooth segmentation network on 3D dental model
    Cui, Zhiming
    Li, Changjian
    Chen, Nenglun
    Wei, Guodong
    Chen, Runnan
    Zhou, Yuanfeng
    Shen, Dinggang
    Wang, Wenping
    [J]. MEDICAL IMAGE ANALYSIS, 2021, 69
  • [6] Diya Sun, 2020, Medical Image Computing and Computer Assisted Intervention - MICCAI 2020. 23rd International Conference. Proceedings. Lecture Notes in Computer Science (LNCS 12264), P703, DOI 10.1007/978-3-030-59719-1_68
  • [7] One-Shot Neural Architecture Search via Self-Evaluated Template Network
    Dong, Xuanyi
    Yang, Yi
    [J]. 2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2019), 2019, : 3680 - 3689
  • [8] Elsken T, 2019, INT C LEARNING REPRE, P551
  • [9] Elsken T, 2019, J MACH LEARN RES, V20
  • [10] Toward Clinically Applicable 3-Dimensional Tooth Segmentation via Deep Learning
    Hao, J.
    Liao, W.
    Zhang, Y. L.
    Peng, J.
    Zhao, Z.
    Chen, Z.
    Zhou, B. W.
    Feng, Y.
    Fang, B.
    Liu, Z. Z.
    Zhao, Z. H.
    [J]. JOURNAL OF DENTAL RESEARCH, 2022, 101 (03) : 304 - 311