Towards a clinically accessible radiology foundation model: open-access and lightweight, with automated evaluation

Abstract

The scaling laws and extraordinary performance of large foundation models motivate the development and utilization of such models in biomedicine. However, despite early promising results on some biomedical benchmarks, there are still major challenges that need to be addressed before these models can be used in real-world clinics. Frontier general-domain models such as GPT-4V still have significant performance gaps in multimodal biomedical applications. More importantly, less-acknowledged pragmatic issues, including accessibility, model cost, and tedious manual evaluation make it hard for clinicians to use state-of-the-art large models directly on private patient data. Here, we explore training open-source small multimodal models (SMMs) to bridge competency gaps for unmet clinical needs in radiology. To maximize data efficiency, we adopt a modular approach by incorporating state-of-the-art pre-trained models for image and text modalities, and focusing on training a lightweight adapter to ground each modality to the text embedding space, as exemplified by LLaVA-Med. For training, we assemble a large dataset of over 697 thousand radiology image-text pairs. For evaluation, we propose CheXprompt, a GPT-4-based metric for factuality evaluation, and demonstrate its parity with expert evaluation. For best practice, we conduct a systematic ablation study on various choices in data engineering and multimodal training. The resulting LlaVA-Rad (7B) model attains state-of-the-art results on standard radiology tasks such as report generation and cross-modal retrieval, even outperforming much larger models such as GPT-4V and Med-PaLM M (84B). The inference of LlaVA-Rad is fast and can be performed on a single V100 GPU in private settings, offering a promising state-of-the-art tool for real-world clinical applications.

References in corpus (10)

• Flamingo: a Visual Language Model for Few-Shot Learning
• ChatGPT Outperforms Crowd-Workers for Text-Annotation Tasks
• PadChest: A large chest x-ray image dataset with multi-label annotated reports
• LLaVA-Med: Training a Large Language-and-Vision Assistant for Biomedicine in One Day
• Phi-3 Technical Report: A Highly Capable Language Model Locally on Your Phone
• Qwen2-VL: Enhancing Vision-Language Model's Perception of the World at Any Resolution
• Attention-based Saliency Maps Improve Interpretability of Pneumothorax Classification
• A Vision-Language Foundation Model to Enhance Efficiency of Chest X-ray Interpretation
• MAIRA-1: A specialised large multimodal model for radiology report generation
• Radiology Report Generation Using Transformers Conditioned with Non-imaging Data

Cited by in corpus (1)

• Merlin: A Computed Tomography Vision-Language Foundation Model and Dataset