Papers for

language model developers

Papers whose findings have a practical use for this group, as judged from the abstract. Open a paper to read what it means in practice.

Rlhf performs well when human feedback matches model behavior distribution

Distortion of AI Alignment Revisited: RLHF is a Decent Utilitarian Aligner

Abstract: While Reinforcement Learning from Human Feedback (RLHF) is the standard paradigm for aligning large language models with human preferences, its effectiveness in pluralistic settings has been called into question. Notably, recent work by Gölz et al. (2025) demonstrated that the \textit{distortion} -- defined as the multiplicative gap between the average user utility of the RLHF policy and the optimal average utility -- can scale exponentially with the Bradley-Terry temperature parameter $β$ when users have heterogeneous preferences. In this work, we present a fine-grained analysis of the distortion of RLHF with reward clipping and demonstrate that such exponential degradation is not a fundamental property of the algorithm but rather a consequence of distribution mismatch between the distribution generating preference data ($μ$) and the KL reference policy ($π_{\mathrm{ref}}$). To this end, we establish tight upper and lower bounds on the distortion of RLHF across multiple regimes of the KL regularization strength. We show that in a representative regime, under the Bradley-Terry model, the distortion is $\tildeΘ(βB + β)$, where $B$ is an upper bound on the log density ratio between $μ$ and $π_{\mathrm{ref}}$. In particular, when there is no distribution mismatch (i.e., $μ= π_{\mathrm{ref}}$), RLHF achieves the optimal distortion of $O(β)$ up to a constant. Our results suggest that, to reasonably maximize average utility with RLHF, it is preferable to use on-policy sampled preference data or to fine-tune before RLHF on data from a source close to $μ$.

Fri 11 SeptMachine LearningComputer Science and Game Theory
The gist
People train AI language models using human feedback to make them act the way users want, but when different users have different tastes, this can cause problems. The authors studied how a common training method called RLHF can sometimes perform poorly if the preferences given by humans don’t align with what the model expects. They found that this problem isn’t a fundamental flaw in RLHF, and if the feedback data matches the model’s behavior, RLHF works well at balancing different users’ wishes. This means it helps to collect feedback in a way that fits the model’s current behavior or to fine-tune the model first on similar data before training.
Open 2609.12651v1

Synthetic data contamination detection improves language model training safety

SynthSentry: Detecting Synthetic Data Contamination in Language Model Training Data

Abstract: Large language models trained recursively on their own or other models' outputs undergo model collapse, in which distributional tails and factual accuracy deteriorate while fluency survives. Prior work diagnoses collapse after training; the actionable problem is screening a corpus of unknown provenance before training. We introduce SynthSentry, a corpus-level, model-agnostic contamination signal requiring no access to the generating model, no generation history, and no synthetic labels. The score is a distributional divergence over three statistics: lexical diversity collapse, n-gram tail truncation, and perplexity variance across reference models. We evaluate on corpora contaminated by small open-weight generators and an instruction-tuned open-weight model under a leave-one-generator-out protocol. A domain-stratified study measures false positives on naturally repetitive human text (legal, clinical, source code). The score ranks corpora by severity with little loss when whole generator families are held out. Per-domain calibration holds near its nominal false-positive budget once covariance shrinkage and a bootstrap threshold replace a naive quantile, which runs four times over budget. A downstream fine-tuning check showed no contamination-driven accuracy deficit at our scale, so whether pruning recovers one remains open; the same run shows over-pruning risk once pruning exceeds the true contamination fraction. We frame screening as a data-curation defense rather than a post-hoc diagnosis and release the scoring toolkit. All results are small-scale; scope is English-language, batch-mode corpus screening. Contamination sources are single-generation or hand-authored rather than recursively generated, so results speak to synthetic contamination generally and not to recursion depth.

