基于委员会查询和自步多样性学习的医学图像分割  

Self-paced learning with diversity for medical image segmentationby using query-by-committee

作　　者：曹源 王妍[3] 王文强 贺小伟[1,2] CAO Yuan;WANG Yan;WANG Wenqiang;HE Xiaowei(School of Information Science and Technology,Northwest University,Xi′an 710127,China;Key Laboratory for Radiomics and Intelligent Sense of Xi′an,Xi′an 710127,China;School of Petroleum Engineering,Xi′an Shiyou University,Xi′an 710065,China)

机构地区：[1]西北大学信息科学与技术学院,陕西西安710127 [2]西安市影像组学与智能感知重点实验室,陕西西安710127 [3]西安石油大学石油工程学院,陕西西安710065

出　　处：《西北大学学报（自然科学版）》2020年第2期285-294,共10页Journal of Northwest University（Natural Science Edition）

基　　金：国家自然科学基金资助项目(61971350,61901374);陕西省自然科学基金资助项目(2017JQ4007)。

摘　　要：由于拥有像素级标记的医学图像数量非常少,制约了卷积神经网络在医学图像分割任务上的应用,因此,该文提出了一种基于委员会查询的自步多样性学习算法,在训练数据有限的情况下提升医学图像分割模型的性能。该文所提算法结合了基于委员会查询的数据选择方法,实现动态地从易到难选择样本,对模型进行训练。同时,该算法通过应用仿射传播聚类,保证了数据选择的多样性,提升了图像分割模型的性能。为了验证所提算法框架的有效性,分别在3类医学图像分割任务的5个不同数据集任务上进行了实验,实验结果表明,该文所提算法可以显著提升分割性能。在使用相同数据的训练的情况下,相比于全监督学习,使用该文算法可以得到更高的Dice评估指标、表面距离和平均交并比值。Because the number of pixel-wise labeled medical images are extremely small,which prevents the application of convolutional neural network(CNN)in medical image segmentation tasks,a query-by-committee based self-paced learning with diversity(SPLD)framework is proposed to boost the performance of medical image segmentation with limited data.The proposed SPLD algorithm combines the data selection method based the query-by-committee to realize the dynamic selection of samples from easy to difficult and train the model.Meanwhile,by applying the affine propagation clustering,the proposed algorithm guarantees the diversity of data and the performance of image segmentation model is further enhanced.To verify the effectiveness of the proposed QBC based SPLD framework,we conducted experiments on three medical image segmentation tasks with five different datasets.The experimental results show that the proposed algorithm can significantly improve the segmentation performance.With the same dataset,the proposed SPLD could significantly improve the segmentation performance and achieve a higher Dice score,surface distance and mean Intersection over Union(mIoU)than fully supervised learning.

关 键 词：自步多样性学习 图像分割 委员会查询 深度学习 

分 类 号：TP3-05[自动化与计算机技术—计算机科学与技术]