Your paper timeline
Scroll AI takes the way you would scroll a great paper aggregator: quick signal first, deeper critique when something earns your attention, and challenges when a claim feels off.
797 papers in cs.LG
Trending mixes fresh papers with community signal.
0
cs.LGcs.CL Areeb Ahmad, Pratinav Seth, Vinay Kumar Sankarapu · Oct 1, 2026

Ablation studies in language models are confounded by self-repair, where downstream components appear to compensate for a removed part. This paper argues that such compensation is not dynamically recruited but is instead a standing, pre-existing opposition released by the intervention. The authors introduce a continuous dose axis for counterfactual interventions and show that downstream contributions follow an affine law predicted in part by static weights. If the finding generalizes, it would reshape how interpretability researchers infer component importance from ablations.

Ablate a component of a language model, and other components often appear to adjust and compensate. This phenomenon, termed self-repair, has been observed repeatedly, but its mechanism remains unclear. The most systematic study to date concluded that self-repair is noisy and unlikely to have a single explanation. We argue that it has one: a gain already present before any ablation. Any intervention on a causally important component can be viewed as a point on a coordinate axis $\lambda$, the signed strength of a counterfactual contrast. Hence, conventional ablation methods are uncalibrated points on this axis. We show that the causal repair response for a fine-grained unit $r$ is governed by an affine law, $E_r(\lambda)=\mathrm{own}_r+\gamma_r\lambda$. The slope $\gamma_r$ is a fixed coefficient that consistently influences the model, with or without ablation, and its sign determines whether the unit counteracts or reinforces the removed signal. On a factual-verdict task across four models from distinct families (Gemma, Qwen, LLaMA, and Mistral), we identify components including MLP neurons, OV neurons, and singular directions that follow this affine law, 68 of 81 downstream directions in all. Moreover, we can anticipate the magnitude of $\gamma_r$ from the fixed weights. On the IOI circuit of GPT-2 Small, seven of the ten heads the intervention can reach follow the law, and all seven are counterweights. From this perspective, what may appear as self-repair is a counterweight performing its usual operation when the contrastive signal emerges at the core.
0
cs.LG Scott W. Viteri (Stanford University), Laura Gomezjurado Gonzalez (Stanford University), Clark Barrett (Stanford University) · Oct 1, 2026

Intrinsic rewards in reinforcement learning aim to drive exploration by valuing experience, yet maximizing them need not yield the most informative data. This paper proposes comparing policies by counterfactual information: one policy dominates another if its interaction histories can be decoded to reproduce the other's history distributions uniformly across possible worlds. Using a simple alarm environment, the authors prove analytically that count-based, prediction-error, empowerment, and several state-based objectives can have strictly dominated optima, while also characterizing when posterior-based rewards succeed and constructing a native score $J_w$ that preserves every strict exploration improvement.

Intrinsic rewards are designed to guide exploration in reinforcement learning by assigning value to an agent's experience, for example through prediction error or learning progress. However, maximizing these rewards need not produce the most informative experience available. We propose a formal criterion for exploration that compares policies by the counterfactual information they acquire: how well their histories can substitute for experience under alternative policies. We construct a single, simple environment in which specified count-based, prediction-error, empowerment, and information-gain objectives have maximizing policies that are Pareto-suboptimal at acquiring counterfactual information. We explain these failures and establish conditions under which existing intrinsic rewards successfully encourage optimal exploration. We also construct an objective that assigns a higher value whenever exploration strictly improves under our criterion.
0
cs.LGmath.OCmath.PR Nikolaos Makras, Sotirios Sabanis · Oct 1, 2026

The paper tackles sampling from Gibbs distributions on matrix spaces with potentials that are neither convex nor globally gradient-Lipschitz. Its core idea is to replace the quadratic kinetic energy of underdamped Langevin dynamics with a non-quadratic, relativistic-like energy $K_\delta(p)=\delta^{-1}\operatorname{Tr}[(I_n+\delta pp^\top)^{1/2}-I_n]$, whose gradient tames momentum singular values smoothly. By carefully coupling a momentum-dependent diffusion matrix $M_\delta(p)$ and its divergence correction $\Gamma_\delta(p)$, the authors preserve the exact target Gibbs measure, recover regularised Muon in the zero-temperature limit, and prove exponential convergence of the diffusion plus uniform-in-time moment bounds for an explicit Euler-Maruyama scheme.

