中文

利用自编码器结合标注与未标注数据实现三维医学图像分割——以CT图像中肝脏分割为例

图像与视频处理 2020-03-20 v1 计算机视觉与模式识别

摘要

利用卷积神经网络(CNN)对解剖结构进行自动分割构成了医学图像分析研究的很大一部分。多数基于 CNN 的方法依赖大量标注数据以妥善训练。标注医学数据往往稀缺,而未标注数据更易获得。这需要超越传统监督学习、并利用未标注数据完成分割任务的方法。本研究考察了自编码器提取特征以改进 CNN 分割的潜力。考虑了两种策略。其一为迁移学习,将预训练的自编码器特征用作分割网络中卷积层的初始化。其二为多任务学习,通过输入重建同时进行并优化分割与特征提取任务。使用卷积自编码器从未标注数据提取特征,并使用多尺度全卷积 CNN 执行 CT 图像中三维肝脏分割这一目标任务。对两种策略,均使用不同数量的标注与未标注训练数据进行了实验。与从头训练相比,所提学习策略在 75%75\% 的实验中改善了结果,并在未标注与标注训练数据比约为 32:132 : 112.5:112.5 : 1 时分别将 dice 系数提高至多 0.0400.0400.0240.024。结果表明,两种训练策略在未标注与标注训练数据占比较大时更为有效。

关键词

引用

@article{arxiv.2003.07923,
  title  = {3D medical image segmentation with labeled and unlabeled data using autoencoders at the example of liver segmentation in CT images},
  author = {Cheryl Sital and Tom Brosch and Dominique Tio and Alexander Raaijmakers and Jürgen Weese},
  journal= {arXiv preprint arXiv:2003.07923},
  year   = {2020}
}