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.AI Junze Zhu, Weihao Chen, Zhiyang Zhou et al. · Oct 14, 2026

Deploying LLMs as agents requires respecting runtime-specific execution contracts, yet standard training leaves harness semantics implicit in action trajectories. This paper introduces Harness Policy Distillation (HPD), which replays reference trajectories through the target harness to compile control-state documents, uses harness verification to filter teacher actions, and jointly trains a shared model on native action prediction, document-conditioned action prediction, and document reconstruction. On TRAJECT-Bench with the Pi harness, HPD lifts Qwen3-8B task success from 3.09% to 22.98% and substantially outperforms supervised, RL, and standard distillation baselines.

Deploying a general-purpose model as an agent means running it inside a product-specific runtime harness that decides which actions are executable, how state advances, and when the loop may terminate. Ordinary trajectory supervision shows which action was taken but leaves the runtime semantics behind that action implicit. We introduce Harness Policy Distillation (HPD), which makes these runtime control semantics an explicit training target alongside actions to align the harness policy. HPD replays reference trajectories through the harness to recover the runtime semantics that are implicit in ordinary trajectory data, and represents them as control documents at each model-call boundary. These documents provide privileged information for a teacher to generate actions, while harness verification ensures that the resulting targets are executable under the runtime contract. We jointly train a shared model on the runtime semantics recovered from control-state privileged information and native history, thereby internalizing these semantics into the model policy. At deployment, the trained model uses the original harness interface without requiring the control document or any additional supplement. On TRAJECT-Bench with the Pi harness, Qwen3-8B trained with HPD reaches 22.98% autonomous task success and 49.23% runtime compliance, against 3.09% and 19.90% for the unadapted model. Harness-verified teacher actions substantially improve the native deployment policy, while joint control-state supervision provides further gains in task completion and runtime compliance. HPD turns executable runtime semantics into dense supervision for learning harness-compatible actions at each decision boundary.
0
cs.LGcs.AI Kintan Saha, Ahana Deb, Anders Jonsson et al. · Oct 14, 2026

Average-reward reinforcement learning with options models continuing hierarchical tasks, but before this paper there were no finite-time convergence guarantees for option-based Q-learning in that setting. The authors prove that multi-step (J-step) inter- and intra-option Bellman operators contract in the span seminorm, design corrected synchronous multi-step estimators, and derive the first $\widetilde{\mathcal{O}}(1/k)$ mean-square convergence rates for average-reward option Q-learning. The results extend to hierarchical reward machines and are supported by small-scale tabular experiments.

We study the finite-sample convergence of Q-learning algorithms for the average-reward formulation of Reinforcement Learning (RL) with options. While options allow temporarily extended and hierarchical actions, the average-reward formulation incorporates continual tasks beyond discounted and episodic RL. We establish the first $\tilde{O}(1/\text{iterations})$ convergence guarantees, in the mean-square sense, for average-reward option Q-learning algorithms in both the inter- and intra-option settings. First, we prove that both inter- and intra-option Bellman operators satisfy multi-step span contractions under unichain assumptions. Then, we propose synchronous estimators of these multi-step option Bellman operators leading to two variants of option Q-learning. Finally, we extend the stochastic approximation framework to prove convergence rates of these two algorithms under the minimal assumptions used in the literature.
0
cs.LGcs.AI Kai Ruan, Jinghao Lin, Tao Chen et al. · Oct 14, 2026

This paper studies policy optimization for long-horizon software-engineering agents that learn from sparse binary test outcomes. The core idea is a training recipe—dubbed SPO++—that combines the Muon optimizer with single-rollout asynchronous updates, truncated importance sampling (TIS), token-averaged loss, and auxiliary length/tool-error penalties. The authors claim this combination achieves higher training success than SPO baselines on repository-level issue resolution for two model families, suggesting Muon is critical for long-horizon RL.

Long-horizon agents must coordinate reasoning, tool use, and environment feedback over extended interactions. Learning from sparse outcome rewards requires weighting trajectories of different lengths and updating the policy while rollouts are still being collected. We examine a single-rollout asynchronous recipe combining Muon, token-averaged loss, truncated importance sampling, and auxiliary length and tool-error penalties. Single-rollout sampling allows asynchronous learning without waiting for multiple trajectories for the same issue. We study optimizer choices and recipe components for repository-level issue resolution across two model families. The evaluated training runs show higher training success with the full recipe than with the baselines.
0
cs.LGcs.AI Mohamed Bouadi, Nassim Bouarour, Aditya Tanna et al. · Oct 14, 2026

