Contrastive Learning with Feature Space Interpolation for Retrieval-Based Chest X-Ray Report Generation

Read full text
Authors: Zahid Ur Rahman, Gwanghyun Yu, Jin Lee, J. Kim
Cited by: 1
References: 27

Abstract

Automated radiology report generation from chest X-rays presents a critical challenge in medical imaging. Traditional image-captioning models struggle with clinical specificity and rare pathologies. Recently, contrastive vision language learning has emerged as a robust alternative that learns joint visual–textual representations. However, applying contrastive learning (CL) to radiology remains challenging due to severe data scarcity. Prior work has employed input space augmentation, but these approaches incur computational overhead and risk distorting diagnostic features. This work presents CL with feature space interpolation for retrieval (CLFIR), a novel CL framework operating on learned embeddings. The method generates interpolated pairs in the feature embedding space by mixing original and shuffled embeddings in batches using a mixing coefficient λ∼U(0.85,0.99). This approach increases batch diversity via synthetic samples, addressing the limitations of CL on medical data while preserving diagnostic integrity. Extensive experiments demonstrate state-of-the-art performance across critical clinical validation tasks. For report generation, CLFIR achieves BLEU-1/ROUGE/METEOR scores of 0.51/0.40/0.26 (Indiana university [IU] X-ray) and 0.45/0.34/0.22 (MIMIC-CXR). Moreover, CLFIR excels at image-to-text retrieval with R@1 scores of 4.14% (IU X-ray) and 24.3% (MIMIC-CXR) and achieves 0.65 accuracy in zero-shot classification on the CheXpert5×200 dataset, surpassing the established vision-language models.

Fields of Study & Subjects

Multimodal Machine Learning Applications COVID-19 diagnosis using AI Domain Adaptation and Few-Shot Learning

Publication details

Journal / source
Applied Sciences
Publisher
MDPI AG
Volume
16
Issue
1
Pages
470–470
Publication date
1 January 2026
Language
EN