We consider the problem of sampling from Gibbs distributions on matrix spaces whose potential energies are neither convex nor globally gradient-Lipschitz. We introduce a family of non-quadratic kinetic energies that lead to a new underdamped Langevin system with momentum preconditioning, in which the gradient of the kinetic energy acts as a smooth spectral taming of the momentum. We prove that, under these relaxed assumptions on the potential, the resulting dynamics leaves the target Gibbs measure invariant, and we establish exponential convergence to equilibrium in a weighted total variation distance. Finally, we show that the corresponding Euler-Maruyama discretization admits moment bounds that are uniform in time, without any modification of the potential gradient, which ensures the stability of the resulting sampling algorithm.
0
cs.CLcs.AIcs.LG Lucheng Fu, Kejing Xia, Yiyang Wang et al. · Oct 1, 2026

Most LLM agents treat repeated use of the same external source as a sequence of independent retrieval events. This paper reframes the deeper problem as source learning: progressively building a persistent, reusable source model that captures how a source’s knowledge is structured, interpreted, and applied. The proposed SourceLearn system combines self-directed study with task-guided refinement, and the authors report large gains across document, code, API, and interactive benchmarks.

Large language model (LLM) agents increasingly rely on persistent external sources to solve sequences of knowledge-intensive tasks. Existing methods improve how source content is accessed and organized, while agent-memory systems preserve reusable knowledge from prior interactions, but repeated use of the same source is still largely treated as repeated access rather than an opportunity to progressively improve understanding of that source. We study source learning: developing reusable source-specific competence over a persistent authoritative source. We represent this competence with a persistent source model that captures reusable understanding of the source, including how its knowledge is structured, interpreted, and applied. To construct and progressively refine such models, we propose SourceLearn, which combines two complementary learning mechanisms. Self-Directed Source Learning identifies what remains incompletely understood and adaptively revisits the source, while Task-Guided Source Learning uses downstream experience to reveal local representational gaps and recurring needs in how source knowledge should be organized. In both cases, learning signals determine what should be reconsidered, while persistent updates are reconstructed from the authoritative source. Across five benchmarks and three LLM backends, SourceLearn achieves the best performance in 13 of 15 settings, with gains of up to 22.6 points over Hybrid RAG and substantial overall improvements over static source representations and experience-based memory baselines.
0
cs.LG Ali Janati, Nikita Kuzmin, Rohit Swamy et al. · Oct 1, 2026

This paper tackles competitive Super Smash Bros. Melee with a single Transformer policy that controls all 26 fighters from one checkpoint. The authors pretrain 10M- and 75M-parameter models on roughly 840k human replays, then apply rank-based supervised curricula, 75M-to-10M distillation, and Fox-only reinforcement learning. The resulting 10M policy wins 240 of 244 benchmark games against prior releases and runs within the 16.67 ms frame budget, while the authors openly acknowledge key confounds in evaluation.

We introduce Faynt, a family of 10M- and 75M-parameter Transformer policies for Super Smash Bros. Melee, each controlling all 26 characters with a single checkpoint. After reinforcement learning (RL), the 10M wins 240 of 244 same-character games (98.4%) against fourteen specialist and multi-character releases on their supported rosters, with a winning record against every release. These opponents retain 21- or 24-frame action delays; Faynt uses no added delay, and we have not isolated the effect of this difference. In a separate evaluation against a privately supplied zero-delay Slippi-AI model, the 10M wins all 68 games across two conditioning settings. We study architecture, optimization, scaling, and hyperparameter transfer to guide pretraining on approximately 840,000 human replays. Post-training combines rank- and outcome-based curricula, 75M-to-10M distillation, and RL restricted to Fox mirror matches. On the initial 152-game benchmark, the supervised 10M wins 69.7% of games, compared with 45.4% for the pretrained 75M, despite higher overall held-out controller-prediction loss. The weighted validation loss used for supervised checkpoint selection agrees with the win-rate ordering of all four pretrained and supervised policies. After supervised post-training, both models take less damage per minute, build larger early leads, and win more often after losing the first life. Optimized inference on recorded game states averages 5.2 ms per decision for the 10M and 8.7 ms for the 75M on an NVIDIA T4, excluding emulator execution and communication. We open-source the weights, both benchmark suites, and a platform for automated model tournaments.
0
cs.LGcs.AIcs.CL Aayush Karan, Sitan Chen, Yilun Du · Oct 1, 2026

The paper tackles how to teach LLMs new skills after pretraining without erasing old ones. While RL resists catastrophic forgetting by staying on-policy, it struggles when the model cannot generate successful trajectories on its own. The authors flip the problem: instead of changing the training loss, they use an MCMC projection sampler to reshape off-policy expert traces so they look more like samples from the base model while preserving correctness. This lets ordinary supervised finetuning match or beat RL and self-distillation on science, math, and medical tasks.

