REVIEW 3 cited by
Zeroth-Order Policy Gradient for Reinforcement Learning from Human Feedback without Reward Inference
Not yet reviewed by Pith; the record is open.
This paper has not been read by Pith yet. Machine review is queued; the pith claim, tier, and objections will appear here once it completes.
SPECIMEN: schema-true, not a live event
T0 review · schema-true
One-sentence machine reading of the paper's core claim.
pith:XXXXXXXX · record.json · timestamp
Signed reviews
read the original abstract
Reward inference (learning a reward model from human preferences) is a critical intermediate step in the Reinforcement Learning from Human Feedback (RLHF) pipeline for fine-tuning Large Language Models (LLMs). In practice, RLHF faces fundamental challenges such as distribution shift, reward model overfitting, and problem misspecification. An alternative approach is direct policy optimization without reward inference, such as Direct Preference Optimization (DPO), which provides a much simpler pipeline and has shown empirical success in LLM applications. However, DPO utilizes the closed-form expression between the optimal policy and the reward function, which is only suitable under the bandit setting or deterministic MDPs. This paper develops two RLHF algorithms without reward inference for general RL problems beyond bandits and deterministic MDPs, and general preference models beyond the Bradley-Terry model. The key idea is to estimate the local value function difference from human preferences and then approximate the policy gradient with a zeroth-order gradient approximator. For both algorithms, we establish polynomial convergence rates in terms of the number of policy gradient iterations, the number of trajectory samples, and human preference queries per iteration. Numerical experiments in stochastic environments validate the performance of our proposed algorithms, outperforming popular RLHF baselines such as DPO and PPO. Our paper shows there exist provably efficient methods to solve general RLHF problems without reward inference.
Forward citations
Cited by 3 Pith papers
-
AI-Driven Stabilization in Power Grids through Controlling Line Admittances
Training an RL agent to rescale line admittances after single-line faults reduced frequency fluctuation by ~53% on a reduced UK-grid swing-equation model and produced a placement ranking that beats PTDF-based siting.
-
Distributed primal-dual algorithm for constrained multi-agent reinforcement learning under coupled policies
Coupled-policy constrained MARL is solved by a distributed primal-dual method that exchanges only local estimates and converges to an ε-first-order stationary point with geometrically decaying truncation error.
-
ElasticZO: A Memory-Efficient On-Device Learning with Combined Zeroth- and First-Order Optimization
ElasticZO uses zeroth-order gradients for most layers and backpropagation only for the last few, improving accuracy over pure ZO training with tiny memory overhead, and its INT8 variant runs integer-only.
Discussion (0). Continue with ORCID to comment.