REAL: Regression-Aware Reinforcement Learning for LLM-as-a-Judge
Not on the map. Why?
Abstract (arXiv)
Large language models (LLMs) are increasingly deployed as automated evaluators that assign numeric scores to model outputs, a paradigm known as LLM-as-a-Judge. However, standard Reinforcement Learning (RL) methods typically rely on binary rewards (e.g., 0-1 accuracy), thereby ignoring the ordinal structure inherent in regression tasks; for instance, they fail to recognize that predicting 4 is significantly better than predicting 1 when the ground truth is 5. Conversely, existing regression-aware approaches are often confined to Supervised Fine-Tuning (SFT), limiting their ability to explore optimal reasoning paths. To bridge this gap, we propose REAL (underlineREgression-underlineAware Reinforcement underlineLearning), a principled RL framework designed to optimize regression rewards, and also proven to be optimal for correlation metrics. A key technical challenge is that the regression objective is explicitly policy-dependent, thus invalidating standard policy gradient methods. To address this, we employ the generalized policy gradient estimator, which naturally decomposes optimization into two complementary components: (1) exploration over Chain-of-Thought (CoT) trajectory, and (2) regression-aware prediction refinement of the final score. Extensive experiments across model scales (8B to 32B) demonstrate that REAL consistently outperforms both regression-aware SFT baselines and standard RL methods, exhibiting significantly better generalization on out-of-domain benchmarks. On Qwen3-32B specifically, we achieve gains of +8.40 Pearson and +7.20 Spearman correlation over the SFT baseline, and +18.30/+11.20 over the base model. These findings highlight the critical value of integrating regression objectives into RL exploration for accurate LLM evaluation.
Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Builds on 31 of its 48 references that are in the map. First posted on arXiv 17 Mar 2026.
Builds on
31 of 48 references.
References · 31
- 2020Language Models are Few-Shot Learnerscited by 32,419
- 2023GPT-4 Technical Reportcited by 18,396
- 2022Training language models to follow instructions with human feedbackcited by 15,971
- 2017Proximal Policy Optimization Algorithmscited by 15,215
- 2023Llama 2: Open Foundation and Fine-Tuned Chat Modelscited by 13,028
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,379
- 2025Qwen3 Technical Reportcited by 8,820
- 2023Judging LLM-as-a-Judge with MT-Bench and Chatbot Arenacited by 8,626
- 2020Scaling Laws for Neural Language Modelscited by 6,586
- 2025DAPO: An Open-Source LLM Reinforcement Learning System at Scalecited by 2,810
- 2023G-Eval: NLG Evaluation using GPT-4 with Better Human Alignmentcited by 2,276
- 2024HybridFlow: A Flexible and Efficient RLHF Frameworkcited by 2,057
19 more
- 2024A Survey on LLM-as-a-Judgecited by 1,423
- 2021A General Language Assistant as a Laboratory for Alignmentcited by 1,055
- 2023Can Large Language Models Be an Alternative to Human Evaluations?cited by 734
- 2023RAFT: Reward rAnked FineTuning for Generative Foundation Model Alignmentcited by 686
- 2023Prometheus: Inducing Fine-grained Evaluation Capability in Language Modelscited by 575
- 2024From Crowdsourced Data to High-Quality Benchmarks: Arena-Hard and BenchBuilder Pipelinecited by 500
- 2024Prometheus 2: An Open Source Language Model Specialized in Evaluating Other Language Modelscited by 446
- 2023JudgeLM: Fine-tuned Large Language Models are Scalable Judgescited by 367
- 2023PandaLM: An Automatic Evaluation Benchmark for LLM Instruction Tuning Optimizationcited by 316
- 2024Reinforcement learningcited by 247
- 2023Generative Judge for Evaluating Alignmentcited by 169
- 2015Gradient Estimation Using Stochastic Computation Graphscited by 121
- 2025Not All Rollouts are Useful: Down-Sampling Rollouts in LLM Reinforcement Learningcited by 84
- 2025J1: Incentivizing Thinking in LLM-as-a-Judge via Reinforcement Learningcited by 75
- 2017Reinforcement Learning for Bandit Neural Machine Translation with Simulated Human Feedbackcited by 72
- 2025Beyond Verifiable Rewards: Scaling Reinforcement Learning for Language Models to Unverifiable Datacited by 29
- 2025TRACT: Regression-Aware Fine-tuning Meets Chain-of-Thought Reasoning for LLM-as-a-Judgecited by 17
- 2024Regression-aware Inference with LLMscited by 14
- 2024Regress, Don't Guess -- A Regression-like Loss on Number Tokens for Language Modelscited by 13
Built on by
Citing papers per year: 2026: 2.
2 papers.
Similar, not cited
Close in wording, no citation either way
- 2025Accelerating RL for LLM Reasoning with Optimal Advantage Regression
- 2024Reinforcement Learning for LLM Post-Training: A Survey
- 2025Response-Level Rewards Are All You Need for Online Reinforcement Learning in LLMs: A Mathematical Perspective
- 2025A Comedy of Estimators: On KL Regularization in RL Training of LLMs
- 2026Reinforcement Learning over Predictive Distributions for LLM Regression
- 2024Exploring RL-based LLM Training for Formal Language Tasks with Programmed Rewards
- 2025TinyV: Reducing False Negatives in Verification Improves RL for LLM Reasoning
- 2024Teaching Large Language Models to Reason with Reinforcement Learning