Tabular foundation models learn to predict from in-context examples without task-specific fine-tuning, but dense architectures apply the same computation to every feature regardless of statistical context. OrionMoE-Small attacks this limitation by inserting a dataset-conditioned mixture-of-experts layer along the feature axis of a dual-axis Transformer, allowing the model to route different features to different expert networks based on both local token content and a global dataset register. If the sparse design truly helps, it could offer a principled way to scale tabular models by allocating computation heterogeneously; however, the paper never proves the MoE itself is responsible for the gains.

Tabular foundation models enable in-context learning by predicting unseen observations from a labeled support table without task-specific parameter updates. Despite recent advances, most existing approaches rely on dense architectures that apply the same transformations across features and datasets. We introduce OrionMoE-Small, a compact tabular foundation model that combines distribution-aware representations with dataset-conditioned sparse mixture-of-experts computation. Built on a dual-axis Transformer architecture, OrionMoE-Small selectively activates experts along the feature axis, with routing decisions informed by both individual feature representations and the broader dataset context. This design enables adaptive allocation of computation across heterogeneous tabular problems while retaining a shared computational pathway across tasks. OrionMoE-Small is pretrained entirely on synthetic classification tasks generated from a structured causal prior and performs downstream prediction without gradient-based adaptation. Experiments across multiple tabular benchmarks demonstrate competitive performance against substantially larger foundation models, with particularly strong results on small-context and high-dimensional datasets. Further analysis reveals favorable relative performance under class imbalance and highlights the importance of feature composition.
0
cs.LGcs.AI Yajie Zhao, Yingjun Mo, Zhiyong Wang et al. · Oct 14, 2026

The paper studies how much context a Transformer must retain to perform in-context temporal-difference (TD) policy evaluation without parameter updates. Existing theoretical constructions store the full trajectory, costing Θ(Nd) memory. The authors prove the intrinsic limit is the Krylov dimension k_L, construct a fixed-parameter attention network that compresses raw transitions to Θ(nk_L) registers, and give tight worst-case approximation error Θ(γ^m max{1,1/(m(1−γ))}) when the context falls below k_L.

Transformers can perform in-context reinforcement learning without parameter updates. However, existing theoretical constructions often require context sizes that grow with the number of sampled transitions, leading to increasing memory costs in large-scale problems. This raises a fundamental scalability question: how much context is intrinsically necessary to carry out such learning? We answer this question for fixed-policy TD evaluation. We show that, with a shared linear decoder, the minimum context dimension required to exactly reproduce the first $L$ TD iterates is the Krylov dimension induced by the sampled transitions. When the available context dimension $m$ falls below this threshold, exact reproduction is no longer possible. In this regime, we characterize the optimal approximation error and show that it decays as $\Theta(\gamma^m\max\{1,1/(m(1-\gamma))\})$ where $\gamma$ is the discounting factor. Moving from representation complexity to actual memory cost, we characterize the optimal retained memory for exact continuous representations as $\Theta(nk_L)$ and show experimentally that on a sampled task whose exact rank is full but whose numerical rank is small, a context accurate to $10^{-2}$ is $88.7\%$ smaller than the sparse sufficient statistics of TD. Finally, we establish the learnability of the shared executor through pretraining by proving global convergence of an auxiliary objective over effective attention parameters under suitable task diversity. Experiments across diverse TD evaluation problems validate parameter recovery and the predicted context--accuracy tradeoff.
0
cs.AIcs.LG Tianyu Liu, Davide Belli, Bence Major et al. · Oct 14, 2026

Explorer-Zero tackles the data-curation bottleneck for agentic RL by training a learned proposer to explore stateful environments, compose multi-step tasks from demonstrated transitions, and record intermediate environment states as verifiable checkpoints. These checkpoints supply fine-grained rewards $R^{\text{subgoal}}$ that prevent group-normalized advantages from collapsing when the solver fails, while alternating proposer–solver training adapts the curriculum automatically. On ALFWorld, ScienceWorld, and AppWorld, the approach outperforms strong self-play baselines and narrows the gap to human-data RL.

