结合代价敏感半监督集成学习的糖尿病视网膜病变分级

Grading of diabetic retinopathy based on cost-sensitive semi-supervised ensemble learning

针对传统糖尿病视网膜病变(糖网)分级诊断系统中,由于数据集中缺少病灶区域的标记和类别分布的不平衡性导致无法有效地进行监督性分类的问题,提出基于代价敏感的半监督Bagging(CS-SemiBagging)的糖网分级方法。首先,从眼底图像上删除视网膜血管,并在此图像上检测疑似的红色病灶(微动脉瘤(MAs)与出血斑(HEMs));然后,从颜色、形状和纹理方面提取22维的特征用于描述每个病灶区域;其次,构建一个CS-SemiBagging模型对MAs与HEMs进行分类;最后,依据不同病灶的数量将糖网划分为4级。通过对国际公共数据集MESSIDOR进行糖网分级评估实验,所提方法获得平均准确率为90.2%,与经典的半监督学习的Co-training方法相比提高了4.9个百分点。实验结果表明,CS-SemiBagging方法在无需提供病灶标注的情况下,能够高效自动地对糖网进行分级,从而既能免除医学图像中标注病灶的费时费力,又可以避免样本类别分布不平衡对分类算法的性能影响,获得较好的效果。

Since the lack of lesion labels and unbalanced data distribution in datasets lead to the problem that the supervised classification model can not effectively classify the lesions in the traditional Diabetic Retinopathy （DR） grading system, a Cost-Sensitive based Semi-supervised Bagging （CS-SemiBagging） algorithm for DR classification was proposed. Firstly, retinal vessels were removed from a fundus image, and then the suspicious red lesions （MicroAneurysms （MAs） and HEMorrhages （HEMs）） were detected on the image without vessels. Secondly, a 22-dimensional feature based on color, shape and texture was extracted to describe each candidate lesion region. Thirdly, a CS-SemiBagging model was constructed for the classification of MAs and HEMs. Finally, the severity of DR was graded into four levels based on the numbers of different lesions. The proposed method was evaluated on the publicly available MESSIDOR database. It achieved an average accuracy of 90.2%, which was 4.9 percentage points higher than that of classical semi-supervised learning method based on Co-training. The CS-SemiBagging algorithm can effectively classify DR without label information of the suspicious lesions, so as to avoid the time-consuming effort of labeling the lesions by specialists and the bad influence of unbalanced samples on the classification.