论文标题

用于医疗图像分割的自我监督RCNN具有有限的数据注释

Self-Supervised-RCNN for Medical Image Segmentation with Limited Data Annotation

论文作者

Felfeliyan, Banafshe, Hareendranathan, Abhilash, Kuntze, Gregor, Cornell, David, Forkert, Nils D., Jaremko, Jacob L., Ronsky, Janet L.

论文摘要

许多用于医学图像分析的成功方法基于机器学习使用监督学习方法,这些方法通常需要专家注释的大型数据集以实现高精度。但是,医学数据注释是耗时且昂贵的,尤其是对于细分任务。为了解决有限标记的医学图像数据的学习问题,在这项工作中提出了一种基于自我监督的预处理预处理的替代深度学习培训策略。我们的训练方法首先,随机将不同的扭曲应用于未标记图像的随机区域,然后预测扭曲的类型和信息丢失。为此,已改进了Mask-RCNN体系结构的改进版本,以定位失真位置并恢复原始图像像素。根据骨关节炎倡议数据集评估了在不同的培训和微调方案中提出的分割任务的有效性。与从头开始的训练相比,使用这种自我监管的预处理方法将骰子得分提高了20%。拟议的自我监督学习简单,有效,适用于不同的医学图像分析任务,包括异常检测,分割和分类。

Many successful methods developed for medical image analysis that are based on machine learning use supervised learning approaches, which often require large datasets annotated by experts to achieve high accuracy. However, medical data annotation is time-consuming and expensive, especially for segmentation tasks. To solve the problem of learning with limited labeled medical image data, an alternative deep learning training strategy based on self-supervised pretraining on unlabeled MRI scans is proposed in this work. Our pretraining approach first, randomly applies different distortions to random areas of unlabeled images and then predicts the type of distortions and loss of information. To this aim, an improved version of Mask-RCNN architecture has been adapted to localize the distortion location and recover the original image pixels. The effectiveness of the proposed method for segmentation tasks in different pre-training and fine-tuning scenarios is evaluated based on the Osteoarthritis Initiative dataset. Using this self-supervised pretraining method improved the Dice score by 20% compared to training from scratch. The proposed self-supervised learning is simple, effective, and suitable for different ranges of medical image analysis tasks including anomaly detection, segmentation, and classification.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源