English

Multi-task Cross-modal Learning for Chest X-ray Image Retrieval

Computer Vision and Pattern Recognition 2026-01-12 v1 Artificial Intelligence Information Retrieval

Abstract

CLIP and BiomedCLIP are examples of vision-language foundation models and offer strong cross-modal embeddings; however, they are not optimized for fine-grained medical retrieval tasks, such as retrieving clinically relevant radiology reports using chest X-ray (CXR) image queries. To address this shortcoming, we propose a multi-task learning framework to fine-tune BiomedCLIP and evaluate improvements to CXR image-text retrieval. Using BiomedCLIP as the backbone, we incorporate a lightweight MLP projector head trained with a multi-task composite loss function that includes: (1) a binary cross-entropy loss to distinguish normal from abnormal CXR studies, (2) a supervised contrastive loss to reinforce intra-class consistency, and (3) a CLIP loss to maintain cross-modal alignment. Experimental results demonstrate that the fine-tuned model achieves more balanced and clinically meaningful performance across both image-to-text and text-to-image retrieval tasks compared to the pretrained BiomedCLIP and general-purpose CLIP models. Furthermore, t-SNE visualizations reveal clearer semantic clustering of normal and abnormal cases, demonstrating the model's enhanced diagnostic sensitivity. These findings highlight the value of domain-adaptive, multi-task learning for advancing cross-modal retrieval in biomedical applications.

Keywords

Cite

@article{arxiv.2601.05399,
  title  = {Multi-task Cross-modal Learning for Chest X-ray Image Retrieval},
  author = {Zhaohui Liang and Sivaramakrishnan Rajaraman and Niccolo Marini and Zhiyun Xue and Sameer Antani},
  journal= {arXiv preprint arXiv:2601.05399},
  year   = {2026}
}
R2 v1 2026-07-01T08:57:07.962Z