Introducing new capabilities to frontier models has long been the goal of posttraining, which predominantly employs supervised finetuning (SFT) and reinforcement learning (RL) to this end. Conventional wisdom dictates that RL enables strong generalization on new tasks without losing existing capabilities, while SFT is prone to weak generalization and catastrophic forgetting. At the same time, SFT can learn from off-policy expert data, whereas RL must rely on a model's ability to find successful trajectories with repeated sampling. In our work, we seek to leverage the strength of on-policy learning while utilizing the privileged information contained in off-policy data. However, rather than modifying the learning objective to accommodate this data, we instead tailor the data distribution to better suit the learner. We introduce a Markov chain Monte Carlo (MCMC) sampling algorithm that progressively transforms off-policy traces to be more on-policy given a reference model for finetuning. Across tasks like scientific skill acquisition, mathematical reasoning, and open-ended expertise, our sampling algorithm enables SFT to rival prevailing posttraining techniques, often generalizing better and forgetting less than strong on-policy baselines. In addition, the resulting finetuned models exhibit strong distributional performance and are capable of learning beyond sharpening the base model distribution. At a higher level, our approach presents sampling as a model-native operator that shapes data for learnability, offering broader utility as a general-purpose primitive throughout the posttraining stack.
0
cs.LGcs.DSmath.OC Han Zhong, Yinyu Ye · Oct 1, 2026

Robust Markov decision processes with polyhedral uncertainty lack exact convex representations because the natural robust Bellman inequalities become bilinear or union-of-polyhedra constraints. This paper shows how to encode a finite sequence of robust policy-iteration steps into a single exact LP whose optimal face recovers the robust optimal value and all optimal stationary randomized policies. At fixed discount, the LP has polynomial dimension and encoding length, and the authors use this representation together with a unified complexity analysis to prove strongly polynomial bounds for several standard uncertainty sets—including $\ell_1$, weighted $\ell_1$, interval, and Wasserstein—that previously lacked such guarantees.

We study linear programming (LP) representations and strongly polynomial algorithms for robust Markov decision processes (RMDPs) with rational polyhedral state-action rectangular uncertainty in rewards and transitions. By encoding a finite sequence of robust policy-iteration steps, we construct a single LP whose optimal solutions recover the robust optimal value and all optimal stationary randomized policies. At fixed discount, the LP has polynomial dimension and encoding length and can be constructed in strongly polynomial time. We also develop a general complexity analysis of robust policy iteration that combines the cost of minimizing over uncertainty sets with the number of iterations needed to evaluate a policy. For a fixed discount factor, we use this analysis to improve the known complexity bounds for $\ell_1$ and $\ell_\infty$ RMDPs and establish new strongly polynomial bounds for general interval, weighted $\ell_1$, and Wasserstein RMDPs, as well as turn-based stochastic games with these uncertainty sets.
0
cs.LGcs.CL Tido Specht, Elias Benedict Krey, Nils Neukirch et al. · Oct 1, 2026

This paper adapts Non-Linear Multi-Dimensional Concept Discovery (NLMCD) from vision to LLMs to model concepts as low-dimensional manifolds rather than linear directions. It introduces a fuzzy generalized Rand index called Concept-Based Alignment (CBA) that compares independently discovered concept spaces across layers, languages, and training stages without requiring shared tokenizers or architectures. The authors claim CBA is more sensitive than linear baselines (CKA, PCA), uses it to reveal cross-layer block structures, and applies it to study multilingual sharing, cross-model alignment, and post-training drift in Tulu-3.

Understanding information processing in large language models (LLMs) requires dissecting the geometric organization of their internal token representations. While existing mechanistic interpretability (MI) methods seek to extract concepts, they are constrained by a strong linearity assumption challenged by evidence of non-linear feature manifolds. We move beyond linear concepts by adapting Non-Linear Multi-Dimensional Concept Discovery (NLMCD) from computer vision to token-level LLM activations, modeling concepts as low-dimensional manifolds. To compare concept manifolds across layers and models, we introduce a concept-based alignment (CBA) score, a generalized Rand index that measures geometric proximity without explicit feature matching. Our analysis yields six key findings: (i) a neighboring-layer sanity check shows CBA is more sensitive than PCA- or CKA-based linear baselines; (ii) layer-by-layer alignment matrices reveal two block structures in intermediate and late layers, consistent across models and obscured by linear metrics; (iii) concept composition remains syntax-dominated through most of the network before giving way to increasingly mixed syntactic-semantic concepts in later layers, with increasing output-orientation toward the final layers; (iv) multilingual concept sharing between English and Mandarin is training-dependent rather than universal, strongest in Qwen, weaker in Llama, and absent in GPT-2; (v) inter-model alignment mirrors this structure, with strong correspondence between same-family Qwen models of different scale but weak alignment across model families; and (vi) across Tulu-3 training stages, alignment is highest between adjacent stages, with the largest shift between the base model and SFT, while subsequent preference-alignment stages (DPO, RLVR) leave early layers largely unchanged and RLVR mostly preserves DPO's concepts in late layers.
0
cs.LG Yifan Guo · Oct 1, 2026

