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

被引:15
作者
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]   The Lovasz-Softmax loss: A tractable surrogate for the optimization of the intersection-over-union measure in neural networks [J].
Berman, Maxim ;
Triki, Amal Rannen ;
Blaschko, Matthew B. .
2018 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2018, :4413-4421
[2]  
Caliva Francesco, 2019, P INT C MED IM DEEP
[3]   Geometric attentional dynamic graph convolutional neural networks for point cloud analysis [J].
Cui, Yiming ;
Liu, Xin ;
Liu, Hongmin ;
Zhang, Jiyong ;
Zare, Alina ;
Fan, Bin .
NEUROCOMPUTING, 2021, 432 :300-310
[4]   TSegNet: An efficient and accurate tooth segmentation network on 3D dental model [J].
Cui, Zhiming ;
Li, Changjian ;
Chen, Nenglun ;
Wei, Guodong ;
Chen, Runnan ;
Zhou, Yuanfeng ;
Shen, Dinggang ;
Wang, Wenping .
MEDICAL IMAGE ANALYSIS, 2021, 69
[5]  
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
[6]   One-Shot Neural Architecture Search via Self-Evaluated Template Network [J].
Dong, Xuanyi ;
Yang, Yi .
2019 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2019), 2019, :3680-3689
[7]  
Elsken T, 2019, INT C LEARNING REPRE, P551
[8]  
Elsken T, 2019, J MACH LEARN RES, V20
[9]   Toward Clinically Applicable 3-Dimensional Tooth Segmentation via Deep Learning [J].
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. .
JOURNAL OF DENTAL RESEARCH, 2022, 101 (03) :304-311
[10]   AutoML: A survey of the state-of-the-art [J].
He, Xin ;
Zhao, Kaiyong ;
Chu, Xiaowen .
KNOWLEDGE-BASED SYSTEMS, 2021, 212