Zhou Zongwei, Shin Jae, Zhang Lei, Gurudu Suryakanth, Gotway Michael, Liang Jianming
Arizona State University.
Mayo Clinic.
Proc IEEE Comput Soc Conf Comput Vis Pattern Recognit. 2017 Jul;2017:4761-4772. doi: 10.1109/CVPR.2017.506. Epub 2017 Nov 9.
Intense interest in applying convolutional neural networks (CNNs) in biomedical image analysis is wide spread, but its success is impeded by the lack of large annotated datasets in biomedical imaging. Annotating biomedical images is not only tedious and time consuming, but also demanding of costly, specialty-oriented knowledge and skills, which are not easily accessible. To dramatically reduce annotation cost, this paper presents a novel method called AIFT (active, incremental fine-tuning) to naturally integrate active learning and transfer learning into a single framework. AIFT starts directly with a pre-trained CNN to seek "worthy" samples from the unannotated for annotation, and the (fine-tuned) CNN is further fine-tuned continuously by incorporating newly annotated samples in each iteration to enhance the CNN's performance incrementally. We have evaluated our method in three different biomedical imaging applications, demonstrating that the cost of annotation can be cut by at least half. This performance is attributed to the several advantages derived from the advanced active and incremental capability of our AIFT method.
在生物医学图像分析中应用卷积神经网络(CNN)的浓厚兴趣广泛存在,但生物医学成像中缺乏大型标注数据集阻碍了其成功应用。标注生物医学图像不仅乏味且耗时,还需要昂贵的、面向专业领域的知识和技能,而这些并不容易获得。为了大幅降低标注成本,本文提出了一种名为AIFT(主动、增量微调)的新方法,将主动学习和迁移学习自然地集成到一个框架中。AIFT直接从预训练的CNN开始,从未标注样本中寻找“有价值”的样本进行标注,并且在每次迭代中通过纳入新标注的样本对(微调后的)CNN进行进一步连续微调,以逐步提高CNN的性能。我们在三种不同的生物医学成像应用中评估了我们的方法,结果表明标注成本可至少降低一半。这种性能归因于我们的AIFT方法先进的主动和增量能力所带来的几个优势。