When a machine-learning benchmark improves, multiple mechanisms could explain why. This paper studies how an autonomous agent should select diagnostic probes from a finite library to discriminate among competing explanations under a tight budget. MECHVAR scores each probe by the posterior-weighted variance of its predicted responses, showing that under a shared Gaussian predictive scale this is exactly proportional to the classical Box–Hill pairwise-KL criterion. The reformulation yields $O(KE)$ vectorized rescoring, an explicit pairwise audit trail, and orders-of-magnitude faster acquisition than quadrature-based expected information gain.

Benchmark gains are often mechanism-ambiguous: reproducing an improvement does not by itself identify why it occurs. We study finite-library mechanism discrimination, where posterior-weighted candidate mechanisms, executable probes, and a limited experimental budget define a sequential experiment-selection problem. MECHVAR selects the next probe by maximizing the posterior-weighted variance of its predicted responses. Under a shared-Gaussian predictive model, this score is exactly proportional to the classical Box--Hill posterior-weighted pairwise-KL criterion, yet it admits O(KE) vectorized rescoring and a transparent additive audit over mechanism pairs. A local expansion further links the score to expected information gain (EIG) when predicted response separations are small. In a 25-block stress audit, MECHVAR outperforms confirmation-first in several moderate misspecification regimes, while its primary comparisons with EIG remain statistically unresolved. In a held-out Digits loop, normalized mechanism-identification AUC is 0.8975 for MECHVAR, 0.7825 for a score-greedy policy, and 0.9092 for EIG. At K = 100, E = 200, median single-thread full-library scoring is 10.36 microseconds for MECHVAR versus 57.69 ms for six-node quadrature EIG in the recorded environment. MECHVAR therefore provides a lightweight, auditable acquisition rule for finite-library experiment selection when a shared predictive scale is a defensible approximation.
0
cs.LGcs.AI Junkang Liu · Oct 1, 2026

FedMIX-P tackles preconditioner coordination in federated learning by mixing a shared reference preconditioner with each client's local adaptive state at every local step. The core idea is to control the operator mismatch that arises when heterogeneous clients transform gradients differently, and the paper demonstrates empirical improvements on vision and language models under severe data heterogeneity. However, the convergence analysis is restricted to linear positive-definite preconditioners, while the headline experiments rely on nonlinear methods such as Muon, creating a notable gap between theory and practice.

Adaptive preconditioners accelerate model training, but heterogeneous client geometries can bias federated updates even when gradients are evaluated at the same model. Round-start synchronization alone cannot prevent this mismatch from reappearing during local training. We propose \texttt{FedMIX-P}, which mixes shared and local preconditioners at every local step, retaining local adaptation while reducing mean-squared operator mismatch by a factor of $\lambda^2$. For smooth nonconvex objectives with stochastic gradients and partial participation, we establish an $O(R^{-1/2})$ stationarity bound using suitable stepsizes and a horizon-dependent mixing weight, without requiring local preconditioners to converge to one another. A two-client counterexample shows that fixed positive mixing can preserve a nonstationary fixed point. The theory covers bounded linear symmetric positive-definite preconditioners. Experiments with SOAP, Sophia, and Muon variants across vision and language tasks show improvements over corresponding local optimizers, including accuracy gains of up to $19.47$ percentage points and lower validation loss for 60M--350M language models. Full nonlinear and momentum-based updates require separate analysis.
0
cs.LG Th\'eau d'Audiffret, Mariia Vladimirova, Jean-Yves Franceschi · Oct 1, 2026

