Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

MedGRPO: Multi-Task Reinforcement Learning for Heterogeneous Medical Video Understanding

About

Large vision-language models struggle with medical video understanding, where spatial precision, temporal reasoning, and clinical semantics are critical. To address this, we first introduce \textbf{MedVidBench}, a large-scale benchmark of 531,850 video-instruction pairs across 8 medical sources spanning video, segment, and frame-level tasks, curated through a rigorous quality assurance pipeline with expert-guided prompting and dual-model validation. While supervised fine-tuning on MedVidBench yields noticeable gains, standard Reinforcement Learning (RL) fails due to imbalanced reward scales across datasets, which destabilizes optimization and leads to training collapse. To overcome this, we introduce \textbf{MedGRPO}, a novel RL framework for balanced multi-dataset training with two key innovations: (1) \emph{cross-dataset reward normalization} that maps each dataset's median performance to a common reward value, ensuring fair optimization regardless of difficulty, and (2) a \emph{medical LLM judge} that evaluates caption quality on five clinical dimensions through comparative similarity scoring. Supervised fine-tuning Qwen2.5-VL-7B on MedVidBench outperforms GPT-4.1 and Gemini-2.5-Flash across all tasks, while MedGRPO further improves the SFT baseline on grounding and captioning. Our work establishes a foundational benchmark and training methodology for advancing medical video understanding with VLMs. Our project website is available at: https://uii-america.github.io/MedGRPO/.

Yuhao Su, Anwesa Choudhuri, Zhongpai Gao, Benjamin Planche, Van Nguyen Nguyen, Meng Zheng, Yuhan Shen, Arun Innanje, Terrence Chen, Ehsan Elhamifar, Ziyan Wu• 2025

Related benchmarks

TaskDatasetResultRank
Critical View of SafetyMedVidBench
CVS Accuracy91.4
5
Dense Video CaptioningMedVidBench
DVCllm3.797
5
Next Action PredictionMedVidBench
NAP Accuracy44.2
5
Region CaptioningMedVidBench
RCllm3.442
5
Surgical ActionMedVidBench
SA Accuracy24.4
5
Surgical Temporal GroundingMedVidBench
STGmIoU0.202
5
Temporal Action GroundingMedVidBench
TAGmIoU@0.321.6
5
Video SummarizationMedVidBench
VSllm Score4.184
5
Showing 8 of 8 rows

Other info

Follow for update