Fine-tune a vision+LLM for medical VQA and report writing using LoRA and one projection layer

January 5, 20248 min

Overview

Production Readiness

0.5

Novelty Score

0.6

Cost Impact Score

0.7

Citation Count

11

Authors

Jinlong He, Pengfei Li, Gang Liu, Genrong He, Zhaolin Chen, Shenjun Zhong

Links

Abstract / PDF

Why It Matters For Business

You can adapt large vision+LLM models to medical VQA and report generation cheaply by tuning small adapter layers; use GPT-4 for scalable semantic QA evaluation instead of brittle lexical metrics.

Summary TLDR

PeFoMed shows you can adapt a general multimodal LLM for medical visual question answering (Med-VQA) and radiology report generation (MRG) while only updating a small fraction of parameters. The authors freeze the ViT and LLM, fine-tune a single linear projection and LoRA adapters (56.63M trainable params), and use a two-stage recipe (image-caption pretrain then task fine-tune). Evaluations use exact-match metrics plus human and GPT-4 semantic judgments; GPT-4 aligns with humans and is more stable. Fine-tuned PeFoMed reaches competitive VQA accuracy and modest MRG scores, but lexical metrics still undercount generative model quality.

Problem Statement

Large multimodal LLMs are promising for medical imaging but full fine-tuning is costly and conventional lexical metrics (exact match, BLEU) miss semantic correctness on generative outputs. The paper asks: can parameter-efficient fine-tuning (PEFT) adapt general MLLMs to Med-VQA and report generation with few trainable parameters, and can GPT-4 provide reliable semantic evaluation?

Main Contribution

A practical PEFT pipeline (freeze ViT+LLM, fine-tune one linear projection + LoRA) for medical Med-VQA and MRG.

Two-stage fine-tuning recipe: image-caption pretraining on medical captions, then downstream VQA/report fine-tune.

Comparative evaluation using exact-match metrics, human annotation (10 people), and GPT-4 semantic scoring (5-point Likert + weighted average).

Empirical claim that GPT-4 semantic scores align with humans and are more stable than human raters, while lexical metrics can mislead.

Key Findings

PEFT keeps trainable footprint tiny: only projection + LoRA updated.

Numbers56.63M trainable vs 7B full LLM

Fine-tuned PeFoMed matches or exceeds closed-ended VQA accuracy of many specialized models.

NumbersVQA-RAD closed 87.1%, overall exact-match 77.4% (exact-match)

Exact-match metrics undercount generative answers; GPT-4 and humans rate higher on open-ended answers.

NumbersVQA-RAD open: exact 62.6% vs GPT-4 79.9% and human 77.7%

GPT-4 semantic scores closely match human judgments and show lower variability.

NumbersVQA-RAD closed GPT-4 87.5% vs human 87.6%; GPT-4 SD ≈0–1.2% vs human SD up to 3.2%

On report generation PeFoMed improves with stage-1 caption fine-tuning but lags best specialized report models by lexical metrics.

NumbersIU-Xray METEOR 0.157, ROUGE-L 0.286, CIDEr 0.462; WASM (GPT-4) 2.79

Results

VQA-RAD exact-match overall

Value77.4%

VQA-RAD exact-match closed

Value87.1%

VQA-RAD exact-match open

Value62.6%

VQA-RAD GPT-4 semantic overall

Value84.4% (WASM/semantic alignment reported)

SLAKE exact-match overall

Value82.1%

PathVQA exact-match overall

Value63.6%

IU-Xray METEOR / ROUGE-L / CIDEr

Value0.157 / 0.286 / 0.462

IU-Xray GPT-4 WASM (semantic 5-pt weighted average)

Value2.79 (1180 samples); 27.5% scored 4

Who Should Care

What To Try In 7 Days

Take a pretrained MiniGPT-v2 + ViT and add a single linear projection + LoRA; run stage-1 image-caption fine-tuning on available medical captions.

Run stage-2 fine-tuning on a small in-domain VQA or report dataset and compare exact-match vs GPT-4 semantic scores to detect evaluation gaps.

Use GPT-4 to label a sample of outputs and compare with a small human panel to validate automated evaluation before scaling.

Agent Features

Tool Use

  • GPT-4 used as an external evaluator

Frameworks

  • LoRA
  • MiniGPT-v2 codebase

Architectures

  • Vision-language (ViT encoder + LLM decoder)
  • LLaMA2-chat (7B) as decoder

Optimization Features

Infra Optimization

  • Runs on 4x NVIDIA A40 (48GB) GPUs

Model Optimization

  • LoRA

System Optimization

  • Group 4 image tokens into one visual embedding to cut compute

Training Optimization

  • Two-stage fine-tune: image-caption pretrain then task fine-tune
  • Freeze backbone weights (ViT and main LLM)

Reproducibility

Data Urls

  • VQA-RAD
  • SLAKE
  • PathVQA
  • IU-Xray
  • ROCO
  • CLEF2022
  • MEDICAT
  • MIMIC-CXR

Code Available

Data Available

Open Source Status

  • partial

Risks & Boundaries

Limitations

  • Lexical metrics (exact-match, CIDEr) underrepresent generative correctness for open-ended answers.
  • Report generation lags specialized MRG models on standard lexical scores despite semantic gains.
  • Evaluation relies on GPT-4 as proxy; GPT-4 may be biased toward higher similarity scores than humans.

When Not To Use

  • Do not rely solely on exact-match metrics to validate generative medical outputs.
  • Avoid using this model in clinical workflows without extra validation and safety checks.
  • Not ideal when you need state-of-the-art lexical scores on MRG benchmarks without extra engineering.

Failure Modes

  • Generative answers that are semantically correct but fail exact-match tests.
  • Hallucinated or clinically incorrect assertions not caught by automated semantic scoring.
  • Performance drops when target image types differ from caption pretraining distribution.

Core Entities

Models

  • PeFoMed (this paper)
  • MiniGPT-v2
  • LLaMA2-chat (7B)
  • LLaVA-Med
  • Med-Flamingo
  • BiomedGPT
  • GPT-4 / GPT-4v
  • EVA (ViT backbone)

Metrics

  • Exact Match
  • Token Recall (used by some baselines)
  • METEOR
  • ROUGE-L
  • CIDEr
  • 5-point Likert WASM (Weighted Average Semantic Metric)
  • GPT-4 semantic similarity

Datasets

  • VQA-RAD
  • SLAKE
  • PathVQA
  • IU-Xray
  • ROCO
  • CLEF2022
  • MEDICAT
  • MIMIC-CXR

Benchmarks

  • VQA-RAD
  • SLAKE
  • PathVQA
  • IU-Xray