Diffusion models amplify demographic biases present in training data, and current debiasing methods usually require costly attribute annotations, model retraining, or sacrifice output diversity. This paper proposes Debias Anything, a training-free guidance method that steers both unconditional and text-conditional diffusion models toward arbitrary target proportions for any sensitive attribute described in free text. By linking the frozen denoiser to a vision-language embedding space via a lightweight adapter, it jointly enforces batch-level fairness through optimal assignment and restores diversity through a semantic disagreement score, offering broad applicability without labeled data.

Although diffusion models produce high-quality images, they also reproduce and amplify demographic imbalances in their training data. Debiasing their generation process post-training w.r.t. some sensitive attribute usually relies on classifier guidance or explicit text extra-conditioning, but this reduces methods' applicability and output diversity. Conversely, methods promoting diversity alone do not ensure fair attribute representation. In this paper, we propose a method tackling fairness and diversity jointly that is generally applicable to any diffusion model and any sensitive attribute. To this end, an adapter connects the frozen diffusion model to a pretrained vision-language embedding space, enabling fairness and diversity guidance without sensitive-attribute annotations. For fairness, pairs of text prompts define attribute directions which guide batch composition towards specific proportions. For diversity, we introduce a score measuring disagreement between the semantic estimates derived from this representation. The formulation supports unconditional and text-conditional diffusion models, while requiring no prior knowledge or data of sensitive attribute. Experiments confirm that our method improves quality and diversity scores at comparable fairness levels.
0
cs.AIcs.LG Jude Waide, Robert Lieck · Oct 1, 2026

Long-term dependencies pose a fundamental challenge for sequential decision making because Transformers suffer quadratic attention scaling and RNNs lack expressive hidden states. This paper proposes Decision Titan (DTi), which grafts Titans-style Test-Time Training layers onto Decision Transformer to store episodic memories in fast weights updated by gradient descent at both train and test time. The authors introduce X-Maze, a controlled extension of T-Maze, and show that DTi can learn dependencies 20× longer than its context window and extrapolate to unseen episode lengths, while visualizing gate dynamics to explain how momentum propagates long-range information. Although the experiments are limited to synthetic memory tasks, the mechanistic ablations and honest characterization of catastrophic forgetting provide a valuable foundation for future research.

Long-term dependencies remain a major challenge for sequential decision-making in the field of AI: RNNs suffer from vanishing gradients and the limited expressivity of vector-based hidden states, whilst Transformer-based models are limited by the quadratic scaling of attention. Recent work has proposed tackling this problem with the Test-Time Training (TTT) framework, which stores episodic memories in the parameters of a neural network through gradient descent at both train and test-time. This approach has seen success in the domain of Natural Language Processing, however, to the best of our knowledge it has not yet been applied to the domain of Reinforcement Learning (RL), nor has there been a study analysing how this memory practically functions. In this paper, we study the potential of the TTT framework for offline RL by augmenting a Decision Transformer with TTT layers, dubbed the Decision Titan. We analyse performance and properties of the model in the X-Maze environment, an extension of T-Maze designed to test sequential memory, and investigate how the memory mechanism learns by visualising gate values over time. Our key findings are that Decision Titan can learn long-term dependencies with ranges 20x longer than the context window, generalises to lengths 1.7x the training data, but crucially temporal generalisation depends on the time embeddings used, and the ability to learn long-term dependencies depends on how the relevant information is encoded.
0
cs.LGmath.OCstat.ML Konstantinos Ziliaskopoulos, Alexander Vinel, Alice E. Smith · Oct 1, 2026

Contextual linear optimization faces a discontinuity: small prediction errors can flip the LP solution to a different vertex, making regret non-differentiable pointwise. This paper proves that, after averaging over the data distribution, the population regret is locally quadratic with curvature concentrated on the walls of the normal fan. The authors derive the exact curvature measure $K_0$, provide a tractable single-projection approximation $\mathsf{K}_{\lambda,s}$, and apply it to scenario generation, showing a $30.8\%$ regret reduction on a battery arbitrage task.

Decision-focused learning for linear optimization is complicated by the discontinuity of the optimizer, where small cost errors may leave the decision unchanged or move it to a different vertex. We show that this non-smooth pointwise behavior becomes locally quadratic after averaging over the data distribution, and we derive the curvature in closed form, specifically, a matrix-valued measure supported on the walls of the normal fan. This measure depends only on the feasible set, with the data distribution entering only as a weight. We then offer a tractable approximation for this curvature, computable with just one projection to the feasible set. We prove that the approximation weakly converges to the true population curvature. We offer one application of our findings, a decision-aware scenario generation method for expected-cost linear optimization. Our experiments test the quadratic and weak convergence laws and show a 30.8% regret improvement over uniform allocation on battery arbitrage.
0
cs.CVcs.LG Ahmed Sharshar, Asif Hanif, Naveen Kumar Kummari et al. · Oct 1, 2026