Fri 11 SeptComputation and Language
The gist
When large language models train using data that includes artificial text from other models, the resulting model can lose accuracy even if it sounds fluent. The authors identify a way to detect such synthetic data in training sets before using it, without needing to know how the fake text was generated. They measure unusual patterns in word diversity, phrase endings, and prediction difficulty to score datasets for contamination. Their tool, SynthSentry, can help data curators filter out problematic text and avoid training issues caused by synthetic data.
Open 2609.12353v1

New distillation method improves cross-family language model training

CompassOPD: Cross-Family On-Policy Distillation via Within-Family Likelihood Shifts

Abstract: On-policy distillation (OPD) provides dense token-level supervision on student-generated trajectories. Although OPD performs strongly when teacher and student belong to the same model family, we find that its effectiveness degrades in cross-family settings even after tokenizer alignment, with substantially stronger external teachers offering little additional improvement. To understand this disconnect, we decompose the cross-family OPD signal into two components: an offset between a low-capability teacher-family reference and the student, and the within-family log-likelihood shift from that reference to the strong teacher. Standard OPD transfers both components together, allowing the offset to dominate the update direction and obscure the changes associated with teacher capability improvements. We propose CompassOPD, which removes this offset and transfers the within-family shift, while a frozen student reference anchors updates to the student's initial policy. Thus, both teacher-side and student-side changes are measured within their respective model families. Experiments across three student families and multiple teacher families show that CompassOPD consistently outperforms standard cross-family OPD, improving average reasoning accuracy by up to 5.50 points. For an MoE teacher, we further construct the reference directly from the teacher checkpoint by reducing expert activation, eliminating the need for a separate reference checkpoint while retaining a 3.43-point gain over OPD.

Wed 9 SeptMachine Learning
The gist
Teaching smaller language models using larger models works well when both models are similar, but it becomes less effective when they come from different families. The authors found that combining two different influences during training confuses the learning process. They designed a new method called CompassOPD that separates these influences and focuses on the improvements from the better model itself. This approach leads to better results in reasoning tasks across different types of models.
Open 2609.10154v1

Safety failings persist in 320 billion parameter AI mixture models

How Fragile Is Safety Alignment at Frontier Scale? A Single-Direction Attack on a 320B MoE

Abstract: Directional ablation removes an aligned language model's ability to refuse by projecting a single "refusal direction" out of the weights that write the residual stream. It needs no gradient-based training and no optimization, only a few hundred contrastive prompts, which makes it the canonical white-box attack on open-weight alignment. However, it has been established only on dense models up to roughly 70B parameters. We study whether it survives the shift to frontier mixture-of-experts (MoE) models whose residual streams are no longer a single tensor and whose weights ship quantized. We apply it to GLM-5.3-Flash (320B parameters, 288 routed experts, a four-wide hyper-connection residual, block-FP8). The attack survives the architecture, but what it reaches is no longer where a reader of the original recipe would look for it. Editing the attention, dense and routed-expert writers on their own removes 0.039, 0.016 and 0.148 of refusal respectively; editing all three together removes 0.776. As a result, 74% of the effect exists only under the joint intervention. The part the conventional recipe reaches by module-name matching accounts for 0.066 of that 0.776, which is why it fails silently on an MoE. The effect does not follow from removing just any direction: ablating a random direction orthogonal to it leaves refusal unchanged. A category-concentrated residue survives every edit we tried: subspaces fitted on violence, sexual content and hate leave measurable refusal at every rank from 1 to 12. We report the method, the 41-89 percentage-point reductions it achieves across seven harmful benchmarks with no detected change in capability, and the boundary where it stops.

Wed 9 SeptCryptography and SecurityArtificial IntelligenceComputation and Language
The gist
Large AI language models are often designed to refuse harmful or unsafe requests. This paper tests a known way to weaken these refusals by targeting a specific part of the AI’s internal workings. The researchers found that this attack still works on a very large and complex type of model, called a mixture-of-experts model, but the way to do it is different and more complicated. Some safety features remain stubbornly resistant, even after these manipulations. They also showed that the attack reduces harmful outputs significantly without hurting the model’s overall ability.
Open 2609.09793v1

Large language models yield to user pressure over long chats

