论文标题

基于深度学习的图像检索系统,用于胸部X光片及其在Covid-19

Deep Metric Learning-based Image Retrieval System for Chest Radiograph and its Clinical Applications in COVID-19

论文作者

Zhong, Aoxiao, Li, Xiang, Wu, Dufan, Ren, Hui, Kim, Kyungsang, Kim, Younggon, Buch, Varun, Neumark, Nir, Bizzo, Bernardo, Tak, Won Young, Park, Soo Young, Lee, Yu Rim, Kang, Min Kyu, Park, Jung Gil, Kim, Byung Seok, Chung, Woo Jin, Guo, Ning, Dayan, Ittai, Kalra, Mannudeep K., Li, Quanzheng

论文摘要

近年来,基于深度学习的图像分析方法已被广泛应用于计算机辅助检测,诊断和预后,并在新型冠状病毒疾病2019(COVID-19)大流行期间显示出其价值。胸部X光片(CXR)一直在COVID-19患者进行分类,诊断和监测中发挥关键作用,尤其是在美国。考虑到CXR中的混合和非特异性信号,提供相似图像和相关临床信息的CXR图像检索模型比直接图像诊断模型更有意义。在这项工作中,我们开发了一种基于深度度量学习的新型CXR图像检索模型。与传统的诊断模型不同,旨在学习从图像到标签的直接映射,而拟议的模型旨在学习图像的优化嵌入空间,其中将带有相同标签和相似内容的图像汇总在一起。它利用多相似性损失具有硬挖掘采样策略和注意机制来学习优化的嵌入空间,并为查询图像提供了相似的图像。该模型在从3个不同来源收集的国际多站点COVID-19数据集上进行了培训和验证。 COVID-19的实验结果图像检索和诊断任务表明,该模型可以作为COVID-19的CXR分析和患者管理的强大解决方案。还测试了该模型在不同的临床决策支持任务上的可传递性,其中将预训练的模型应用于从新数据集中提取图像特征,而无需进行任何进一步的培训。这些结果表明,我们基于公制的图像检索模型在CXR检索,诊断和预后非常有效,因此对于COVID-19患者的治疗和管理具有巨大的临床价值。

In recent years, deep learning-based image analysis methods have been widely applied in computer-aided detection, diagnosis and prognosis, and has shown its value during the public health crisis of the novel coronavirus disease 2019 (COVID-19) pandemic. Chest radiograph (CXR) has been playing a crucial role in COVID-19 patient triaging, diagnosing and monitoring, particularly in the United States. Considering the mixed and unspecific signals in CXR, an image retrieval model of CXR that provides both similar images and associated clinical information can be more clinically meaningful than a direct image diagnostic model. In this work we develop a novel CXR image retrieval model based on deep metric learning. Unlike traditional diagnostic models which aims at learning the direct mapping from images to labels, the proposed model aims at learning the optimized embedding space of images, where images with the same labels and similar contents are pulled together. It utilizes multi-similarity loss with hard-mining sampling strategy and attention mechanism to learn the optimized embedding space, and provides similar images to the query image. The model is trained and validated on an international multi-site COVID-19 dataset collected from 3 different sources. Experimental results of COVID-19 image retrieval and diagnosis tasks show that the proposed model can serve as a robust solution for CXR analysis and patient management for COVID-19. The model is also tested on its transferability on a different clinical decision support task, where the pre-trained model is applied to extract image features from a new dataset without any further training. These results demonstrate our deep metric learning based image retrieval model is highly efficient in the CXR retrieval, diagnosis and prognosis, and thus has great clinical value for the treatment and management of COVID-19 patients.

扫码加入交流群

加入微信交流群

微信交流群二维码

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