Medical imaging models often fail when deployed across scanners or hospitals because they overfit to domain-specific textures and appearance statistics. PhaseAT addresses this by adversarially perturbing the Fourier phase of the luminance channel while keeping the amplitude spectrum fixed, generating hard training views that stress spatial organization under matched style. By encouraging the model to ignore brittle amplitude-driven correlations and focus on stable geometric structure, PhaseAT achieves large gains on both single-source and multi-source domain generalization benchmarks.

Reliable clinical deployment of deep medical image models is hindered by distribution shifts across scanners, sites, and acquisition protocols. Existing domain generalization (DG) methods often focus on style or intensity diversification, but they can still leave networks dependent on domain-specific texture correlations. Inspired by evidence that Fourier phase encodes semantic structure, we introduce PhaseAT, a phase-aware adversarial training framework for medical DG. PhaseAT forms phase-perturbed training views in the Fourier domain by iteratively updating a bounded phase perturbation while keeping the amplitude spectrum unchanged, thereby stressing spatial organization under matched appearance statistics. Perturbations are applied only to the luminance channel in YCbCr color space to avoid chromatic artifacts. Additionally, a simple phase-saliency mask concentrates updates on the most influential frequencies. The model is trained with a weighted combination of losses on clean and phase-perturbed samples, supporting both single-source and multi-source DG. We validate our method on two challenging medical datasets and demonstrate that PhaseAT achieves over 20% improvement in single-source domain generalization, outperforming several state-of-the-art DG methods. The code implementation is available at: https://github.com/ahmed-sharshar/PhaseAT.
0
cs.LGcs.CLcs.CR Sahil Kadadekar · Oct 1, 2026

Can response safety be scored by cosine similarity to a safe-response embedding centroid? The paper shows that such a raw positive prototype identifies a location, not a separating direction, and formalizes why positive-only observations cannot determine the sign of the safe-unsafe gap. On three prompt-controlled safety corpora and four frozen encoders, the unreferenced prototype is weak or inverted while an explicit safe-minus-unsafe reference recovers useful ranking; the result is carefully scoped to the centroid rule, not all one-class methods.

Can response safety be scored by cosine similarity to the mean embedding of known-safe responses? A recent sleeper-agent detector proposes exactly this score, yet the raw positive-centroid rule is not identified: positive observations locate the safe class relative to an encoder origin, but do not determine which direction separates safe from unsafe responses. We audit the rule on two prompt-controlled, human-labeled corpora and one auxiliary jury-labeled source control, using four frozen encoders and prompt-grouped splits. On the human-labeled corpora the safe prototype reaches ROC-AUC 0.457-0.545, with two cells significantly below chance and one above, while an explicit safe-minus-unsafe reference reaches 0.588-0.738 on the same embeddings; on the jury control the prototype is inverted (0.358-0.405) and the reference reaches 0.754-0.793. At validation-calibrated 5% false-safe thresholds, the reference accepts more safe responses on PKU-SafeRLHF (0.153-0.263 versus 0.039-0.061 across encoders) and Aegis (0.189-0.291 versus 0.004-0.045), but not reliably on BeaverTails. A fully unlabeled held-out reference recovers part to most of the referenced ranking, much less when only 5% of the pool is unsafe, whereas 80-634 labeled unsafe responses recover most of it. Prompt-only ablations show that prompt-label composition can inflate uncontrolled evaluations. This is a bounded result about a raw positive centroid, not all one-class methods or safety-specialized guards. A class mean is a location, not necessarily a safety direction; a declared reference with enough unsafe mass identifies orientation.
0
cs.LGcs.SE Tianshuo Qiao, Naiqian Zheng, Xiaopeng Liu et al. · Oct 1, 2026

CI pipelines for LLM training frameworks are drowning in GPU-heavy end-to-end tests that can take hours per run. FastCI attacks this bottleneck by replacing coarse file-level mappings with a runtime evidence graph built from lightweight stack sampling, then pruning redundant tests via caller/callee context, scheduling risky tests first, and shrinking workloads along dimensions outside each test's validation scope. Deployed at ByteDance, it claims a 77.5% latency reduction and 63.9% GPU savings, though the evaluation rests on an internal codebase with no open-source artifacts.

