mlsanityData to 14 Oct 2026

Trust Your Guide Only When Certain: Uncertainty-Aware Sparse Alignment at Inference Time

arXiv 2609.00624 ↗First posted 1 Sep 2026cs.CLcited by 0

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

Abstract (arXiv)

A prominent paradigm in inference-time alignment employs lightweight supervisors to steer Large Language Models (LLMs). Through empirical analysis, we identify a structural mismatch in this paradigm: weak supervisors exhibit pervasive high entropy across the vast majority of tokens, yet prevailing dense intervention approaches mandate supervision at every decoding step. This leads to frequent low-confidence interventions that can disrupt valid base-model reasoning and incur substantial utility costs. To resolve this, we propose TUSA (Trust-based Uncertainty Sparse Alignment). Moving away from continuous oversight, TUSA reframes alignment as a dynamic arbitration process, introducing an uncertainty-aware arbiter that authorizes intervention only when two conditions are met: the supervisor is confident and the token is semantically salient. This mechanism effectively filters out uncertainty-driven noise and redundant supervision. Extensive experiments across multiple models and benchmarks show that TUSA consistently improves both safety alignment and general helpfulness. By bypassing approximately 50% of alignment steps, it not only enhances safety preference by up to 15.6%, but also boosts general preference rates by up to 12.0% compared to the dense baseline, demonstrating that selective, high-precision alignment can outperform continuous supervision.

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 28 of its 41 references that are in the map. First posted on arXiv 1 Sep 2026.

Builds on

28 of 41 references.

Closest ancestors · 10

Same direction, newest first

  1. 2025ConfPO: Exploiting Policy Model Confidence for Critical Token Selection in Preference Optimization10 Jun
  2. 2025Inference-time Alignment in Continuous Space26 May
  3. 2025Token-level Accept or Reject: A Micro Alignment Approach for Large Language Models26 May
  4. 2025PARM: Multi-Objective Test-Time Alignment via Preference-Aware Autoregressive Reward Model6 May
  5. 2024Weak-to-Strong Preference Optimization: Stealing Reward from Weak Aligned Model24 Oct
  6. 2024GenARM: Reward Guided Generation with Autoregressive Reward Model for Test-time Alignment10 Oct
  7. 2024Decoding-Time Language Model Alignment with Multiple Objectives27 Jun
  8. 2024Aligner: Efficient Alignment by Learning to Correct4 Feb
  9. 2024ARGS: Alignment as Reward-Guided Search23 Jan
  10. 2022Training language models to follow instructions with human feedback4 Mar

Other references · 18

  1. 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 8,065
  2. 2017On Calibration of Modern Neural Networkscited by 4,960
  3. 2018Soft Actor-Critic: Off-Policy Maximum Entropy Deep Reinforcement Learning with a Stochastic Actorcited by 4,885
  4. 2017Deep reinforcement learning from human preferencescited by 3,912
  5. 2022Language Models (Mostly) Know What They Knowcited by 1,856
  6. 2025gpt-oss-120b & gpt-oss-20b Model Cardcited by 1,386
  7. 2023Representation Engineering: A Top-Down Approach to AI Transparencycited by 1,336
  8. 2023Inference-Time Intervention: Eliciting Truthful Answers from a Language Modelcited by 1,302
  9. 2023Llama Guard: LLM-based Input-Output Safeguard for Human-AI Conversationscited by 1,207
  10. 2023Steering Language Models With Activation Engineeringcited by 922
  11. 2023BeaverTails: Towards Improved Safety Alignment of LLM via a Human-Preference Datasetcited by 860
  12. 2019Plug and Play Language Models: A Simple Approach to Controlled Text Generationcited by 799
6 more
  1. 2022Contrastive Decoding: Open-ended Text Generation as Optimizationcited by 609
  2. 2023Weak-to-Strong Generalization: Eliciting Strong Capabilities With Weak Supervisioncited by 467
  3. 2021DExperts: Decoding-Time Controlled Text Generation with Experts and Anti-Expertscited by 441
  4. 2024PKU-SafeRLHF: Towards Multi-Level Safety Alignment for LLMs with Human Preferencecited by 261
  5. 2023Red-Teaming Large Language Models using Chain of Utterances for Safety-Alignmentcited by 223
  6. 2025Token-Level Density-Based Uncertainty Quantification Methods for Eliciting Truthfulness of Large Language Modelscited by 18

Built on by

Not cited yet.

Similar, not cited

Close in wording, no citation either way

  1. 2025On Almost Surely Safe Alignment of Large Language Models at Inference-Time
  2. 2026Sensitivity Uncertainty Alignment in Large Language Models ↗
  3. 2025Trusted Uncertainty in Large Language Models: A Unified Framework for Confidence Calibration and Risk-Controlled Refusal
  4. 2026Inference-time Alignment via Sparse Junction Steering
  5. 2026Deeper is Not Always Better: Mitigating the Alignment Tax via Confident Layer Decoding
  6. 2024Uncertainty Aware Learning for Language Model Alignment
  7. 2025An Uncertainty-Driven Adaptive Self-Alignment Framework for Large Language Models
  8. 2025Safety Alignment Depth in Large Language Models: A Markov Chain Perspective