Training LLM agents on long-horizon interactive tasks with reinforcement learning typically relies on human-annotated tasks and reliable verification, both expensive to construct for stateful environments. Self-play offers an alternative, but must generate achievable, compositional tasks and provide feedback beyond binary outcomes. We propose Explorer-Zero, a self-play framework that jointly generates tasks and fine-grained, verifiable rewards through environment interaction. A learned proposer explores the environment, constructs tasks, and records intermediate and final states as verification targets. These checkpoints reward partial completion, while alternating proposer--solver training adapts the generated curriculum. We show theoretically that subtask rewards reduce collapse among unsuccessful rollouts when they complete different numbers of subtasks. Without human-annotated training tasks, Explorer-Zero outperforms the evaluated self-play baselines across ALFWorld, ScienceWorld, and AppWorld; on AppWorld, our 30B agent reaches 59.67% task success, approaching 62.65% for RL on human-annotated tasks. Further analyses support task-relevant composition and benefits on complex tasks, while self-play initialization improves subsequent human-data RL across all three benchmarks.
0
cs.CVcs.AIcs.LG Taha Sabir, Alberto Montero Molina, Thinh Quang Ly et al. · Oct 14, 2026

Medical image parsing requires a single model to produce heterogeneous outputs—categorical labels, bounding boxes, counts, numeric estimates, and free-text reports—across disparate imaging modalities. This work studies task-specific adaptation of a shared 4-bit Qwen3.5-9B backbone using lightweight LoRA adapters, where a question-only GLiClass router selects the adapter, image budget, and prompt for each case on the FLARE 2026 Task 3 benchmark. The core finding is that adaptation recipes should vary by task—rank, resolution, and prompts allocated according to demand—rather than applied uniformly.

Adapting a single vision-language model to medical image parsing requires training choices that reflect the demands of each task. We study task-specific adaptation with Qwen3.5-9B for FLARE 2026 Task 3, covering six task families and seven imaging modalities. Our system couples a shared, frozen 4-bit backbone with LoRA adapters selected from the input question. We customize adapter ranks and prompts, rebalance rare classification targets, and supplement regression training with dermatology classification examples. On the fixed internal holdout, the task-specific adapters achieve a dataset-macro balanced accuracy of 0.7791 for classification, a dataset-macro micro-F1 of 0.6286 for multi-label classification, a counting mean absolute error (MAE) of 11.73, a dataset-macro detection F1 of 0.8914, a dataset-macro regression MAE of 9.78, and a report-generation CRIMSON score of 0.8204. On the organizer's 1,703-example hidden-validation set, the submitted system achieves 0.8496 classification balanced accuracy, 0.5338 sample-averaged multi-label F1, and 14.5950 regression MAE. The final 30.1 GB container averages 3.1 seconds per case, including model loading, and produces identical outputs across repeated verification runs. The benefits vary across tasks, supporting the use of task-specific adapter configurations and image budgets within a shared medical vision-language model. Code is available at https://github.com/TahaSabir0/FLARE26-Hexapus.
0
cs.LGcs.AI Yongtao Wu, Mahdi Mansouri, Zhenyu Zhu · Oct 14, 2026

This paper studies the implicit bias of the Muon optimizer when its idealized spectral-norm steepest-descent update is replaced by the standard practical implementation: finite Newton–Schulz iterations with a positive stabilization constant. The authors prove that, contrary to the spectral max-margin behavior established for idealized Muon, the practical variant instead converges in direction to the Frobenius max-margin separator—effectively recovering the implicit bias of ordinary gradient descent. They further characterize how the iteration count, stabilization, and polynomial choice interact, and provide conditions (logarithmically growing iterations) under which spectral bias can be recovered.

Prior work on implicit bias shows that gradient descent selects the Frobenius max-margin direction on linearly separable problems, whereas Fan et al. (2025) established spectral max-margin behavior for spectral-norm steepest descent methods, including idealized Muon with exact orthogonalization. In practice, however, Muon approximates orthogonalization using finite Newton-Schulz iterations and a positive stabilization constant. We prove that these implementation choices cause Muon to select the Frobenius rather than the spectral max-margin direction in linear classification with separable data. We also establish this behavior for both exponential moving average and Nesterov-style momentum. The key technical ingredient is a summable perturbation argument showing that the resulting parameter trajectories remain at bounded distance from those of gradient descent. We then disentangle the roles of Newton-Schulz iteration count, the stabilization constant, and the Newton-Schulz polynomial, revealing distinct asymptotic behaviors, including directions that achieve neither Frobenius nor spectral max-margin. Finally, we show that Muon can recover its spectral max-margin bias with Newton-Schulz iterations growing logarithmically over training.
0
cs.LGcs.AI Mario Mallea, Angela Nebot, Franscisco Mugica · Oct 14, 2026

Sequence models usually require costly backpropagation and large datasets, leaving open the question of how far simple random reservoirs can go in few-shot settings. This paper systematically tests Echo State Network variants on the STREAM benchmark of synthetic seq2seq tasks, showing that orthonormal input initialization preserves a quasi-linear memory regime and that a proposed data-dependent gate (G-ESN) together with shared deep readouts pushes the solvable frontier toward structural tasks like adding and bracket matching. By benchmarking these cheap reservoir models against gradient-based SOTA, the study offers a practical map of when random features suffice and when fully trained models remain necessary.

Efficient sequence modeling remains a fundamental challenge in machine learning. Despite the success of Transformers, recurrent neural networks (RNNs) have recently re-emerged as efficient alternatives. Particularly, Echo State Networks (ESNs) offer a compelling framework of randomized RNNs that have been extensively studied for time-series forecasting and dynamical systems. This paper presents an empirical analysis of distributional, architectural, and optimization variants of ESNs evaluated on few-shot sequence-to-sequence (seq2seq) pattern tasks. These synthetic tasks utilize discrete symbol sequences, characterized by delays, inactive phases, and triggered markers to isolate and evaluate memory and pattern recognition capabilities. The minimality of these tasks is crucial to understand the strengths and limitations of machine learning models. Furthermore, when the input signal utilizes a one-hot encoding, ESN dynamics are governed strictly by their weight distributions, offering an ideal lens through which to study reservoir computing. Our findings map the performance boundaries of ESNs relative to state-of-the-art (SOTA) models. We show that preserving a linear state regime is vital for memory tasks, such as pattern completion or simple copy. We propose a data-dependent modulation mechanism for nonlinear input injection into the reservoir (G-ESN). G-ESN functions as a statistical filter capable of solving tasks that conventional ESNs cannot (the adding problem and bracket matching). Finally, for Deep ESNs, we evaluate a shared readout mechanism across layers that outperforms conventional concatenated readouts on structural tasks, including selective copy, sorting problem, and bracket matching.
0
cs.LGcs.AI Shuaijie Ma, Yongkang Hu, Yu Cheng et al. · Oct 14, 2026

Federated fine-tuning lets LLMs learn from distributed data, but malicious clients can poison the global model and their attack patterns may evolve as new tasks arrive. This paper proposes Safe-FCL, which continually adapts a risk detector using historical protective subspaces and then performs risk-aware, module-level aggregation of LoRA updates. The goal is to keep the global model safe across sequential tasks without forgetting previously learned attack signatures.

Federated fine-tuning enables large language models (LLMs) to learn collaboratively from distributed client data, but malicious client updates can compromise global model safety. As federated tasks evolve, changing malicious update patterns challenge the ability of existing defenses to maintain reliable risk detection over time. To address this challenge, we propose Safe-FCL, a framework integrating continual risk awareness with fine-grained safety aggregation for federated continual learning of LLMs. Specifically, the Continual Risk Awareness (CRA) mechanism learns emerging risk patterns across sequential tasks and constrains detector updates using a protective subspace derived from historical features, mitigating forgetting of previously learned malicious patterns while adapting to new risks. To translate risk awareness into aggregation-level safety control, the Fine-Grained Safety Aggregation (FGSA) mechanism combines client-level risk assessments from CRA with module-level deviations from historical update behavior to adaptively regulate aggregation contributions, suppressing potentially malicious updates while minimizing unnecessary restrictions on benign updates. Extensive experiments demonstrate that Safe-FCL outperforms current SOTA methods, improving global-model Rule and MD-Judge scores in federated continual learning by 0.41% and 0.64%, respectively, while increasing average risk recognition accuracy by 2.01%.
0
cs.LGcs.AI Radovan Halu\v{s}ka, Christopher Solinas, Nathan R. Sturtevant et al. · Oct 14, 2026

Neural networks require vector inputs, yet many problems involve higher-order objects like probability distributions, functions, and graphs that lack compact representations. This paper proposes a unified encoder-decoder framework that embeds families of such objects into fixed-size vectors using only finite sample sets, with a domain-specific decoder providing the training signal. The core idea is to replace per-object fitting with amortized learning: a permutation-invariant encoder compresses samples into an embedding that conditions a decoder, enabling generalization to unseen family members. The authors instantiate this on distributions, functions, and graphs, and claim it is the first framework general enough to work across object types.

Neural networks operate naturally on finite-dimensional vectors, yet many problems require reasoning about higher-order objects such as distributions, functions, and graphs. Such objects capture relationships within a ground set and rarely admit an explicit compact representation. Moreover, we typically want a compact representation not of a single higher-order object but of an entire family of such objects, including members unseen during training. While a few approaches exist for particular object types, none is designed to work across object types. We propose a unified framework that represents any higher-order object as a fixed-size embedding vector learned over a family of such objects. An encoder maps an object, seen only through a finite set of samples, to an embedding, and a domain-specific decoder supplies the training signal by modeling the object conditioned on that embedding. Since the encoder works only from samples, one architecture serves different object types. Because the decoder is a model already suited to single instances of the object type, the framework inherits that model's native operations and inductive biases. We instantiate the framework on distributions, functions, and graphs. In each case we show that a single trained model represents an entire family and generalizes to members unseen during training. Furthermore, we demonstrate the downstream utility of these embeddings: interpolating distributions in embedding space, solving differential equations in embedding space, and predicting unseen edges from a graph's embedding.
0
cs.CLcs.AIcs.LG Fei Wu, Zhenrong Zhang, Jianshu Zhang et al. · Oct 14, 2026

On-policy self-distillation gives reasoning models dense token-level supervision via a teacher that sees reference solutions, but this signal can drift toward style and ignores whether the student can finish correctly on its own. VASPO repurposes the same privileged reference inside PPO to train both a prefix-value critic and a bounded self-teacher that reweights selected positive advantages. The paper reports average gains of roughly 2.6 percentage points over PPO on math benchmarks and leading accuracy on Knights and Knaves.

On-policy self-distillation provides dense supervision for reasoning models via a privileged self-teacher. However, this supervision reflects local token preferences rather than the expected return of an unguided student continuation. Recent methods combine this signal with GRPO. They use GRPO for update directions and the teacher for signal scaling. However, we demonstrate a more effective integration within PPO. Specifically, we identify complementary positional dynamics: privileged information enhances a critic's value-prediction accuracy toward the response's end, whereas self-teacher guidance is strong initially, drops mid-generation, and rebounds finally. These insights motivate Value-Augmented Self-Distillation Policy Optimization (VASPO), wherein a self-teacher and critic evaluate identical privileged information at student-generated prefixes. The critic computes token-level PPO advantages from predicted continuation returns, while the self-teacher modulates selected positive advantages via bounded probability ratios. Across six mathematical reasoning benchmarks, VASPO improves mean accuracy over PPO by 2.59 and 1.81 percentage points on Qwen3-4B and OLMo3-7B variants, respectively, and achieves the highest mean accuracy among evaluated methods on Knights and Knaves. Our code is available at https://github.com/cii030/VASPO.
0
cs.CLcs.AIcs.LG Davide Baldelli, Robyn Kozierok, Mirko Bronzi et al. · Oct 14, 2026

Probes on hidden activations are a promising tool for detecting deception and harmful intent, but they assume models cannot volitionally alter their own representations. Baldelli et al. ask whether open-weight LLMs can control their activations through in-context neurofeedback, devising a cross-model writer-subject protocol to disentangle genuine self-access from in-context pattern matching. Across 42 semantically opaque readouts and six safety monitors, they find that free-generation effects are largely explained by sentence-selection competence, while fixed-sentence control is tiny and monitor evasion is essentially absent. The results rigorously bound the current threat but flag a capability that may grow with model scale.

Probes on activation vectors are widely used to detect misbehavior in language models, but their success relies on the assumption that models cannot control what their activations encode. Humans can learn, through neurofeedback, to control the firing of even single neurons, and recent studies investigated whether this ability extends to language models, reaching conflicting conclusions. No experimental design so far disentangles competence at the task from genuine access to one's own activations. We fill this gap with in-context neurofeedback sessions, in which a "subject" model is asked to maximize (or minimize) a score derived from its activations, shown back to it at every turn. In our first experiment, the sentences are generated either by the subject or by a second "writer" model, while the score always derives from the activations of the subject. We test ten open-weight models from five families, with sizes from 9B to 32B. Some models move their own scores considerably, but no more than a writer of comparable competence moves them. Is this because the task requires no privileged access, or because the subject controls its activations while processing the sentences, whoever wrote them? To answer this, we prefill a fixed sentence as the subject's reply at every turn and ask it to increase or decrease the score. Three out of ten models show small but consistent control effects. We further ask whether models can control their activations to the point of fooling an activation monitor. We make models repeat sentences a monitor would flag, such as a deceptive answer for a deception monitor, and ask them to raise or lower the monitor's score. No model learns to lower it, but one model can raise it. In-context control of a model's own activations might not yet be a threat to activation monitors, but it exists in a few models and needs to be tracked as models evolve.
0
cs.LGcs.AI Christos Panourgias, Roberto Stanzione, Adrien Petralia et al. · Oct 14, 2026

Model selection for time-series anomaly detection remains a critical open problem because no single detector dominates across heterogeneous datasets. This paper proposes RAMSAD, a retrieval-based framework that replaces trainable meta-learners with non-parametric similarity search over a knowledge base of past series and their detector-performance profiles. By matching new queries to previously seen instances via time-series foundation model embeddings—optionally augmented with multimodal semantic alignment—the authors argue they can achieve stronger in-distribution accuracy and out-of-distribution robustness at a fraction of the computational cost.

Anomaly detection is a fundamental task for time-series analytics, with important implications for the downstream performance of many applications. Despite the large number of anomaly detection methods proposed in the literature, recent benchmark studies have shown that no single detector performs best across highly heterogeneous time series. Therefore, a practical and scalable solution is to develop a model-selection method that, for a given time series, selects the anomaly detector most likely to perform well. Nevertheless, the model selection approaches proposed in the literature suffer from a significant drop in accuracy when applied in Out-of-Distribution (OOD) settings. In this paper, we tackle the aforementioned limitation and propose RAMSAD, a retrieval-based framework for model selection in time-series anomaly detection. Given a new time series, our method queries a knowledge base of previously observed instances, retrieves the top-k most similar series, and transfers detector recommendations from their performance profiles. The framework can operate with standard similarity measures as well as embedding-based representations. Our results show that RAMSAD outperforms existing baselines in In-Distribution (ID) settings and is the most robust selector in OOD, where, when selecting a small subset of detectors, it matches or exceeds ensembles that run the entire detector pool, at a fraction of their cost. Overall, we demonstrate that similarity-based retrieval constitutes a strong and efficient foundation for model selection in time-series anomaly detection.
0
cs.CLcs.AIcs.LG Jia Tong, Senyou Deng, Min Peng et al. · Oct 14, 2026

Agent skills let frozen models adapt through editable external instructions, but optimizing them faces a tension: batching many changes obscures which edit mattered, while judging each edit only by its immediate score risks discarding useful stepping stones. AtomSearch proposes contrastive atomic editing—one primitive rule edit tied to execution evidence—and Skill-State Tree Search (STS), which tests short edit compositions and keeps a composed state only if it beats its primitive child. The hope is to find compact, inspectable skills without greedy myopia.

Agent skills adapt frozen models through editable external instructions. Revising several rules at once makes changes difficult to inspect, while judging each local edit only by its immediate score can discard useful intermediate states. We introduce AtomSearch, which keeps revision actions local while using subsequent skill states to guide search. Contrastive atomic editing links each primitive rule edit to failed and successful executions, a behavioral hypothesis, and an intended scope. Skill-State Tree Search (STS) evaluates short edit compositions and selectively retains them as executable states that can be revisited and extended. Outcome memory supplies later proposals with observed repair and regression statistics. We evaluate AtomSearch across four tasks and five target models. With deepseek-v4-flash, it exceeds SkillOpt on three of four tasks, including a 9.68-percentage-point gain on LiveMathematicianBench, and SkillOpt's final skills contain 2.3-19.7 times as many characters. In 67% of completed searches, the best validation state passes through an ancestor that was not highest-scoring at its depth. These results show how local rule edits can support the exploration of useful revision paths and yield compact skills, with additional offline evaluations used to explore alternative states.
0
cs.LGcs.AI Sifan Yang, Hongyi Tao, Dingzhi Yu et al. · Oct 14, 2026

