English

Injecting Distributional Awareness into MLLMs via Reinforcement Learning for Deep Imbalanced Regression

Computation and Language 2026-05-12 v2 Computer Vision and Pattern Recognition Machine Learning

Abstract

Multimodal large language models (MLLMs) struggle with numerical regression under long-tailed target distributions. Token-level supervised fine-tuning (SFT) and point-wise regression rewards bias learning toward high-density regions, leading to regression-to-the-mean behavior and poor tail performance. We identify the lack of cross-sample relational supervision as a key limitation of existing MLLM training paradigms. To address it, we propose a distribution-aware reinforcement learning framework based on Group Relative Policy Optimization, which introduces batch-level comparison-based supervision via the Concordance Correlation Coefficient-based reward to align predicted and ground-truth distributions in terms of correlation, scale, and mean. The framework is plug-and-play, requiring no architectural modification. Experiments on a unified suite of long-tailed regression benchmarks show consistent improvements over SFT and existing MLLM regression methods, with particularly strong gains in medium- and few-shot regimes.

Keywords

Cite

@article{arxiv.2605.01402,
  title  = {Injecting Distributional Awareness into MLLMs via Reinforcement Learning for Deep Imbalanced Regression},
  author = {Yao Du and Shanshan Song and Xiaomeng Li},
  journal= {arXiv preprint arXiv:2605.01402},
  year   = {2026}
}

Comments

Accepted by ICML 2026

R2 v1 2026-07-01T12:46:36.752Z