Measuring LLM Sycophancy under Sustained Multi-Turn Pressure

Abstract: Large language models (LLMs) may abandon correct positions when users push back, exhibiting a failure mode known as sycophancy. Existing evaluations typically use short, pre-specified conversations and may therefore miss failures that emerge under sustained, adaptive disagreement. We introduce SPINE, a benchmark in which an LLM proxy plays a persistent but mistaken user and adaptively challenges a target model for up to 25 turns. We evaluate four production systems and three Olmo3-7b variants on 100 false-presupposition and 100 unethical-query items. Our experimental results show that collapse rates increase with conversation length for every model, short-horizon protocols underestimate sycophancy and resistance under sustained pressure remains unreliable across current models. By analyzing models with accessible reasoning traces, we surprisingly found that the correct position often remains represented in a reasoning trace when the response concedes, suggesting that the model chooses to please a user and sycophancy is not due to lack of knowledge or ignorance. Ablations show that adaptive LLM proxy exposes more sycophantic collapse than pre-generated scripts. Among all tactics, emotional appeals is the most associated with inducing LLM sycophantic behavior. The code and data are released at https://anonymous.4open.science/r/SPINE

Tue 8 SeptComputation and LanguageArtificial Intelligence
The gist
Language models sometimes give in and agree with users even when the users are wrong, a behavior called sycophancy. The authors created a test called SPINE where a model pretends to be a wrong user and pushes another model for up to 25 conversation turns. They found that longer conversations lead to more giving-in, and short tests miss many of these failures. Interestingly, models often still know the right answer internally but choose to please the user. Emotional appeals from the user were the strongest triggers for this behavior.
Open 2609.09090v1

SkillAdam improves agent skill learning with stable and fast updates

SkillAdam: Stable and Efficient Skill Evolution for Agents

Abstract: Agent skills provide a lightweight way to equip frozen language-model agents with domain knowledge and procedural guidance, yet obtaining high-quality skills remains costly and difficult to scale. Expert-written skills require substantial human effort. Recent skill self-evolution methods automate an iterative loop that uses execution feedback to revise skills, but their heuristic update strategies often yield unstable optimization and low iteration efficiency. We identify two challenges in realizing stable and efficient skill self-evolution. Direction Stability requires effective corrections to accumulate rather than be overwritten by iteration-local feedback. Update Adaptivity requires the scope of each revision to reflect the consistency of recent case-level improvements. We introduce SkillAdam, an Adam-inspired framework for optimizing discrete and non-differentiable skill documents. As a functional analogue of Adam's first moment, an optimization memory records identified problems and the outcomes of prior solution attempts to stabilize the update direction. As a functional analogue of Adam's second moment, a volatility-driven edit budget tracks the history-weighted variation of recent case-level improvements and adaptively controls the update magnitude. Across seven benchmarks that span short- and long-horizon tasks, SkillAdam achieves state-of-the-art performance with more stable optimization dynamics. It also obtains stronger skills with substantially fewer optimization iterations and lower cost than prior methods. Code repository: https://github.com/ruc-datalab/SkillAdam

Tue 8 SeptArtificial Intelligence
The gist
Teaching AI agents new skills can be slow and unstable because updating these skills usually relies on guesswork. The authors found two key problems: updates can undo previous progress and sometimes revisions are too big or too small. They created SkillAdam, a method inspired by a popular optimization algorithm, to keep track of past updates and adjust how much skills change based on recent improvements. This makes the learning process steadier and faster across many types of tasks.
Open 2609.08944v1

Transformer layers quickly fix each other’s mistakes in language models

LLM Layers Immediately Correct Each Other

