mlsanityData to 7 Oct 2026

S2T-RLHF: Hierarchical Credit Assignment for Stable Preference-Based RLHF

arXiv 2607.18258 ↗First posted 15 May 2026cs.AIcited by 0

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

  1. 2025Preference Learning for AI Alignment: a Causal Perspective6 Jun
  2. 2025SCAR: Shapley Credit Assignment for More Efficient RLHF26 May
  3. 2025Information-Theoretic Reward Decomposition for Generalizable RLHF8 Apr
  4. 2024MA-RLHF: Reinforcement Learning from Human Feedback with Macro Actions3 Oct
  5. 2024TLCR: Token-Level Continuous Reward for Fine-grained Reinforcement Learning from Human Feedback23 Jul
  6. 2024Regularizing Hidden States Enables Learning Generalizable Reward Model for LLMs14 Jun
  7. 2024A Critical Look At Tokenwise Reward-Guided Text Generation12 Jun
  8. 2024DPO Meets PPO: Reinforced Token Optimization for RLHF29 Apr
  9. 2024InfoRM: Mitigating Reward Hacking in RLHF via Information-Theoretic Reward Modeling14 Feb
  10. 2024Dense Reward for Free in Reinforcement Learning from Human Feedback1 Feb
  11. 2023Contrastive Preference Learning: Learning from Human Feedback without RL20 Oct
  12. 2023Confronting Reward Model Overoptimization with Constrained RLHF6 Oct
7 more
  1. 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Model29 May
  2. 2023RAFT: Reward rAnked FineTuning for Generative Foundation Model Alignment13 Apr
  3. 2022Scaling Laws for Reward Model Overoptimization19 Oct
  4. 2022Training a Helpful and Harmless Assistant with Reinforcement Learning from Human Feedback12 Apr
  5. 2019Fine-Tuning Language Models from Human Preferences18 Sep
  6. 2018Learning to Understand Goal Specifications by Modelling Reward5 Jun
  7. 2017Deep reinforcement learning from human preferences12 Jun

Other references · 23

  1. 2020Language Models are Few-Shot Learnerscited by 32,199
  2. 2023GPT-4 Technical Reportcited by 18,132
  3. 2022Training language models to follow instructions with human feedbackcited by 15,712
  4. 2017Proximal Policy Optimization Algorithmscited by 14,441
  5. 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
  6. 2023Judging LLM-as-a-Judge with MT-Bench and Chatbot Arenacited by 8,393
  7. 2024GPT-4o System Cardcited by 4,467
  8. 2021Finetuned Language Models Are Zero-Shot Learnerscited by 3,747
  9. 2023Universal and Transferable Adversarial Attacks on Aligned Language Modelscited by 2,914
  10. 2020Learning to summarize from human feedbackcited by 2,562
  11. 2020RealToxicityPrompts: Evaluating Neural Toxic Degeneration in Language Modelscited by 1,235
  12. 2024ORPO: Monolithic Preference Optimization without Reference Modelcited by 556
11 more
  1. 2022ToxiGen: A Large-Scale Machine-Generated Dataset for Adversarial and Implicit Hate Speech Detectioncited by 419
  2. 2022The Alignment Problem from a Deep Learning Perspectivecited by 246
  3. 2022Multi-Task Learning as a Bargaining Gamecited by 209
  4. 2023Retroformer: Retrospective Large Language Agents with Policy Gradient Optimizationcited by 110
  5. 2024Panacea: Pareto Alignment via Preference Adaptation for LLMscited by 78
  6. 2024Stepwise Alignment for Constrained Language Model Policy Optimizationcited by 30
  7. 2024Spread Preference Annotation: Direct Preference Judgment for Efficient LLM Alignmentcited by 27
  8. 2023Privately Aligning Language Models with Reinforcement Learningcited by 17
  9. 2024Fairness-Aware Meta-Learning via Nash Bargainingcited by 6
  10. 2023Hindsight-DICE: Stable Credit Assignment for Deep Reinforcement Learningcited by 4
  11. 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

  1. 2026From Reasoning to Agentic: Credit Assignment in Reinforcement Learning for Large Language Models
  2. 2024RED: Unleashing Token-Level Rewards from Holistic Feedback via Reward Redistribution
  3. 2026Reinforcement Learning from Human Feedback: A Statistical Perspective
  4. 2026Credit Assignment with Resets in Language Model Reasoning
  5. 2026PACT: From Credit Assignment to Critic Alignment
  6. 2026DelTA: Discriminative Token Credit Assignment for Reinforcement Learning from Verifiable Rewards
  7. 2024T-REG: Preference Optimization with Token-Level Reward Regularization
  8. 2026ARCA: Adapter-Residual Credit Assignment When Token Signals Degenerate