Meta Ordinal Regression Forest for Medical Image Classification With Ordinal Labels  

作　　者：Yiming Lei Haiping Zhu Junping Zhang Hongming Shan 

机构地区：[1]Shanghai Key Laboratory of Intelligent Information Processing,the School of Computer Science,Fudan University,Shanghai 200433,China [2]with Huawei,Shanghai 200120,China [3]Institute of Science and Technology for Brain-Inspired Intelligence and MOE Frontiers Center for Brain Science,Fudan University,Shanghai 200433 [4]Shanghai Center for Brain Science and Brain-Inspired Technology,Shanghai 200031,China [5]IEEE

出　　处：《IEEE/CAA Journal of Automatica Sinica》2022年第7期1233-1247,共15页自动化学报（英文版）

基　　金：This work was supported in part by the Natural Science Foundation of Shanghai(21ZR1403600);the National Natural Science Foundation of China(62176059);Shanghai Municipal Science and Technology Major Project(2018SHZDZX01);Zhang Jiang Laboratory,Shanghai Sailing Program(21YF1402800);Shanghai Municipal of Science and Technology Project(20JC1419500);Shanghai Center for Brain Science and Brain-inspired Technology.

摘　　要：The performance of medical image classification has been enhanced by deep convolutional neural networks(CNNs),which are typically trained with cross-entropy(CE)loss.However,when the label presents an intrinsic ordinal property in nature,e.g.,the development from benign to malignant tumor,CE loss cannot take into account such ordinal information to allow for better generalization.To improve model generalization with ordinal information,we propose a novel meta ordinal regression forest(MORF)method for medical image classification with ordinal labels,which learns the ordinal relationship through the combination of convolutional neural network and differential forest in a meta-learning framework.The merits of the proposed MORF come from the following two components:A tree-wise weighting net(TWW-Net)and a grouped feature selection(GFS)module.First,the TWW-Net assigns each tree in the forest with a specific weight that is mapped from the classification loss of the corresponding tree.Hence,all the trees possess varying weights,which is helpful for alleviating the tree-wise prediction variance.Second,the GFS module enables a dynamic forest rather than a fixed one that was previously used,allowing for random feature perturbation.During training,we alternatively optimize the parameters of the CNN backbone and TWW-Net in the meta-learning framework through calculating the Hessian matrix.Experimental results on two medical image classification datasets with ordinal labels,i.e.,LIDC-IDRI and Breast Ultrasound datasets,demonstrate the superior performances of our MORF method over existing state-of-the-art methods.

关 键 词：Terms-Convolutional neural network(CNNs) medical image classification META-LEARNING ordinal regression random forest 

分 类 号：R445[医药卫生—影像医学与核医学]