Training large language models is memory-intensive, and subspace projection methods like GaLore can diverge when stochastic noise dominates the selected low-rank gradient space. This paper proposes SMRC and PMRC, which keep optimizer states in a low-dimensional subspace while adding the orthogonal residual gradient back into the parameter update. The authors prove $O(T^{-1/4})$ convergence rates for momentum SGD and for Muon under a mixed stationarity measure, and demonstrate empirical gains on LLM pretraining and fine-tuning up to $1.3$B parameters.

Memory-efficient subspace optimization has emerged as a practical approach to reducing the memory cost of LLM training while preserving optimization performance. Previous algorithms typically rely on the leading singular subspace of stochastic gradients, yet the gradient noise may dominate the subspace in later training stages and can prevent convergence. To overcome this limitation, we adopt a residual correction framework, which maintains the optimizer states in the subspace and adds the orthogonal residual gradient to the parameter update. We first integrate this scheme into momentum SGD to develop SMRC, which attains an $O(T^{-1/4})$ convergence rate for a fixed orthonormal basis, matching that of full-space momentum SGD. Our analysis introduces a subspace virtual iterate to resolve the coupling between the momentum and the residual correction. Furthermore, we extend SMRC to a multi-stage variant which periodically recomputes bases to track the evolving dominant subspace and retains the same rate. Since each basis refresh breaks the consistency of the virtual iterates, we devise a boundary correction approach that aligns them across stages. Beyond momentum SGD, we incorporate our framework into Muon and propose PMRC, which applies the matrix sign to the projected momentum. Through a weighted momentum tracking analysis that accounts for the residual gradient, we establish an $O(T^{-1/4})$ convergence rate for PMRC, which matches the rate of full-space Muon. We also design a multi-stage PMRC with periodically refreshed bases, which maintains a convergence rate of the same order. Extensive experimental results confirm the effectiveness and efficiency of our algorithms.
0
cs.LGcs.AIstat.ML Amartya Banerjee, Harlin Lee, Caroline Moosm\"uller · Oct 14, 2026

Population dynamics are often observed only as unpaired snapshots, leaving the evolution between measurements fundamentally underdetermined. This paper introduces Neural Wasserstein Splines (NWSP), a method that learns a continuous curve of distributions by penalizing the intrinsic Wasserstein acceleration rather than the acceleration of individual particles, which is representation-dependent. It couples a neural spline generator with a finite-dictionary projection of velocity and its time derivative, achieving competitive results on synthetic and single-cell benchmarks.

Population dynamics are often observed only as unpaired snapshots, which leave the evolution between measurements underdetermined. Second-order regularization can constrain it, but a penalty on the accelerations of sampled particles is a prior on joint particle paths, and different particle representations of the same marginal curve can have different acceleration costs. We therefore introduce Neural Wasserstein Splines (NWSP), a sample-based method that learns a continuous population curve under a Wasserstein smoothness prior. A neural network maps latent variables to the control points of a cubic spline in time, so samples at any time come from the spline basis without numerical integration. We project the generator's velocity onto gradients of a finite dictionary of scalar potentials and differentiate this measure-dependent projection in time, obtaining a finite-dimensional surrogate for the intrinsic Wasserstein acceleration, trained jointly with distributional fit losses. For a fixed dictionary and ridge parameter, we prove that the population penalty depends only on the marginal curve and bound its sensitivity to moment-estimation errors. Learned potentials (NWSP-N) sharpen acceleration estimates. On various synthetic and single-cell benchmarks, NWSP is highly competitive with transport and flow-based methods, ranking in the top two in 11 of 16 settings. The penalty improves on an unregularized spline in 13 of 16 settings and, while competitive with a particle-acceleration penalty, adds provable representation invariance.
0
cs.LGcs.AI Wei Chen, Yubing Wu, Zhou Xu et al. · Oct 14, 2026

Hindsight-Privileged RLVR tackles the inefficiency of group-based RLVR methods like GRPO, which require multiple rollouts per prompt and lose their learning signal when all sampled outcomes are identical. The core idea is to replace group comparisons with a single rollout per prompt, using the trajectory's own realized future as training-only privilege: a frozen snapshot of the generating policy scores each token under both causal and hindsight views, and normalized Jensen--Shannon divergence between these full-vocabulary distributions weights a historically centered reward. This yields bounded token-level credit without a learned critic or reference solution, training $1.2$--$2.3\times$ faster than GRPO on Qwen3-8B but with task-dependent accuracy tradeoffs.

