MedLoc-R1: Performance-Aware Curriculum Reward Scheduling for GRPO-Based Medical Visual Grounding

View PDF HTML (experimental)

Abstract:Medical visual grounding serves as a crucial foundation for fine-grained multimodal reasoning and interpretable clinical decision support. Despite recent advances in reinforcement learning (RL) for grounding tasks, existing approaches such as Group Relative Policy Optimization~(GRPO) suffer from severe reward sparsity when directly applied to medical images, primarily due to the inherent difficulty of localizing small or ambiguous regions of interest, which is further exacerbated by the rigid and suboptimal nature of fixed IoU-based reward schemes in RL. This leads to vanishing policy gradients and stagnated optimization, particularly during early training. To address this challenge, we propose MedLoc-R1, a performance-aware reward scheduling framework that progressively tightens the reward criterion in accordance with model readiness. MedLoc-R1 introduces a sliding-window performance tracker and a multi-condition update rule that automatically adjust the reward schedule from dense, easily obtainable signals to stricter, fine-grained localization requirements, while preserving the favorable properties of GRPO without introducing auxiliary networks or additional gradient paths. Experiments on three medical visual grounding benchmarks demonstrate that MedLoc-R1 consistently improves both localization accuracy and training stability over GRPO-based baselines. Our framework offers a general, lightweight, and effective solution for RL-based grounding in high-stakes medical applications. Code & checkpoints are available at \hyperlink{}{this https URL}.

Comments: 2026 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR)

Subjects:

Computer Vision and Pattern Recognition (cs.CV)

Cite as: arXiv:2603.28120 [cs.CV]

(or arXiv:2603.28120v1 [cs.CV] for this version)

https://doi.org/10.48550/arXiv.2603.28120

arXiv-issued DOI via DataCite (pending registration)

Submission history

From: Yang Guangjing [view email] [v1] Mon, 30 Mar 2026 07:31:21 UTC (2,202 KB)