Direct Advantage Regression: Aligning LLMs with Online AI Reward

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: He, Li, Zhao, He, Wan, Stephen, Wang, Dadong, Yao, Lina, Liu, Tongliang
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910915091234816
author He, Li
Zhao, He
Wan, Stephen
Wang, Dadong
Yao, Lina
Liu, Tongliang
author_facet He, Li
Zhao, He
Wan, Stephen
Wang, Dadong
Yao, Lina
Liu, Tongliang
contents Online AI Feedback (OAIF) presents a promising alternative to Reinforcement Learning from Human Feedback (RLHF) by utilizing online AI preference in aligning language models (LLMs). However, the straightforward replacement of humans with AI deprives LLMs from learning more fine-grained AI supervision beyond binary signals. In this paper, we propose Direct Advantage Regression (DAR), a simple alignment algorithm using online AI reward to optimize policy improvement through weighted supervised fine-tuning. As an RL-free approach, DAR maintains theoretical consistency with online RLHF pipelines while significantly reducing implementation complexity and improving learning efficiency. Our empirical results underscore that AI reward is a better form of AI supervision consistently achieving higher human-AI agreement as opposed to AI preference. Additionally, evaluations using GPT-4-Turbo and MT-bench show that DAR outperforms both OAIF and online RLHF baselines.
format Preprint
id arxiv_https___arxiv_org_abs_2504_14177
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Direct Advantage Regression: Aligning LLMs with Online AI Reward
He, Li
Zhao, He
Wan, Stephen
Wang, Dadong
Yao, Lina
Liu, Tongliang
Artificial Intelligence
Computation and Language
Human-Computer Interaction
Online AI Feedback (OAIF) presents a promising alternative to Reinforcement Learning from Human Feedback (RLHF) by utilizing online AI preference in aligning language models (LLMs). However, the straightforward replacement of humans with AI deprives LLMs from learning more fine-grained AI supervision beyond binary signals. In this paper, we propose Direct Advantage Regression (DAR), a simple alignment algorithm using online AI reward to optimize policy improvement through weighted supervised fine-tuning. As an RL-free approach, DAR maintains theoretical consistency with online RLHF pipelines while significantly reducing implementation complexity and improving learning efficiency. Our empirical results underscore that AI reward is a better form of AI supervision consistently achieving higher human-AI agreement as opposed to AI preference. Additionally, evaluations using GPT-4-Turbo and MT-bench show that DAR outperforms both OAIF and online RLHF baselines.
title Direct Advantage Regression: Aligning LLMs with Online AI Reward
topic Artificial Intelligence
Computation and Language
Human-Computer Interaction
url https://arxiv.org/abs/2504.14177