计算机应用 ›› 2019, Vol. 39 ›› Issue (12): 3456-3461.DOI: 10.11772/j.issn.1001-9081.2019049101

• 第十七届中国机器学习会议(CCML 2019)论文 • 上一篇    下一篇

基于精细化残差U-Net的新生儿局灶性脑白质损伤分割模型

刘亚龙, 李洁, 王颖, 仵赛飞, 邹佩   

  1. 西安电子科技大学 电子工程学院, 西安 710071
  • 收稿日期:2019-04-29 修回日期:2019-06-26 出版日期:2019-12-10 发布日期:2019-12-17
  • 作者简介:刘亚龙(1994-),男,江苏苏州人,硕士研究生,CCF会员,主要研究方向:深度学习、医学图像分割;李洁(1972-),女,陕西西安人,教授,博士,主要研究方向:图像处理、智慧交通;王颖(1981-),女,陕西西安人,副教授,博士,CCF会员,主要研究方向:影像处理与分析、模式识别;仵赛飞(1995-),男,河南省周口人,硕士研究生,主要研究方向:机器学习、计算机视觉;邹佩(1993-),女,湖北孝感人,硕士研究生,主要研究方向:影像处理与分析。
  • 基金资助:
    国家自然科学基金资助项目(61671339)。

Segmentation model of neonatal punctate white matter lesion based on refined deep residual U-Net

LIU Yalong, LI Jie, WANG Ying, WU Saifei, ZOU Pei   

  1. School of Electronic Engineering, Xidian University, Xi'an Shaanxi 710071, China
  • Received:2019-04-29 Revised:2019-06-26 Online:2019-12-10 Published:2019-12-17
  • Contact: 刘亚龙
  • Supported by:
    This work is partially supported by the National Natural Science Foundation of China (61671339).

摘要: 针对新生儿局灶性脑白质损伤的病灶区域小而样本差异大导致的检测与分割病灶较为困难的问题,提出一种精细化深度残差U-Net模型,以对病灶进行精细的语义分割。首先,把核磁共振(MRI)图像裁剪成较小的图像块;其次,利用残差U-Net提取出每个图像块不同层次的深度特征;然后,将特征进行融合并输出每个图像块的病灶分布概率图;最后,由全连接条件随机场对拼接后的概率图进行优化得到最终的分割结果。在某合作医院提供的数据集上的评估结果显示,在仅使用T1序列单模态数据的情况下,该模型在分割新生儿局灶性脑白质损伤时,病灶边缘的分割精度得到提高,且模型抗干扰能力较好。该模型的Dice相似性系数达到了62.51%,敏感度达到69.76%,特异性达到99.96%,修正的Hausdorff距离降低到33.67。

关键词: 局灶性脑白质损伤, 新生儿, 脑部肿瘤分割, 深度学习, 语义分割, 深度残差U-Net模型

Abstract: The tiny lesion area and the large difference between samples of neonatal punctate white matter lesion make it difficult to detect and segment the lesion. To solve the problem, a refined deep residual U-Net was proposed to realize the fine semantic segment of the lesion. Firstly, a Magnetic Resonance Imaging (MRI) image was cut into small patches. Secondly, the deep features of multiple layers of each image patch were extracted by the residual U-Net. Then, the features were fused and the probability map of the lesion distribution of each image patch was obtained. Finally, the probability map after splicing was optimized by the fully-connected condition random field to obtain the final segmentation results. The performance of the algorithm was evaluated on a dataset provided by a cooperative hospital. The results show that with only T1 order unimodal data used, the proposed model has the lesion's edge segmented more precisely, and the anti-interference ability of the model is prominent. The model has the Dice similarity coefficient of 62.51%, the sensitivity of 69.76%, the specificity of 99.96%, and the modified Hausdorff distance reduced to 33.67.

Key words: Punctate White Matter Lesion (PWML), neonate, brain tumor segmentation, deep learning, semantic segmentation, deep residual U-Net

中图分类号: