Localization Lens for Improving Medical Vision-Language Models
Organizations: Lahore University of Management Sciences, Lahore, Pakistan · The University of Western Australia, Perth, Australia · Information Technology University of the Punjab, Lahore, Pakistan
Abstract
Medical Vision-Language Models (Med-VLMs) have demonstrated strong capabilities in clinical tasks. However, they often struggle to understand anatomical structures and spatial positioning, which are crucial for medical reasoning. To address this, we propose a localization-aware enhancement to the Med-VLM pipeline, introducing improvements at three levels: data,architecture, and alignment. First, we introduce localization lens, a set of expert-validated representations that provide richer anatomical and positional context. However, as these representations increase input complexity, we integrate pixel shuffle within the model architecture to filter and refine representations, enhancing spatial information processing while preserving anatomical continuity. Lastly, to effectively align the localization lens representations with textual features, we incorporate decoupled contrastive loss (DCL) alongside the standard loss function. This ensures better feature discrimination and robustness, particularly in data limited medical settings. Through extensive evaluations on medical visual question answering (Med-VQA) datasets, we show that our methodology improves localization-driven performance across different Med-VLM architectures. Our analysis of localization-based questions further reveals that improvements in anatomy and spatial reasoning directly enhance the overall accuracy of Med-VQA upto 6.2%. The proposed approach is model-agnostic and can be seamlessly integrated into existing Med-VLM pipelines. The dataset, code, and trained models will be made publicly available at https://github.com/CVLABLUMS/localizationlens.
Figures & tables
| Model | VQA-RAD [ 9 ] | SLAKE [ 12 ] | Model Size | ||
|---|---|---|---|---|---|
| Open | Closed | Open | Closed | ||
| Medical Vision Language Models | |||||
| RaDialog [ 19 ] | 54.6 | 57.9 | 51.4 | 56.3 | 7B |
| RaDialog (with Lens) | 55.3 0.7 | 60.0 2.1 | 53.4 2.0 | 61.1 4.8 | 7B |
| Med-Flamingo [ 17 ] | 64.1 | 70.7 | 61.2 | 68.2 | 7B |
| Med-Flamingo (with Lens) | 65.8 1.7 | 72.1 1.4 | 65.0 3.8 | 71.6 3.4 | 7B |
| Model | Anatomical | Positioning |
|---|---|---|
| LLAVA-Med [ 10 ] | 4.7% | 5.6% |
| Med-Flamingo [ 17 ] | 3.3% | 5.2% |
| Phi* [ 7 , 30 ] | 2.8% | 3.6% |
| SmolVLM [ 23 ] (0.5B) | 4.2% | 5.8% |
| SmolVLM [ 23 ] (1.7B) | 3.7% | 6.1% |
| Lens | Loss | Shuffle | Metrics | ||||
|---|---|---|---|---|---|---|---|
| SAM | Med-SAM | Mask | InfoNCE | DCL | Pixel | Overall | Localization |
| ✗ | ✗ | ✗ | ✓ | ✗ | ✗ | 55.4% | - |
| ✓ | ✗ | ✗ | ✗ | ✓ | ✓ | 57.2% | 4.9% |
| ✗ | ✓ | ✗ | ✗ | ✓ | ✓ | 56.8% | 3.6% |
| ✓ | ✓ | ✗ | ✗ | ✓ | ✓ | 59.4% | 7.4% |
| ✓ | ✓ | ✓ | ✓ | ✗ | ✗ | 57.3% | 5.2% |
From Reasoning to Pixels: Grounded Medical Multimodal LLMs for VQA and Segmentation
[SEG] tokens within the MLLM's hidden states. Furthermore, a Reasoning-to-Visual (R2V) fusion mechanism is proposed to effectively inject these reasoning-aware features into a segmentation pipeline for accurate mask decoding. To facilitate this paradigm, we construct MedRAVS-13K, a comprehensive dataset comprising 13,824 expertly validated samples across four diverse imaging modalities. Extensive experiments demonstrate that MedREAL significantly outperforms state-of-the-arts, achieving 68.49% gIoU and 70.47% cIoU on benchmark evaluations. By generating evidence masks that are strictly consistent with textual diagnoses, MedREAL provides a robust, interpretable framework for reasoning-driven medical image analysis.