计算机应用 ›› 2017, Vol. 37 ›› Issue (8): 2253-2257.DOI: 10.11772/j.issn.1001-9081.2017.08.2253

• 人工智能 • 上一篇    下一篇

基于多分类AdaBoost改进算法的TEE标准切面分类

王莉莉1,2, 付忠良1,2, 陶攀1,2, 朱锴1,2   

  1. 1. 中国科学院 成都计算机应用研究所, 成都 610041;
    2. 中国科学院大学, 北京 100049
  • 收稿日期:2017-03-01 修回日期:2017-04-12 出版日期:2017-08-10 发布日期:2017-08-12
  • 通讯作者: 王莉莉
  • 作者简介:王莉莉(1987-),女,河南周口人,博士研究生,主要研究方向:机器学习、模式识别、数据挖掘;付忠良(1967-),男,重庆合川人,教授,硕士,主要研究方向:机器学习、模式识别;陶攀(1988-),男,河南安阳人,博士研究生,主要研究方向:机器学习、数据挖掘;朱锴(1991-),男,贵州安顺人,博士研究生,主要研究方向:机器学习、数据挖掘。
  • 基金资助:
    四川省科技支撑计划项目(2016JZ0035);中国科学院西部之光项目。

TEE standard plane classification based on improved multi-class AdaBoost algorithm

WANG Lili1,2, FU Zhongliang1,2, TAO Pan1,2, ZHU Kai1,2   

  1. 1. Chengdu Institute of Computer Application, Chinese Academy of Sciences, Chengdu Sichuan 610041, China;
    2. University of Chinese Academy of Sciences, Beijing 100049, China
  • Received:2017-03-01 Revised:2017-04-12 Online:2017-08-10 Published:2017-08-12
  • Supported by:
    This work is partially supported by the Sichuan Science and Technology Support Project (2016JZ0035),the West Light Project of the Chinese Academy of Sciences.

摘要: 针对超声图像样本冗余、不同标准切面因疾病导致的高度相似性、感兴趣区域定位不准确问题,提出一种结合特征袋(BOF)特征、主动学习方法和多分类AdaBoost改进算法的经食管超声心动图(TEE)标准切面分类方法。首先采用BOF方法对超声图像进行描述;然后采用主动学习方法选择对分类器最有价值的样本作为训练集;最后,在AdaBoost算法对弱分类器的迭代训练中,根据临时强分类器的分类情况调整样本更新规则,实现对多分类AdaBoost算法的改进和TEE标准切面的分类。在TEE数据集和三个UCI数据集上的实验表明,相比AdaBoost.SAMME算法、多分类支持向量机(SVM)算法、BP神经网络和AdaBoost.M2算法,所提算法在各个数据集上的G-mean指标、整体分类准确率和大多数类别分类准确率都有不同程度的提升,且比较难分的类别分类准确率提升最为显著。实验结果表明,在包含类间相似样本的数据集上,分类器的性能有显著提升。

关键词: 多分类AdaBoost, 主动学习, 特征袋模型, 标准切面分类, 超声图像分类

Abstract: Due to redundancy of ultrasound image samples, high similarity between different planes caused by disease, and inaccurate positioning of region-of-interest, a classification method of TransEsophageal Echocardiography (TEE) standard plane was proposed by combining with Bag of Features (BOF) model, active learning and improved multi-class AdaBoost algorithm. Firstly, BOF model was constructed to describe ultrasound image. Secondly, active learning was adopted to select the most informative samples for classifiers as training data set. Lastly, improved multi-class AdaBoost algorithm was proposed, where the weight update rule of multi-class AdaBoost was modified according to the classfication results of temporary strong learner, and the TEE standard plane was classified by the improved multi-class AdaBoost algorithm. The experimental results on TEE data set and three UCI data sets showed that, compared with AdaBoost.SAMME, multi-class Support Vector Machine (SVM), BP neural network and AdaBoost.M2, the G-mean value, the total classification accuracy and the classification accuracy in most classes of the proposed method were improved in varying degrees, the classification accuracy of easily misclassified class was improved most significantly. The experimental results illustrate that the improved multi-class AdaBoost algorithm can significantly improve the G-mean value and accuracy of easily misclassified class in the datasets containing similar samples between classes.

Key words: multi-class AdaBoost, active learning, Bag of Features (BOF) model, standardized plane classification, ultrasound image classification

中图分类号: