atlas

Research map / Vision-language and multimodal / Human and embodied modeling

PromptMRG: Diagnosis-Driven Prompts for Medical Report Generation

AAAI 2024 · Vision-language and multimodal › Human and embodied modeling · approach: Medical vision-language models · 123 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes PromptMRG, which converts a disease classification branch's outputs into token prompts to guide a chest X-ray report decoder, and adds retrieval-based feature enhancement and self-adaptive logit-adjusted loss to improve clinical efficacy on MIMIC-CXR and IU X-Ray.

Problem

Medical report generation from chest X-rays often produces diagnostically inaccurate reports, and rare diseases are underrepresented, so their diagnostic performance is unreliable.

Approach

Diagnostic outputs of an auxiliary classification branch are turned into special token prompts ([POS], [NEG], [BLA], [UNC]) that explicitly condition the text decoder; retrieval of similar reports via a frozen CLIP enhances classification features, and an adaptive logit-adjusted loss, driven by per-disease prediction scores, rebalances rare diseases.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
add diagnosis-driven token prompts (ablation)PromptMRG · MIMIC-CXR testF1: 0.370 -> 0.444Helped
add auxiliary disease labeling (ablation)PromptMRG with DDP · MIMIC-CXR testF1: 0.444 -> 0.451Helped
add cross-modal feature enhancement (ablation)PromptMRG with DDP and ADL · MIMIC-CXR testF1: 0.451 -> 0.464Helped
add self-adaptive disease-balanced learning (ablation)PromptMRG with DDP and ADL · MIMIC-CXR testF1: 0.451 -> 0.468Helped
add diagnosis-driven prompts (ablation)PromptMRG · MIMIC-CXR testBLEU-4: 0.116 -> 0.106Hurt
replace token prompts with text prompts (ablation)PromptMRG · MIMIC-CXR testF1: 0.451 -> 0.446Hurt
replace token prompts with feature prompts (ablation)PromptMRG · MIMIC-CXR testF1: 0.370 -> 0.357Hurt
replace token prompts with embedding prompts (ablation)PromptMRG · MIMIC-CXR testF1: 0.370 -> 0.434Helped
add SDL (ablation)PromptMRG · MIMIC testMacro-averaged F1: 0.319 -> 0.381Helped
add SDL (ablation)PromptMRG · IU testMacro-averaged F1: 0.193 -> 0.246Helped
use token prompts instead of text prompts (ablation)PromptMRG · MIMIC-CXR testF1: 0.446 -> 0.451Helped

Surprising findings

Limitations

Experimental setup

Datasets: MIMIC-CXR IU X-Ray

Benchmarks: MIMIC-CXR IU X-Ray

Metrics: CE precision CE recall CE F1 BLEU-1 BLEU-4 METEOR ROUGE-L

Baselines: R2Gen M2TR MKSG CliBert CVT2Dis. M2KT METrans. KiUT DCL RGRG

Builds on: logit-adjusted loss CLIP CheXbert R2Gen

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy