mlsanityData to 14 Oct 2026

Tail-Aware Information-Theoretic Bounds for LLM Alignment under Heavy-Tailed Rewards

arXiv 2604.10727 ↗First posted 12 Apr 2026stat.ML cs.AI cs.LGcited by 0

On the map: LLM alignment / RLHF / preference optimisation → LLM preference optimization methods

Abstract (arXiv)

Classical information-theoretic learning bounds typically rely on KL mutual information and moment-generating-function (MGF) arguments, which are well matched to bounded or sub-Gaussian losses but can be ineffective when losses or rewards are heavy-tailed. We develop a tail-aware information-theoretic framework for sub-Weibull data, where the tail parameter θ controls the tail heaviness: θ=2 corresponds to sub-Gaussian, θ=1 to sub-exponential, and 0<θ<1 to genuinely heavy tails. Our key technical ingredient is a decorrelation lemma that bounds change-of-measure expectations using a shifted-log fθ-divergence, which admits explicit comparisons to R'enyi divergence without MGF arguments. On the empirical-process side, we establish sharp maximal inequalities and a Dudley-type chaining bound for sub-Weibull processes, with logarithmic and entropy terms raised to the power 1/θ. These tools yield tail-adaptive selection bounds and a multiscale information-theoretic Dudley inequality based on shifted-log and R'enyi mutual information. We apply our theory to large language models (LLMs) in the context of reward hacking within reinforcement learning from human feedback (RLHF). We show that R'enyi-regularized alignment provides finite reward guarantees and ensures that best-of-N policies remain well-controlled, thereby mitigating the catastrophic Goodhart effects where standard KL-regularization fails. We illustrate R'enyi-regularized RLHF by experiments, including controlled heavy-tailed rewards and token-space reward attacks.

Key facts. No citing papers in the map yet (data through 14 Oct 2026). Placed in LLM preference optimization methods, one of 10 directions in LLM alignment / RLHF / preference optimisation. Builds on 32 of its 45 references that are in the map. First posted on arXiv 12 Apr 2026.

Builds on

32 of 45 references.

Closest ancestors · 4

Same direction, newest first

  1. 2024Correcting the Mythos of KL-Regularization: Direct Alignment without Overoptimization via Chi-Squared Preference Optimization18 Jul
  2. 2024Information Theoretic Guarantees For Policy Alignment In Large Language Models9 Jun
  3. 2024Asymptotics of Language Model Alignment2 Apr
  4. 2022Training language models to follow instructions with human feedback4 Mar

Other references · 28

  1. 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 9,379
  2. 2025Qwen3 Technical Reportcited by 8,820
  3. 2020Learning to summarize from human feedbackcited by 2,588
  4. 2021WebGPT: Browser-assisted question-answering with human feedbackcited by 1,745
  5. 2024WildChat: 1M ChatGPT Interaction Logs in the Wildcited by 629
  6. 2017Information-theoretic analysis of generalization capability of learning algorithmscited by 303
  7. 2025Skywork-Reward-V2: Scaling Preference Data Curation via Human-AI Synergycited by 214
  8. 2024Theoretical guarantees on the best-of-n alignment policycited by 124
  9. 2019Tightening Mutual Information Based Bounds on Generalization Errorcited by 117
  10. 2015How much does your data exploration overfit? Controlling bias via information usagecited by 95
  11. 2018Chaining Mutual Information and Tightening Generalization Boundscited by 65
  12. 2023Generalization Bounds: Perspectives from Information Theory and PAC-Bayescited by 61
16 more
  1. 2018Moving Beyond Sub-Gaussianity in High-Dimensional Statistics: Applications in Covariance Estimation and Linear Regressioncited by 42
  2. 2024Bayesian Reward Models for LLM Alignmentcited by 35
  3. 2024Reward-Robust RLHF in LLMscited by 32
  4. 2024Catastrophic Goodhart: regularizing RLHF with KL divergence does not mitigate heavy-tailed reward misspecificationcited by 23
  5. 2024WildVis: Open Source Visualizer for Million-Scale Chat Logs in the Wildcited by 20
  6. 2019Lecture Notes: Selected topics on robust statistical learning theorycited by 17
  7. 2021Optimistic Rates: A Unifying Theory for Interpolation Learning and Regularization in Linear Regressioncited by 17
  8. 2020Sharp Concentration Results for Heavy-Tailed Distributionscited by 15
  9. 2023Algorithmic Stability of Heavy-Tailed SGD with General Loss Functionscited by 15
  10. 2021Sharper Sub-Weibull Concentrationscited by 8
  11. 2022Non-Asymptotic Guarantees for Robust Statistical Learning under Infinite Variance Assumptioncited by 6
  12. 2023A unified framework for information-theoretic generalization boundscited by 6
  13. 2025Do Large Language Models (Really) Need Statistical Foundations?cited by 5
  14. 2026Beyond Semantic Manipulation: Token-Space Attacks on Reward Modelscited by 3
  15. 2025Log-Sum-Exponential Estimator for Off-Policy Evaluation and Learningcited by 2
  16. 2023Concentration inequalities for high-dimensional linear processes with dependent innovationscited by 1

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2026Best-of-Tails: Bridging Optimism and Pessimism in Inference-Time Alignment
  2. 2026Tail Annealing for Heavy-Tailed Flow Matching
  3. 2026Autoregressive Learning in Joint KL: Sharp Oracle Bounds and Lower Bounds
  4. 2024Large Deviation Upper Bounds and Improved MSE Rates of Nonlinear SGD: Heavy-tailed Noise and Power of Symmetry
  5. 2026ExpBoN: Exponential-Noise Best-of-n for Efficient Test-Time LLM Alignment
  6. 2026Robust Langevin Sampling under State-Dependent Heavy-Tailed Noise with Applications to LLM Fine-Tuning
  7. 2026Error Bounds for Statistical Estimators in BTL Model with Parametric Multivariate Utility Functions
  8. 2026Heavy-Tailed and Long-Range Dependent Noise in Stochastic Approximation: A Finite-Time Analysis