S2T-RLHF: Hierarchical Credit Assignment for Stable Preference-Based RLHF
On the map: LLM alignment / RLHF / preference optimisation → Reward modeling and RLHF
Abstract (arXiv)
Reinforcement learning from human feedback (RLHF) with preference-based reward models often exhibits unstable training dynamics. A key contributing factor is that standard RLHF relies on a single sequence-level scalar reward, which is propagated to token-level policy updates and leaves credit assignment within a response inherently ambiguous. Recent work has attempted to address this issue by refining rewards into denser token-level supervision, often relying on the implicit assumption that finer-grained credit assignment improves optimization. We argue that this assumption is incomplete: when preference signals are noisy and only defined at the response level, overly fine-grained reward refinement can amplify reward uncertainty and destabilize learning. To address this problem, we propose a granularity-aware principle for hierarchical credit assignment, emphasizing stability-oriented reward design rather than maximal allocation precision. Under this principle, sentences serve as a natural intermediate granularity, balancing semantic coherence with robustness to token-level noise. Guided by this view, we introduce S2T-RLHF. This sentence-to-token reward decomposition framework first allocates sequence-level preference rewards across sentences and then applies bounded token-level refinement within each sentence, without reward-model retraining or token-level supervision. Experiments across multiple datasets and optimization settings show that S2T-RLHF improves training stability and robustness while maintaining competitive preference alignment.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Reward modeling and RLHF, one of 10 directions in LLM alignment / RLHF / preference optimisation. Builds on 42 of its 47 references that are in the map. First posted on arXiv 15 May 2026.
Builds on
42 of 47 references.
Closest ancestors · 19
Same direction, newest first
- 2025Preference Learning for AI Alignment: a Causal Perspective6 Jun
- 2025SCAR: Shapley Credit Assignment for More Efficient RLHF26 May
- 2025Information-Theoretic Reward Decomposition for Generalizable RLHF8 Apr
- 2024MA-RLHF: Reinforcement Learning from Human Feedback with Macro Actions3 Oct
- 2024TLCR: Token-Level Continuous Reward for Fine-grained Reinforcement Learning from Human Feedback23 Jul
- 2024Regularizing Hidden States Enables Learning Generalizable Reward Model for LLMs14 Jun
- 2024A Critical Look At Tokenwise Reward-Guided Text Generation12 Jun
- 2024DPO Meets PPO: Reinforced Token Optimization for RLHF29 Apr
- 2024InfoRM: Mitigating Reward Hacking in RLHF via Information-Theoretic Reward Modeling14 Feb
- 2024Dense Reward for Free in Reinforcement Learning from Human Feedback1 Feb
- 2023Contrastive Preference Learning: Learning from Human Feedback without RL20 Oct
- 2023Confronting Reward Model Overoptimization with Constrained RLHF6 Oct
7 more
- 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Model29 May
- 2023RAFT: Reward rAnked FineTuning for Generative Foundation Model Alignment13 Apr
- 2022Scaling Laws for Reward Model Overoptimization19 Oct
- 2022Training a Helpful and Harmless Assistant with Reinforcement Learning from Human Feedback12 Apr
- 2019Fine-Tuning Language Models from Human Preferences18 Sep
- 2018Learning to Understand Goal Specifications by Modelling Reward5 Jun
- 2017Deep reinforcement learning from human preferences12 Jun
Other references · 23
- 2020Language Models are Few-Shot Learnerscited by 32,199
- 2023GPT-4 Technical Reportcited by 18,132
- 2022Training language models to follow instructions with human feedbackcited by 15,712
- 2017Proximal Policy Optimization Algorithmscited by 14,441
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
- 2023Judging LLM-as-a-Judge with MT-Bench and Chatbot Arenacited by 8,393
- 2024GPT-4o System Cardcited by 4,467
- 2021Finetuned Language Models Are Zero-Shot Learnerscited by 3,747
- 2023Universal and Transferable Adversarial Attacks on Aligned Language Modelscited by 2,914
- 2020Learning to summarize from human feedbackcited by 2,562
- 2020RealToxicityPrompts: Evaluating Neural Toxic Degeneration in Language Modelscited by 1,235
- 2024ORPO: Monolithic Preference Optimization without Reference Modelcited by 556
11 more
- 2022ToxiGen: A Large-Scale Machine-Generated Dataset for Adversarial and Implicit Hate Speech Detectioncited by 419
- 2022The Alignment Problem from a Deep Learning Perspectivecited by 246
- 2022Multi-Task Learning as a Bargaining Gamecited by 209
- 2023Retroformer: Retrospective Large Language Agents with Policy Gradient Optimizationcited by 110
- 2024Panacea: Pareto Alignment via Preference Adaptation for LLMscited by 78
- 2024Stepwise Alignment for Constrained Language Model Policy Optimizationcited by 30
- 2024Spread Preference Annotation: Direct Preference Judgment for Efficient LLM Alignmentcited by 27
- 2023Privately Aligning Language Models with Reinforcement Learningcited by 17
- 2024Fairness-Aware Meta-Learning via Nash Bargainingcited by 6
- 2023Hindsight-DICE: Stable Credit Assignment for Deep Reinforcement Learningcited by 4
- 2025RLHF: A comprehensive Survey for Cultural, Multimodal and Low Latency Alignment Methodscited by 2
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026From Reasoning to Agentic: Credit Assignment in Reinforcement Learning for Large Language Models
- 2024RED: Unleashing Token-Level Rewards from Holistic Feedback via Reward Redistribution
- 2026Reinforcement Learning from Human Feedback: A Statistical Perspective
- 2026Credit Assignment with Resets in Language Model Reasoning
- 2026PACT: From Credit Assignment to Critic Alignment
- 2026DelTA: Discriminative Token Credit Assignment for Reinforcement Learning from Verifiable Rewards
- 2024T-REG: Preference Optimization with Token-Level Reward Regularization
- 2026ARCA: Adapter-Residual Credit Assignment When Token Signals Degenerate