Yao Tianyuan, Qu Chang, Long Jun, Liu Quan, Deng Ruining, Tian Yuanhan, Xu Jiachen, Jha Aadarsh, Asad Zuhayr, Bao Shunxing, Zhao Mengyang, Fogo Agnes B, Landman Bennett A, Yang Haichun, Chang Catie, Huo Yuankai
Vanderbilt University, Department of Computer Science, Nashville, TN, USA 37215.
Central South University, Big Data Institute, Changsha, Hunan, China 410083.
J Mach Learn Biomed Imaging. 2022 Aug;1. Epub 2022 Sep 4.
With the rapid development of self-supervised learning (e.g., contrastive learning), the importance of having large-scale images (even without annotations) for training a more generalizable AI model has been widely recognized in medical image analysis. However, collecting large-scale task-specific unannotated data at scale can be challenging for individual labs. Existing online resources, such as digital books, publications, and search engines, provide a new resource for obtaining large-scale images. However, published images in healthcare (e.g., radiology and pathology) consist of a considerable amount of compound figures with subplots. In order to extract and separate compound figures into usable individual images for downstream learning, we propose a simple compound figure separation (SimCFS) framework without using the traditionally required detection bounding box annotations, with a new loss function and a hard case simulation. Our technical contribution is four-fold: (1) we introduce a simulation-based training framework that minimizes the need for resource extensive bounding box annotations; (2) we propose a new side loss that is optimized for compound figure separation; (3) we propose an intra-class image augmentation method to simulate hard cases; and (4) to the best of our knowledge, this is the first study that evaluates the efficacy of leveraging self-supervised learning with compound image separation. From the results, the proposed SimCFS achieved state-of-the-art performance on the ImageCLEF 2016 Compound Figure Separation Database. The pretrained self-supervised learning model using large-scale mined figures improved the accuracy of downstream image classification tasks with a contrastive learning algorithm. The source code of SimCFS is made publicly available at https://github.com/hrlblab/ImageSeperation.
随着自监督学习(例如对比学习)的快速发展,在医学图像分析中,拥有大规模图像(即使没有注释)对于训练更具通用性的人工智能模型的重要性已得到广泛认可。然而,对于单个实验室来说,大规模收集特定任务的无注释数据可能具有挑战性。现有的在线资源,如图书、出版物和搜索引擎,为获取大规模图像提供了新的资源。然而,医疗保健领域(如放射学和病理学)中发布的图像包含大量带有子图的复合图。为了将复合图提取并分离成可用于下游学习的单个可用图像,我们提出了一个简单的复合图分离(SimCFS)框架,该框架不使用传统所需的检测边界框注释,而是采用了一种新的损失函数和硬案例模拟。我们的技术贡献有四个方面:(1)我们引入了一个基于模拟的训练框架,最大限度地减少了对资源密集型边界框注释的需求;(2)我们提出了一种针对复合图分离进行优化的新的辅助损失;(3)我们提出了一种类内图像增强方法来模拟硬案例;(4)据我们所知,这是第一项评估利用自监督学习进行复合图像分离效果的研究。从结果来看,所提出的SimCFS在ImageCLEF 2016复合图分离数据库上取得了领先的性能。使用大规模挖掘图的预训练自监督学习模型通过对比学习算法提高了下游图像分类任务的准确性。SimCFS的源代码已在https://github.com/hrlblab/ImageSeperation上公开提供。