As large language models (LLMs) keep growing in size and complexity, their training frameworks evolve at a rapid pace as well. Therefore, continuous integration (CI) is critical for maintaining the quality and stability of these frameworks. However, unlike traditional software, CI for LLM training frameworks relies on GPU-intensive tests, which usually involve complete model training or evaluation. This leads CI itself to become a new bottleneck for fast-paced development. In this paper, we introduce FastCI, a framework that improves the efficiency of CI for LLM training frameworks. FastCI leverages runtime evidence to select affected tests and prune tests that execute changed code in equivalent contexts. Then FastCI prioritizes high-risk tests to expose potential failures earlier, and optimizes test workloads along dimensions outside the intended validation scope of each test. Evaluated on the CI workload of our LLM training framework, FastCI reduces the CI latency by 77.5% and the GPU resource usage by 63.9%, while improving the modified code coverage retention by 3.2%, compared with the currently deployed CI pipelines. FastCI has now been integrated into the CI pipelines of our LLM training framework at ByteDance.
0
cs.LG Kuo Zhou, ZiXion Yang, Lu Zhang · Oct 1, 2026

SkillEvoLean tackles the challenge of improving LLM agents for Lean theorem proving without updating model parameters. The core idea is to evolve natural-language procedural skills through progressive refinement from execution trajectories and, crucially, a mutation operator that perturbs skills in a semantic concept space to escape local optima when all proof attempts fail. This matters because existing skill-evolution methods stagnate on hard theorems where no sampled trajectory succeeds, leaving the solver with no signal for improvement.

Skill evolution offers a promising way to improve large language model agents without updating their parameters, but its use in formal theorem proving remains underexplored. Existing methods mainly target natural-language reasoning, improving skills by analyzing successful and failed trajectories and incrementally revising solving strategies. Although the Lean verifier provides reliable execution feedback, when all sampled trajectories fail, existing skill evolution methods lack successful trajectories from which to infer effective update directions. Furthermore, these methods also focus mainly on the root instruction file, thus underexploring the evolution of reference knowledge including mathematical concepts and proving techniques. To address these limitations, we propose a mutation-enhanced skill self-evolution framework for building skill-augmented Lean provers. The framework jointly evolves a high-level solving policy and its reference knowledge through progressive and mutation-based updates. Progressive evolution derives local improvements from successful and failed trajectories, while mutation is triggered when no complete proof can be generated, sampling mathematical concepts to produce and select new skill candidates under verifier feedback. We evaluate our method on MiniF2F, PutnamBench, the 2025 International Mathematical Olympiad (IMO 2025), and the 2026 USA Mathematical Olympiad (USAMO 2026). Under the same backbone model, trajectorysampling budget, and test-time compute, our method achieves proof success rates of 100.0%, 90.6%, 4/6, and 4/6, respectively, with GPT-5.5, outperforming the baseline methods. Further analysis shows that concept-guided mutation outperforms random-text-guided mutation by 6.9 and 8.2 percentage points on MiniF2F and PutnamBench, respectively, while solving one additional problem on both IMO 2025 and USAMO 2026.
0
cs.AIcs.LG Changdae Oh, Qi Zeng, Qi Qi et al. · Oct 1, 2026

Does RL post-training create new agentic capabilities or just redistribute probability mass toward high-reward behaviors that a base model already possesses? This paper shows that, across 14 base/post-trained checkpoint pairs and three process-centric agentic benchmarks, pre-trained base models equipped with a lightweight harness often surpass their post-trained counterparts in solution coverage (pass@$K$) given enough test-time compute. To diagnose this coverage-accuracy trade-off, the authors propose Sharpening Tax, a single-number metric for the loss in test-time scalability after post-training, and present Posterior-Tempered Group Sampling (PTGS), a Bayesian sampler that mitigates the tax during RL training.

An emerging hypothesis about reinforcement learning (RL) post-training of large language models (LLMs) is that it merely sharpens existing behaviors of a base model, improving single-shot accuracy at the cost of solution coverage. Although this trade-off has been observed in math and coding tasks, it need not extend to agentic tasks, where multi-turn tool use and interaction may require capabilities newly acquired during post-training. Our surprising finding is that pre-trained LLMs, equipped with a light inference harness, can serve as capable agents. Despite far lower accuracy (pass@1), they often surpass their post-trained counterparts in solution coverage (pass@K) given a sufficient test-time budget. We further analyze the underlying mechanism and show that post-training pushes tasks toward two extremes, always solved or never solved, and thereby improves sampling efficiency and consistency at the cost of solution coverage. To measure this cost, we propose Sharpening Tax, a diagnostic metric that quantifies the loss in test-time scalability after post-training. Across 14 base/post-trained model pairs from four families and three agentic benchmarks (42 cases in total), the tax is prevalent in most settings, can be estimated from a few rollouts, and correlates well with other metrics. Finally, we present posterior-tempered group sampling (PTGS), a simple plug-and-play Bayesian sampler that adapts the sampling temperature per prompt to its estimated difficulty. Applied during RL training in two agentic environments, PTGS pays a smaller tax than the fixed-temperature baseline, solving more tasks under repeated sampling while also improving single-shot accuracy.
0
cs.AIcs.CLcs.LG Manar Aljohani, Brandon Ho, Kenneth McKinley et al. · Oct 1, 2026

