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

1 State Key Laboratory for Novel Software Technology, Nanjing University
2 School of Computer Science and Artificial Intelligence, Zhengzhou University
Corresponding author


Preprint 2026

S2T-RLHF framework


A global sequence-level reward is first decomposed into sentence-level credits via inter-sentence bargaining (Stage I), and then refined into token-level rewards through Dirichlet posterior inference (Stage II), enabling stable and interpretable credit assignment in RLHF without modifying the reward model.

Abstract

Reinforcement learning from human feedback (RLHF) often exhibits unstable training dynamics. One contributing reason is that RLHF relies on a single sequence-level scalar reward that is propagated to token-level updates during optimization, making credit assignment inherently ambiguous. Recent work explores reward refinement to provide denser supervision in RLHF, often motivated by increasingly accurate reward decomposition and the implicit assumption that finer-grained credit assignment improves optimization. We argue that this assumption is suboptimal: overly fine-grained reward refinement can amplify noise and uncertainty in the reward signal, ultimately degrading model performance and training stability. In light of this, we propose a granularity-aware reward refinement principle that emphasizes stability-oriented reward design rather than maximal allocation precision. Under this principle, we identify sentences as a natural intermediate granularity for reward allocation, balancing robustness to noise with the ability to convey semantically meaningful supervision. Guided by this principle, we introduce S2T-RLHF, a sentence-to-token reward decomposition framework that enforces structured, semantically grounded credit assignment while requiring no reward-model retraining or token-level supervision. Experiments across multiple datasets demonstrate that S2T-RLHF yields more stable optimization dynamics and improved robustness and generalization.

BibTeX

@article{chen2026s2t,
  title={S2T-RLHF: Hierarchical Credit Assignment for Stable Preference-Based RLHF},
  author={Chen, Wei and Zhu, Guanghui and Li, Yafei and Wang, Limin and Huang, Yihua},
  journal={arXiv preprint arXiv:2607.18258},
  year={2026}
}