Finding Dense Supervoxel Correspondence of Cone-Beam Computed Tomography Images

• Published in: Machine Learning in Medical Imaging pp. 114 - 122
• Main Authors: Pei, Yuru, Yi, Yunai, Ma, Gengyu, Guo, Yuke, Chen, Gui, Xu, Tianmin, Zha, Hongbin
• Format: Book Chapter
• Language: English
• Published: Cham Springer International Publishing 07.09.2017
• Series: Lecture Notes in Computer Science
• ISBN: 9783319673882; 3319673882
• ISSN: 0302-9743; 1611-3349
• DOI: 10.1007/978-3-319-67389-9_14

Dense correspondence establishment of cone-beam computed tomography (CBCT) images is a crucial step for attribute transfers and morphological variation assessments in clinical orthodontics. However, the registration by the traditional large-scale nonlinear optimization is time-consuming for the craniofacial CBCT images. The supervised random forest is known for its fast online performance, thought the limited training data impair the generalization capacity. In this paper, we propose an unsupervised random-forest-based approach for the supervoxel-wise correspondence of CBCT images. In particular, we present a theoretical complexity analysis with a data-dependent learning guarantee for the clustering hypotheses of the unsupervised random forest. A novel tree-pruning algorithm is proposed to refine the forest by removing the local trivial and inconsistent leaf nodes, where the learning bound serves as guidance for an optimal selection of tree structures. The proposed method has been tested on the label propagation of clinically-captured CBCT images. Experiments demonstrate the proposed method yields performance improvements over variants of both supervised and unsupervised random-forest-based methods.