CrossMed: A Multimodal Cross-Task Benchmark for Compositional Generalization in Medical Imaging

TL;DR

CrossMed introduces a comprehensive benchmark to evaluate the ability of multimodal medical AI models to generalize compositionally across unseen combinations of imaging modalities, anatomical structures, and tasks using a unified VQA format.

Contribution

This work presents CrossMed, a novel benchmark reformulating multiple medical imaging datasets into a VQA format to assess compositional generalization in multimodal LLMs.

Findings

Models perform well on related splits but struggle with unrelated and zero-overlap splits.

Cross-task transfer improves segmentation performance by 7% cIoU.

Multimodal LLMs excel at compositional generalization compared to traditional models.

Abstract

Recent advances in multimodal large language models have enabled unified processing of visual and textual inputs, offering promising applications in general-purpose medical AI. However, their ability to generalize compositionally across unseen combinations of imaging modality, anatomy, and task type remains underexplored. We introduce CrossMed, a benchmark designed to evaluate compositional generalization (CG) in medical multimodal LLMs using a structured Modality-Anatomy-Task (MAT) schema. CrossMed reformulates four public datasets, CheXpert (X-ray classification), SIIM-ACR (X-ray segmentation), BraTS 2020 (MRI classification and segmentation), and MosMedData (CT classification) into a unified visual question answering (VQA) format, resulting in 20,200 multiple-choice QA instances. We evaluate two open-source multimodal LLMs, LLaVA-Vicuna-7B and Qwen2-VL-7B, on both Related and Unrelated MAT splits, as well as a zero-overlap setting where test triplets share no Modality, Anatomy, or Task with the training data. Models trained on Related splits achieve 83.2 percent classification accuracy and 0.75 segmentation cIoU, while performance drops significantly under Unrelated and zero-overlap conditions, demonstrating the benchmark difficulty. We also show cross-task transfer, where segmentation performance improves by 7 percent cIoU even when trained using classification-only data. Traditional models (ResNet-50 and U-Net) show modest gains, confirming the broad utility of the MAT framework, while multimodal LLMs uniquely excel at compositional generalization. CrossMed provides a rigorous testbed for evaluating zero-shot, cross-task, and modality-agnostic generalization in medical vision-language models.