arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2212.07050cs.LGcs.CVeess.IV

通过微调预训练图像-文本编码器显著提升零样本X光病理分类

Significantly improving zero-shot X-ray pathology classification via fine-tuning pre-trained image-text encoders

LG AI Research(LG人工智能研究院);KAIST(韩国科学技术院)

Jongseong Jang, Daeun Kyung, Seung Hwan Kim, Honglak Lee, Kyunghoon Bae, Edward Choi

AI总结:

本文提出包含正对损失松弛和随机句子采样的新微调策略,解决了医学图像-报告对的多标签性质被忽略的问题,无需外部知识即可在多个数据集上显著提升零样本X光病理分类性能,并超越放射科医生。

摘要:

深度神经网络正日益广泛地应用于病理分类等医学影像任务,但由于缺乏高质量、专家标注的训练数据而面临挑战。近期研究利用了像CLIP这样的预训练对比图像-文本模型,通过使用胸部X光图像及对应报告微调模型来进行零样本病理分类,从而消除了对特定病理注释的需求,使其适应医学用途。然而,大多数研究仍继续使用与通用领域相同的对比学习目标,忽略了医学图像-报告对的多标签性质。在本文中,我们提出了一种新的微调策略,包括正对损失松弛和随机句子采样。我们旨在不依赖外部知识的情况下提升零样本病理分类的性能。我们的方法可应用于任何预训练对比图像-文本编码器,并且由于不使用外部数据,可以无需进一步训练轻松迁移到域外数据集。我们的方法在四个胸部X光数据集和三个预训练模型上持续提升了整体零样本病理分类性能,平均宏观AUROC提升了4.3%。此外,我们的方法优于现有最先进技术,并在CheXpert数据集中五种竞赛病理的零样本分类上略微超过了获得委员会认证的放射科医生。

英文摘要:

Deep neural networks are increasingly used in medical imaging for tasks such as pathological classification, but they face challenges due to the scarcity of high-quality, expert-labeled training data. Recent efforts have utilized pre-trained contrastive image-text models like CLIP, adapting them for medical use by fine-tuning the model with chest X-ray images and corresponding reports for zero-shot pathology classification, thus eliminating the need for pathology-specific annotations. However, most studies continue to use the same contrastive learning objectives as in the general domain, overlooking the multi-labeled nature of medical image-report pairs. In this paper, we propose a new fine-tuning strategy that includes positive-pair loss relaxation and random sentence sampling. We aim to improve the performance of zero-shot pathology classification without relying on external knowledge. Our method can be applied to any pre-trained contrastive image-text encoder and easily transferred to out-of-domain datasets without further training, as it does not use external data. Our approach consistently improves overall zero-shot pathology classification across four chest X-ray datasets and three pre-trained models, with an average macro AUROC increase of 4.3%. Additionally, our method outperforms the state-of-the-art and marginally surpasses board-certified radiologists in zero-shot classification for the five competition pathologies in the CheXpert dataset.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