AUTHOR=Gu Lin , Zhang Xiaowei , You Shaodi , Zhao Shen , Liu Zhenzhong , Harada Tatsuya 

TITLE=Semi-Supervised Learning in Medical Images Through Graph-Embedded Random Forest

JOURNAL=Frontiers in Neuroinformatics

VOLUME=Volume 14 - 2020

YEAR=2020

URL=https://www.frontiersin.org/journals/neuroinformatics/articles/10.3389/fninf.2020.601829

DOI=10.3389/fninf.2020.601829

ISSN=1662-5196

ABSTRACT=In this paper, we propose a novel semi-supervised random forest to tackle the challenging problem of lacking annotation in medical imaging analysis. Observing that the bottleneck of standard random forest is the biased information gain estimation, we replace it with a novel graph-embedded entropy which incorporates information from both labelled and unlabelled data. Empirical results show that our information gain is more reliable than the one used in the traditional random forest under insufficient labelled data. By slightly modifying the training process of standard random forest, our algorithm significantly improves the performance while preserving the virtue of random forest. Our method has shown superior performance with very limited data in both brain imaging analysis and machine learning benchmark.