English

DIVA-GRPO: Enhancing Multimodal Reasoning through Difficulty-Adaptive Variant Advantage

Artificial Intelligence 2026-03-03 v1

Abstract

Reinforcement learning (RL) with group relative policy optimization (GRPO) has become a widely adopted approach for enhancing the reasoning capabilities of multimodal large language models (MLLMs). While GRPO enables long-chain reasoning without a critic, it often suffers from sparse rewards on difficult problems and advantage vanishing when group-level rewards are too consistent for overly easy or hard problems. Existing solutions (sample expansion, selective utilization, and indirect reward design) often fail to maintain enough variance in within-group reward distributions to yield clear optimization signals. To address this, we propose DIVA-GRPO, a difficulty-adaptive variant advantage method that adjusts variant difficulty distributions from a global perspective. DIVA-GRPO dynamically assesses problem difficulty, samples variants with appropriate difficulty levels, and calculates advantages across local and global groups using difficulty-weighted and normalized scaling. This alleviates reward sparsity and advantage vanishing while improving training stability. Extensive experiments on six reasoning benchmarks demonstrate that DIVA-GRPO outperforms existing approaches in training efficiency and reasoning performance. Code: https://github.com/Siaaaaaa1/DIVA-GRPO

Keywords

Cite

@article{arxiv.2603.01106,
  title  = {DIVA-GRPO: Enhancing Multimodal Reasoning through Difficulty-Adaptive Variant Advantage},
  author = {Haowen Gao and Zhenyu Zhang and Liang Pang and Fangda Guo and Hongjian Dou and Guannan Lv and Shaoguo Liu and Tingting Gao and Huawei Shen and Xueqi Cheng},
  journal= {arXiv preprint arXiv:2603.01106},
  year   = {2026}
}

Comments

Accepted to ICLR 2026. Code and models are available at https://github.com/Siaaaaaa1/DIVA-GRPO

R2 v1 2026-07-01T10:57:58.819Z