Reinforcement learning with verifiable rewards (RLVR) trains language models from checked task outcomes, but assigning a terminal reward to individual reasoning tokens or actions remains difficult. Group-relative methods such as Group Relative Policy Optimization (GRPO) sample several attempts per prompt, but lose their reward-based learning signal when all attempts receive the same outcome. We propose Hindsight-Privileged RLVR (HP-RLVR), a single-rollout method that turns a completed trajectory's own future into token-level update weights. The generating policy snapshot scores each eligible token under its causal history and under a hindsight context containing the realized future and verifier feedback. A normalized Jensen--Shannon divergence between the two views weights the reward relative to a historical baseline, yielding bounded token credit for both successful and failed attempts, with no reference solution or learned critic. In paired Qwen3-8B runs over a shared $100$-step schedule, HP-RLVR trains $1.21\times$, $1.46\times$, and $2.30\times$ faster than GRPO on mathematics, code generation, and ALFWorld, using one eighth the rollouts per prompt. Accuracy is comparable overall, with HP-RLVR ahead on mathematics at all three sizes and GRPO ahead on code and ALFWorld. In ALFWorld, $43.2\%$ (4B) and $49.0\%$ (8B) of GRPO's gradient-norm records are zero, versus none for HP-RLVR.
0
cs.LGcs.AImath.OC Kaifei Wang, Zikang Shan, Liwei Wang et al. · Oct 14, 2026

Momentum is a pillar of modern optimization, yet its benefits in reinforcement learning with verifiable rewards (RLVR) remain unclear. This paper studies a KL-regularized bandit formulation of RLVR and proves that, with exact policy gradients, larger EMA coefficients strictly accelerate convergence within a safe overshoot-free range. Its central insight is that stochastic sampling obscures this gain unless the coefficient scales with the learning rate so that $\eta/(1-\beta)$ stays constant, a principle validated in bandits and in LLM post-training on Qwen models.

Momentum is a classical tool for accelerating convex optimization and a standard component of modern large language model training. Yet recent empirical work has questioned the usefulness of first-moment momentum in reinforcement learning with verifiable rewards (RLVR). We study this question through a bandit formulation of RLVR with bias-corrected exponential moving average (EMA) momentum, where the EMA coefficient $\beta$ controls how quickly the weights on past gradients decay. When policy gradients are evaluated exactly, we prove that increasing $\beta$ within a range that prevents overshoot strictly accelerates optimization. When gradients are estimated from sampled responses, a positive expected gain can be obscured by sampling fluctuations. We show that as the learning rate $\eta$ decreases, the expected gain with fixed $\beta$ vanishes faster than these fluctuations, whereas increasing $\beta$ to keep $\eta/(1-\beta)$ within the safe regime preserves a nonvanishing gain that eventually dominates them. Bandit simulations quantitatively validate these predictions. RLVR experiments on Qwen3-4B-Base using SGD with EMA and AdamW support our principle of choosing larger coefficients in practical LLM post-training. On Qwen2.5-32B with AdamW, simply increasing $\beta_1$ improves mean validation accuracy by nearly $2$ percentage points over the conventional choice, demonstrating the principle's effectiveness at a larger model scale.
0
cs.LGcs.AI Haruki Settai, Tadashi Kozuno, Shinji Ito · Oct 14, 2026

Semi-gradient TD with nonlinear function approximation is notoriously unstable because the bootstrap target moves as the parameters evolve. This paper studies the problem in the mean-field regime for two-layer neural networks and identifies a stability regime where entropy-driven relaxation dominates the sensitivity of the moving target. Under the resulting condition, the authors prove exponential last-iterate convergence and quantify the regularization bias, then extend the same mechanism to soft Q-learning.

Semi-gradient temporal-difference learning is a core algorithmic component of reinforcement learning, but bootstrapping with function approximation generally induces a non-conservative vector field and can lead to instability. We analyze this effect for two-layer neural networks in the mean-field regime and identify a simple stability mechanism in which entropy-induced relaxation dominates the sensitivity of the moving bootstrap target. Under this condition, we prove exponential convergence to a unique fixed point and quantify its bias relative to the unregularized Bellman solution. We also show that the same mechanism extends to entropy-regularized off-policy control through soft Q-learning.