Emergency Department triage assigns acuity scores that can be skewed by non-clinical patient descriptors. This paper asks whether open-source LLMs change their pediatric Emergency Severity Index predictions when only demographic, socioeconomic, or contextual variables are injected into otherwise identical clinical vignettes. The authors audit ten models and find that a small fine-tuned Qwen2.5-7B is the most counterfactually stable, challenging the assumption that larger or medically pretrained models are necessarily fairer. The study matters because locally deployable open-source LLMs are increasingly considered for privacy-preserving clinical decision support, yet their fairness profiles across model families were poorly understood.

Emergency department (ED) triage is a high-stakes prioritization task in which demographic, socioeconomic, and system-context information may improperly influence acuity assignment. Although open-source large language models (LLMs) are increasingly considered for local and privacy-preserving clinical decision support, it remains unclear how counterfactual bias varies across model families, sizes, medical-domain models, and domain-adapted models. We present a comparative counterfactual audit of ten open-source LLMs for pediatric Emergency Severity Index (ESI) prediction. Starting from real and handbook-style clinical vignettes, we construct paired counterfactual variants that change only one injected demographic, socioeconomic, healthcare-access, behavioral, social, or system-context variable while holding the clinical presentation fixed. Models include Qwen2.5-7B, Qwen2.5-14B-Instruct, a QLoRA fine-tuned Qwen2.5-7B, MedGemma variants, MedLLaMA2-7B, GPT-OSS-20B, and GPT-OSS-120B. We measure any counterfactual shift, undertriage, overtriage, shifts greater than one ESI level, mean shift, and mean absolute shift. Counterfactual sensitivity varied substantially and did not consistently decrease with larger model size or medical-domain pretraining. The fine-tuned Qwen2.5-7B showed the lowest overall sensitivity, with a 5.27% any-shift rate and mean absolute shift of 0.0534, versus 16.02% and 0.1706 for the base model. Several larger or medical-domain models showed more significant shifts. Stratified and correlation analyses further revealed clinically important directionality and shared failure patterns hidden by aggregate rates. These findings support counterfactual auditing as a lightweight, clinically interpretable framework for comparing fairness risks in open-source LLMs before clinical deployment.
0
cs.LGcs.CV Si Qi Goh, Cap Dang Xuan Kiet, Tat-Jen Cham et al. · Oct 1, 2026

SIEVE tackles selective unlearning in vision-language models (VLMs), where a model must forget specific sensitive image-question-answer associations about an individual while retaining permitted facts about the same person. The core idea is to suppress attention-value representations for forget examples toward zero and anchor retain-example values to a frozen reference model, coupled with sequence-level negative cross-entropy. This matters because VLMs can recover biographical information from faces, creating privacy risks, and prior methods struggle to disentangle closely related forgotten and retained knowledge without degrading utility.

The ability of vision-language models (VLMs) to associate visual identities with biographical information creates a need for selective unlearning of personally identifiable information (PII) while preserving permitted knowledge about the same individual. This setting is challenging because both sensitive and retained information can share the same visual inputs and intermediate representations. We introduce SIEVE, a simple and effective framework for selective VLM unlearning. SIEVE directly regularizes attention-value representations while also controlling model outputs. SIEVE suppresses attention values for forget examples toward a constant zero, while preserving retain-example representations by matching them to a frozen reference model. These objectives are combined with sequence-level forget and retain supervision, enabling targeted forgetting without largely affecting retained knowledge. Extensive experiments show that SIEVE achieves state-of-the-art performance on unlearning with multiple model-modality settings, while maintaining competitive retained utility. Ablation studies further show that value suppression and negative cross-entropy contribute complementary forgetting signals, while reference-based value matching substantially reduces utility degradation. These results demonstrate that attention values provide an effective intervention point for selective multimodal unlearning when sensitive and retained knowledge are closely related.