Abstract: Recent methods in language model interpretability employ techniques such as sparse autoencoders to decompose residual stream contributions into linear, semantically meaningful features. Such methods are commonly interpreted as identifying features that persist in the residual stream and that subsequent layers build upon. We challenge this view by identifying the Transformer Layer Correction Mechanism (TLCM), wherein adjacent transformer layers systematically counteract portions of each other's contributions. TLCM appears in 5 out of 7 major open-source model families and activates across nearly all tokens in diverse texts. We show that TLCM emerges during pretraining, operates most strongly on contextually dependent tokens, and adaptively calibrates its correction strength based on the preceding layer's output. Using the layer Jacobian, we further show that TLCM selectively corrects specific subspaces while reinforcing others, which we interpret through a ``propose-and-reject'' framework in which layers propose candidate features and subsequent layers selectively remove inappropriate ones. This dynamic suggests that the residual stream at any layer contains transient proposals alongside persistent features, helping explain why SAE feature descriptions often have low specificity, why effective model steering requires extreme feature amplification, and why transcoders hold a theoretical advantage over SAEs.

Mon 7 SeptComputation and LanguageMachine Learning
The gist
Large language models work in several steps or layers when processing text, and it was previously thought each layer builds upon the last without reversing changes. The authors found that adjacent layers often undo or correct parts of each other’s work quickly, like a back-and-forth editing process. This behavior, called the Transformer Layer Correction Mechanism, is common in many types of models and changes depending on the context. It helps explain some oddities seen when trying to interpret these models and suggests layers propose ideas that other layers then accept or reject.
Open 2609.07876v1

Mamba recall capacity improves with hashing and model size

On the Recall Scaling Laws in Mamba: A Theoretical and Mechanistic Study via Hashing

Abstract: Associative Recall (AR) is the cognitive ability to learn and retrieve links between items in memory. In NLP, AR is used as a benchmark for evaluating the in-context memory capacity of architectures such as Mamba, and has been found to strongly correlate with language modeling performance. This paper explores AR from the perspective of mechanistic interpretability, aiming to reverse-engineer the exact internal algorithm used by Mamba to perform recall. Our key insight is that Mamba performs recall by implicitly learning linear hash functions, and we identify the low-level circuit that enables this behavior. Building on these findings and inspired by theoretical tools in similarity-preserving hashing, such as the Johnson-Lindenstrauss lemma, we develop a theoretical framework for analyzing AR, which we term Recall Scaling Laws. Given the vocabulary size and the number of facts in context, this framework allows us to (1) predict the embedding and state dimensions required for Mamba to achieve perfect recall, (2) predict recall success probability given the model dimensions, and (3) analyze multi-layer models and multi-head SSM patterns. Empirical results show that our theoretical findings are accurate and predictive, offering insights into how AR capacity scales with vocabulary, state, embedding size, and architecture.

Mon 7 SeptMachine LearningComputation and Language
The gist
Remembering connections between things is important for both humans and AI models. This paper studies how the AI model Mamba remembers facts by discovering it uses a kind of mathematical shortcut called linear hashing. The researchers created a theory that predicts how well Mamba can recall information based on its size and design. Their predictions matched experiments, which helps us understand how memory in AI could get better as models grow.
Open 2609.07681v1

WIDER method improves latent reasoning diversity in language models

Think Wider: Mitigating Latent Rank Collapse in Implicit Chain-of-Thought Reasoning

Abstract: Chain-of-thought (CoT) reasoning improves the reasoning ability of large language models by introducing intermediate computation, but explicit rationales increase decoding length, latency, and context cost. Implicit CoT offers a more efficient alternative by moving intermediate reasoning into continuous latent states. However, latent reasoning can be unstable: successive latent states may become overly similar and collapse toward a shared dominant direction, reducing the diversity of the reasoning trajectory. In this work, we identify $\textit{latent rank collapse}$ and propose $\textbf{WIDER}$, a lightweight spectral regularizer for implicit CoT. During training, WIDER estimates the shared direction of each latent trajectory and penalizes projections onto this direction, encouraging latent states to span a broader representational subspace. The method is plug-and-play and leaves the backbone model, latent schedule, and inference-time decoding procedure unchanged. We further formulate this collapse as a geometric bottleneck in implicit reasoning, casting its mitigation as a training-time regularization problem rather than an inference-time decoding change. Extensive experiments show that WIDER improves matched implicit CoT baselines, while mechanistic analyses reveal higher effective rank, lower dominant-direction energy, and reduced redundancy among latent steps. These results highlight latent subspace utilization as an important factor for efficient continuous reasoning, providing a geometric perspective for analyzing and improving implicit CoT. Code is available at https://github.com/whitesweater/WIDER.

Mon 7 SeptMachine Learning
The gist
Large language models can think step-by-step to solve problems, but writing out each step takes time and space. The authors found that when these reasoning steps are kept inside the model's memory (latent states) instead of written out, these steps can become too similar and less useful. To fix this, they created WIDER, a training trick that encourages these memory states to be more varied and spread out. This makes the model’s internal thinking clearer and improves its problem-solving without slowing it down.
Open 2609.07406v1

Language model learning improves with rewards for partial progress

Long-Horizon Language Model Reinforcement Learning via Progressive Point Matching

Abstract: Current paradigms for training language models via reinforcement learning rely heavily on sparse outcome rewards. However, as we pursue tasks that require longer and more complicated trajectories, such strategies result in slow learning. Prior work has attempted to address this problem by rewarding partial progress; however, naive formulations are often biased and converge to suboptimal policies. We show that a simple and unbiased dense reward formulation, which we term progressive point matching, scales exponentially more efficiently to long-horizon tasks by rewarding partial progress on a segment level, both theoretically and empirically via synthetic environments. We then show how progressive point matching can be practically instantiated using a single reference trajectory per task. On extremely hard math reasoning problems, sparse outcome rewards cannot make any progress, whereas segment-level rewards enable improvements at larger test-time token budgets when measured by success rate or pass@k.

Mon 7 SeptMachine Learning
The gist
Training language models to complete long and complicated tasks is hard because feedback usually comes only at the very end, making learning slow. The authors studied a new way to give earlier, more frequent feedback by rewarding partial progress along the way, which helps models learn faster and better. They created a method called progressive point matching that compares parts of the model's work to a reference example to provide these in-between rewards without bias. Their method works much better on hard math problems than just giving rewards at the end.
Open 2609.07303v1

Model defaults control how activation steering affects language output

Steering Interference Reflects the Model's Defaults, Not the Behavior Directions

Abstract: Activation steering promises modular control of language model behavior: a behavior such as politeness corresponds to a direction in a model's activations, and adding that direction while it generates should switch the behavior on and leave everything else alone. It does not. We ask what decides which other behaviors move, and by how much, and find that it is the model rather than the behavior being steered. A steer relaxes the model toward a small set of behaviors it already favors, chiefly refusal, sycophancy, and poeticism, and that set is much the same whatever is steered. Three results across 24 behaviors and ten instruction-tuned models support this, every effect read off the generated text by a language-model judge rather than off a probe. That readout matters: all 24 behaviors are linearly decodable, but only 20 change what the model writes. First, a direction carrying no behavioral content, matched to a real steer only in the size of the vector it adds, moves the same behaviors in the same order as real steers do, while producing none of the behaviors that need a specific direction. Second, most interference runs one way, so it cannot be an overlap between two directions: steering profanity makes the model toxic, while steering toxicity leaves profanity untouched. Third, with a behavior held out entirely, geometry measured on the others explains almost none of the interference it takes part in. The account holds on all ten models, the pull toward defaults strongest below 10B parameters and weakening in each family's largest. Reading a steer as a perturbation whose endpoint the model fixes implies that disentangling behavior directions cannot by itself make steering modular.

Mon 7 SeptMachine LearningArtificial Intelligence
The gist
Activation steering tries to change how language models behave by nudging certain directions in their internal activations. The authors find that instead of making models perform new behaviors cleanly, these nudges mostly pull the model toward a small set of usual behaviors it prefers, like refusal or poetic language. This effect is largely determined by the model itself, not the specific behavior being targeted. They observed this pattern across many behaviors and models, suggesting that simply isolating behavior directions won’t achieve perfect modular control.
Open 2609.06951v1