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.CLcs.LG Shuxuan Li, Jialiang Kang, Yanghao Zhou et al. · Oct 14, 2026

Speculative decoding drafters trained on short sequences collapse on long-context inputs because their position encodings extrapolate poorly. FOLD fixes this without retraining by keeping exact positions in a local window and compressing all distant context into a shared-position far memory. A concentration factor $\alpha$ in a decoupled softmax prevents attention dilution as the memory grows. Across six target–drafter pairs and contexts up to 256K tokens, the method consistently raises speedup, in the best case more than doubling it.

Speculative decoding accelerates LLM generation through lightweight drafting and parallel verification. It is particularly promising for long inputs, where KV-cache access costs can be amortized over multiple accepted tokens. Yet most drafters are trained on short sequences and degrade on long inputs, while long-context training is costly. Existing training-free approaches restrict the context window, rescale positions, or augment recent context with retrieved distant entries. Drafters, however, read target hidden states that already encode contextual and positional information, and their lightweight architecture makes full-history attention affordable. We propose FOLD, a training-free method that preserves exact local positions and treats all distant entries as far memory at a shared position. To counter attention dilution as the memory grows, we sharpen attention within it to emphasize relevant entries while separately controlling its total weight relative to the local window. Across six target-drafter pairs and diverse benchmarks of up to 256K tokens, FOLD predominantly delivers the highest speedup, with the best case more than doubling the speedup over autoregressive decoding, from 2.08$\times$ to 4.22$\times$.
0
stat.MLcs.LG Soundouss Messoudi, Youness Mansar · Oct 14, 2026

LLM workflows rely on fast binary decisions like moderation and reranking, but autoregressive generation is too slow for tight latency budgets. This paper studies System One models—lightweight scorers that emit bounded decision scores—and proposes Learn Then Test (LTT) to certify thresholds that maximize the fraction of inputs receiving the utility-maximizing action while controlling the False Omission Rate (FOR) with a finite-sample guarantee. The authors show across six datasets that naive thresholding at the target risk level exceeds the budget in roughly half of settings, whereas LTT violates it in only 7.8% of cases, making a strong case that a probability-like interface is insufficient for safe deployment under asymmetric risk.

LLM workflows and agents rely on a set of binary decisions like allow/block an input, or discard/retain a retrieved document. When human deferral is unavailable, uncertain cases must still receive an action. We study System One decision scores in this setting by defaulting to a conservative fallback when uncertain, and controlling the false omission risk within the subset of inputs assigned the utility-maximizing action. Across six datasets covering moderation, safety, and reranking, we compare general System One models with task baselines and a local open-weights fine-tuned System One model. We show that naive baselines for threshold selection exceed the target risk in roughly half of settings with defined test risk, compared with 7.8\% for Learn Then Test (LTT), which leverages finite-sample statistical guarantees. Our results show why relying on a probability-like interface is insufficient for deployment decisions and how risk-budgeted selection trades useful action against observed error.
0
cs.LG Fanyao Li, Yuxi Liu, Xiaohao Liu et al. · Oct 14, 2026

Low-rank adapters can waste their parameter budget because not every learned direction improves downstream performance. ReGRA attacks this by splitting adaptation into two stages: first train a standard LoRA adapter, then estimate task gradients, project out the components already captured by the first stage, and use the residual directions to initialize a second adapter with an orthogonality regularizer. By merging both updates into the backbone, it keeps inference cost identical while increasing average accuracy within the same total rank budget.

Low-rank adaptation (LoRA) enables parameter-efficient fine-tuning by learning low-rank weight updates. Increasing rank expands adaptation capacity, but does not guarantee that all learned directions improve downstream performance. In a diagnostic experiment with rank-16 LoRA, we show that retaining only the top singular components achieves the best performance, whereas adding further components degrades it. Motivated by this, we propose Residual-Gradient Adaptation (ReGRA), a two-stage method for making more effective use of a fixed rank budget. After the first stage, ReGRA estimates task gradients on a calibration set and removes their projections onto the two-sided subspace induced by the first-stage update. It then aggregates the residual-gradient directions to initialize a second-stage adapter without changing the first-stage update. An orthogonality regularizer encourages the two updates to capture complementary directions. Both updates can be merged into the pretrained weights, incurring no additional inference overhead. Across diverse benchmark sets, ReGRA achieves stronger performance than alternatives with the same total rank and even matches LoRA adapters with twice the rank. Ablation studies and further analysis support the contribution of its core design choices. Code is available at https://github.com/Fanyao-Li/ReGRA.
0
cs.LG Olayiwola Arowolo, Oluwadara Adedeji, Peter Palensky et al. · Oct 14, 2026

Power grid steady-state analysis demands fast, generalizable surrogates for power flow, contingency screening, and voltage stability margin estimation. This paper investigates whether a pretrained tabular foundation model (TabPFN) can serve as a unified neural solver through in-context learning, without task-specific fine-tuning. The authors evaluate the approach on 118- and 2000-bus systems, arguing that it offers a flexible alternative to training separate graph or MLP baselines for each task and topology.

Foundation models have shown strong generalization across scientific domains, but their use in power systems remains largely task-specific. We investigate whether a pretrained tabular foundation model (TFM), TabPFN, can serve as a general-purpose neural solver for power grid steady-state analysis. We evaluate the TFM on three tasks: power flow (PF) prediction, N-k contingency screening (NCS), and voltage stability margin (VSM) prediction. We performed experiments on 118- and 2000-bus systems, comparing the TFM against task-specific neural baselines. Without retraining or fine-tuning, the TFM matches or outperforms the best task-specific model in several settings across the three tasks. For 118-bus PF prediction, a variant of the TFM achieves errors of $1\times 10^{-3}$ pu for voltage magnitudes and $0.30^\circ$ for voltage angles, outperforming all baselines. On unseen N-1 contingency and close-to-infeasible PF cases, the TFM is competitive with the DCPF approximation. For millions of unseen N-2 contingencies, the TFM achieves AUCs of 0.92 and 0.95 for branch and voltage violation screening, respectively. The TFM also attains accurate VSM prediction performance for the 2000-bus system with limited labelled data. The TFM attains strong results across the tested tasks despite pretraining without any power-system-specific dataset and requiring no fine-tuning step. These results offer a glimpse into the potential of pretrained TFMs as a neural solver for a range of power grid analysis tasks.
0
cs.DScs.LG Liang-Yu Zou · Oct 14, 2026

The paper studies proper hypothesis selection: given $n$ candidate distributions and sample access to an unknown target $P$, the goal is to return one candidate with total variation distance at most $3\min_i d_{TV}(P,H_i)+\varepsilon$. The core idea is to maintain weights over the candidates, define a weighted median of their densities as an unnormalized intermediate measure, and perform multiplicative weight updates using balanced tests derived from this median. The result achieves the optimal factor-3 approximation and sample complexity $O(\varepsilon^{-2}\log(n/\delta))$ while improving the running time to $O(n\varepsilon^{-2}\log(n/\delta))$, answering an open question of Aamand et al. (COLT 2026) by reducing the failure-probability dependence from $1/\delta$ to $\log(1/\delta)$.

Given $n$ distributions (hypotheses) $\mathcal H=\{H_1,\ldots,H_n\}$ and samples from an unknown target distribution $P$, proper hypothesis selection seeks a distribution whose total variation distance to $P$ is close to the best achievable within the known distributions. We give an algorithm that, using $O(\varepsilon^{-2}\log(n/\delta))$ target samples, running in $O(n\varepsilon^{-2}\log(n/\delta))$ time, with probability at least $1-\delta$, returns $H_{\hat i}$ satisfying $d_\mathrm{TV}(P,H_{\hat i})\leq 3\min_{i\in[n]}d_\mathrm{TV}(P,H_i)+\varepsilon.$ The algorithm achieves the optimal approximation factor and sample complexity. The running time improves the dependence on the failure probability from polynomial to logarithmic in the state-of-the-art bound $\tilde{O}(n\varepsilon^{-2}\delta^{-1})$ of Aamand, Aliakbarpour, Chen, and Silwal (COLT 2026), thereby answering their Question~2.
0
cs.LGmath.OCstat.ML Angelos Assos, Anand Brahmbhatt, Elad Hazan · Oct 14, 2026

This paper studies online convex optimization against comparators that switch $k$ times among only $m$ distinct points, generalizing the classic experts tracking problem to continuous domains. While discretizing the domain and applying the Bousquet–Warmuth long-term-memory algorithm yields information-theoretic regret $\widetilde O(md+k)$, this is computationally inefficient. The authors prove that, under the hypergraphic planted-clique conjecture, no polynomial-time algorithm can achieve $\widetilde O(k^a d^b)$ regret whenever the exponents violate $\theta a + b < \theta + 1/2$, establishing a fundamental computational barrier for exp-concave losses even when $m=2$.

For the fundamental problem of prediction from expert advice, Freund asked whether one could compete with a comparator that switches $k$ times among only $m$ experts. For a discrete set of experts, this problem was resolved by Bousquet and Warmuth (2020), who gave an efficient algorithm with optimal regret matching the information-theoretic limit. We study the analogous problem in online convex optimization, where the action space is continuous. An inefficient algorithm can be derived by discretizing the action space and applying the Bousquet--Warmuth algorithm, giving regret of $\widetilde{O}(md+k)$, while the best known efficient algorithm for this problem has regret $\widetilde{O}(dk)$. We show that this computational inefficiency is inherent, establishing a separation between the information-theoretic and polynomial-time regret achievable for this problem. Specifically, under the hypergraphic planted-clique conjecture, we show that for every polynomial-time algorithm there exist instances for which it incurs at least $(k\sqrt d)^{1-o(1)}$ regret.
0
cs.LGstat.ML Yuling Jiao, Lican Kang, Jerry Zhijian Yang et al. · Oct 14, 2026

Gaussian diffusion models poorly cover extreme regions for heavy-tailed data because their light-tailed corruption kernels yield exponentially sparse tail samples. This paper introduces Annealed Lévy Diffusion (ALD), which specifies time-dependent $\alpha(t)$-stable perturbation marginals via a Gaussian scale-mixture representation, enabling a two-phase schedule that transitions from heavy-tailed exploration to Gaussian refinement. The construction retains conditional-Gaussian score matching with only one added scalar input, making it compatible with standard diffusion training while targeting distributions with polynomial tails.

Gaussian diffusion models offer a practical framework for generative modeling. Denoising L\'{e}vy Probabilistic Models (DLPM) introduce heavy-tailed perturbations with a fixed tail index throughout denoising. We introduce Annealed L\'{e}vy Diffusion (ALD). ALD varies the stability index $\alpha(t)$ to combine heavy-tailed exploration with Gaussian refinement. Our marginal-first formulation represents stable noise as a scalar Gaussian scale mixture. This formulation enables standard Gaussian noise-prediction training with one additional scale input. We study two sampling processes. ALD-renoise estimates the clean sample and injects fresh marginal noise. ALD-posterior preserves dependence between successive states through an accumulated conditional variance and a Gaussian bridge update. Both samplers reuse the same trained predictor with different scalar inputs. We evaluate ALD on four 2D benchmarks and an imbalanced nine-mode mixture. We also examine training efficiency and the effects of prediction perturbations on clipped Holtsmark data. Additional experiments evaluate MNIST and CelebA64 generation and CIFAR-10LT augmentation. ALD preserves local geometry on the evaluated light-tailed and manifold-structured targets.
0
cs.LG Jonghae Park, Sungil Son, Jonghun Shin et al. · Oct 14, 2026

C-Slerp addresses long-horizon, sparse-reward goal-conditioned reinforcement learning without prior data, rewards, or demonstrations by leveraging hyperspherical contrastive representations for adaptive waypoint planning. Instead of training a separate high-level policy, it infers intermediate subgoals in closed form via spherical linear interpolation on a learned unit hypersphere and dynamically adjusts the number of waypoints to match each task's difficulty. Experiments on robotic manipulation and navigation show the method solves tasks like eight-cube tower stacking from scratch where strong baselines entirely fail, suggesting that adaptive task decomposition—not merely added hierarchy—is key to scaling goal-reaching RL.

We develop a self-supervised reinforcement learning algorithm that learns to build towers of up to eight cubes from scratch, without prior data, rewards, demonstrations, or pretrained models. Our method, Contrastive Planning via Spherical Linear Interpolation (C-Slerp), uses contrastive representations as reusable building blocks and repeatedly composes them to infer a sequence of intermediate waypoints. Under our hyperspherical parameterization, the learned representations allow intermediate waypoints to be inferred in closed form through spherical linear interpolation, without training a separate high-level policy or performing graph search. Importantly, the representation geometry also tells us how to adjust the number of waypoints based on the difficulty of each goal-reaching problem. Experiments show that our method succeeds on long-horizon navigation and robotic manipulation tasks, including settings where strong baselines fail to achieve a single success. Ablations further show that using a fixed number of waypoints substantially degrades performance, highlighting the importance of adapting the plan to task difficulty.
0
cs.LGstat.ML Kapilan Balagopalan, Hao Qin, Kwang-Sung Jun et al. · Oct 14, 2026

Contextual partial monitoring is a sequential learning problem where the learner observes only partial feedback instead of incurred losses. The paper proposes E2D.PMSide, a computationally efficient instantiation of the Estimation-to-Decisions (E2D) meta-algorithm that reduces contextual partial monitoring to online regression. Its core contribution is replacing E2D's intractable per-round minimax exploration rule with tractable surrogates—epsilon-greedy for globally observable games and inverse gap weighting for locally observable ones—thereby achieving optimal $O(T^{2/3})$ and $O(\sqrt{T})$ regret rates. The work matters because it provides the first oracle-efficient algorithms for this setting with general function approximation and demonstrates practical benefits on LLM routing and label-efficient prediction.

We study finite stochastic partial monitoring (PM) games with side information, a paradigm of sequential learning with partial feedback. On each round $t \in [T]$, both player and environment observe a context and player chooses an action and environment simultaneously chooses an outcome. The player neither observes the outcome nor the loss; instead, they only observe a feedback symbol which is partially informative about the outcome. We utilize the Estimation-to-Decisions (E2D) meta-algorithm of Foster et al.(2021) to design E2D.PMSide, a computationally efficient framework for contextual partial monitoring, which uses an online regression oracle to efficiently estimate the outcome distributions based on historical (context, action, feedback symbol) tuples. Since the exploration rule given by the original E2D algorithm is intractable to compute, we propose two computationally efficient exploration rules that achieve optimal regret orders in $T$: in the globally observable setting, our exploration rule is based on $\epsilon$-greedy, while in the locally observable setting, our exploration rule uses inverse gap weighting (Foster & Rakhlin, 2020), which was previously only known to be effective in the bandit setting. We show that a variant of E2D.PMSide named OE2D.PMSide can be combined with offline regression oracles to give an efficient algorithm with optimal regret when the contexts are drawn independently from a fixed distribution. We evaluate our algorithm on a synthetic label-efficient prediction game and an LLM routing task, in which a learner decides from prompt, uncertainty, and internal state features whether to escalate a query from a weak model to a strong one. We show that our method has superior performance to prior approaches.
0
cs.LG Woobin Park, Jaeyeon Kim, Yilun Du et al. · Oct 14, 2026

One-step generative models typically demand complex distillation or fragile end-to-end distributional training. This paper introduces Priming Models, a two-stage framework that first pretrains with a simple reconstruction objective—surprisingly, even fixed-noise denoising suffices—and then refines the network with feature-space distributional losses. The core claim is that this reconstruction stage fully removes the training difficulty of distributional optimization, enabling state-of-the-art quality at roughly one-tenth the compute of prior one-step methods.

Diffusion and flow matching are now dominant frameworks for generative modeling in continuous domains, but they typically require multiple network evaluations at inference. This has motivated extensive research on one-step generative modeling primarily built on diffusion and flow, though existing approaches often rely on complex training machinery. In this work, we introduce Priming Models, a simple two-stage framework for training one-step generators. Motivated by the difficulty of distributional training from scratch, Priming Models are first trained on a reconstruction objective, e.g., pixel-space denoising at a fixed noise level, before undergoing feature-space distributional post-training. Our central scientific finding is that this simple reconstruction stage is already enough to entirely alleviate the difficulty of distributional training. Empirically, across class-conditional image generation, text-to-image generation, and world modeling, Priming Models significantly advance the Pareto frontier of training compute vs. generation quality. On ImageNet at $256 \times 256$ resolution, we achieve a state-of-the-art FID of $1.19$ among one-step generators using approximately one-tenth the training compute of prior methods with comparable FID. We open-source our codebase at https://github.com/JaeyeonKim01/priming.
0
cs.LG Abdel-Rahim Mezidi, Jordan Patracone, Saverio Salzo et al. · Oct 14, 2026

Neural operators learn mappings between function spaces, yet training deep architectures remains difficult. This paper proposes viewing each operator layer as the minimizer of a Bregman-regularized functional optimization problem in Banach spaces. This unifies classical neural operators under a proximal framework and yields a new Bregman variant that adds an inverse activation term, showing improved depth scaling and accuracy on standard PDE benchmarks.

We present several advances on neural operators by viewing the action of operator layers as the minimizers of Bregman regularized optimization problems over Banach function spaces. The proposed framework allows interpreting the activation operators as Bregman proximity operators from dual to primal space. This novel viewpoint is general enough to recover classical neural operators as well as a new variant, coined Bregman neural operators, which includes the inverse activation operator and features the same expressivity of standard neural operators. Numerical experiments support the added benefits of the Bregman variant of Fourier neural operators for training deeper and more accurate models.
0
cs.CLcs.LG Ian Magnusson, Noah A. Smith, Pang Wei Koh · Oct 14, 2026

COUNT tackles the GPU-memory bottleneck of scaling language-model embeddings by replacing learned n-gram tables with precomputed corpus counts stored on cheap host RAM. It constructs input embeddings via a weighted sum of the model's own unembedding vectors, requiring no extra learned parameters and enabling training-free domain adaptation at inference. On 60M--1B-parameter backbones, the authors demonstrate consistent perplexity and downstream gains without increasing GPU memory pressure.

The scaling relationship of compute and performance for language models (LMs) has been extensively studied; more recently, models are being designed to scale memory use, as well. In particular, adding embeddings for n-grams scales memory separately from compute use, because each pass only processes individual embedding rows, which are much smaller than the storage used for the whole embedding table. However, existing methods train embeddings from scratch, which needs expensive GPU memory for fast reading and writing. Meanwhile, the ordinary memory of the host system is cheap and underutilized by these training methods. To scale on host memory, we introduce COUNT (Counting Offline Unlocks N-gram Transformers), which builds n-gram embeddings without adding learned parameters by storing precomputed n-gram counts on host. We use the corpus frequencies to weight together the LM's own unembeddings of the next tokens most often following the current n-gram history. At a fixed compute budget, COUNT can be added as an additional input embedding for pretraining 1B-parameter LMs to improve validation loss as much as an LM with 57% more compute. If GPU memory is limited, using COUNT to let backbone parameters fill GPU memory gives better performance than splitting memory with additional learned n-gram embeddings. Beyond efficiency, COUNT allows switching in a different table of n-gram counts from a domain-specific corpus at inference time to adapt to a new task without any training.
0
cs.LGcs.CL Minsuh Joo, Youna Kim, Choonghyun Park et al. · Oct 14, 2026

Current evaluations of Knowledge Extraction Enhancement (KEE) techniques collapse recovery and disruption into a single accuracy score, concealing critical trade-offs. MAPLE introduces a unified diagnostic framework built on a pretraining-verified benchmark and three complementary metrics—surface-level $\text{Gain}/\text{Loss}$, probability-level $\text{PRS}$, and beyond-sampling $\text{SGR}/\text{DGR}$—to dissect intervention effects at multiple levels. Across six techniques and three model families, MAPLE shows that prompting recovers deep failures but disrupts many successes, while decoding and steering often reshuffle token probabilities without consistent benefit. The work offers a principled, actionable standard for comparing KEE methods beyond headline accuracy.

LLMs absorb vast knowledge from training data, yet their inherently probabilistic nature offers no guarantee that this knowledge will surface in their outputs. To bridge this gap, Knowledge Extraction Enhancement (KEE) techniques such as activation steering, decoding adjustments, and prompting modify inference, potentially reinforcing existing predictions or surfacing less accessible answers. Yet their evaluations centered on aggregate accuracy, which obscures diverse effects by collapsing recovered failures and disrupted successes into a single score. We introduce MAPLE, a unified framework for characterizing how KEE techniques affect parametric knowledge extraction. Its diagnostic metrics disentangle these effects at the surface, probability, and beyond-sampling levels by measuring answer recovery and disruption, gold-token rank shifts on unrecovered failures, and recovery of answers missed by fixed-budget sampling. We evaluate six KEE techniques spanning decoding, activation steering, and prompting across three model families: OLMo2, Qwen2.5, and Llama3. Our analysis reveals distinct effects and trade-offs across techniques, highlighting the value of MAPLE for understanding and comparing KEE beyond aggregate accuracy.
0
stat.MLcs.LG Zhenisbek Assylbekov, Rustem Takhanov · Oct 14, 2026

Feldman (2021) gave an influential lower bound linking the cost of not fitting training labels to excess population risk via frequency-dependent coefficients $\tau_{n,\ell}$. This paper demonstrates that Feldman's central conditioning identity fails because the counts of other points convey information about the sampling distribution, and it provides an exact three-point counterexample together with an exponentially large multiplicative gap. The authors recover a valid bound within the same model by replacing the count-only coefficient with the exact full-sample posterior mean, which depends on both multiplicity and the complete frequency profile.

Feldman's memorization lower bound assigns a frequency coefficient based on how often a point appears in the training sample. We give an exact three-point counterexample to its conditional and averaged forms, using independent fair labels and a permutation-equivariant learner. A sequence of sampling priors makes the ratio of the proposed penalty to actual excess risk grow exponentially, ruling out a universal positive constant-factor repair. The priors vary with sample size, the offending profiles become rare, and both quantities vanish. Retaining the original model, we obtain an exact replacement using full-sample posterior probabilities. Under exchangeability, these depend on multiplicity and the complete frequency profile. We characterize the largest multiplicity-only coefficients valid for unrestricted learners. The original coefficient has the posterior-mean form of Good's exact frequency formula and is an occupancy-weighted average of the corrected coefficients.
0
cs.LGstat.ML Junlin Chen, Wenhao Wang, Junchao Xu et al. · Oct 14, 2026

Class-incremental graph learning suffers from catastrophic forgetting because models are repeatedly fine-tuned on new classes, causing severe recency bias where earlier sessions are misclassified as recent ones. MAEM addresses this by freezing a dedicated Mixture-of-Experts expert per session and routing test nodes via the reconstruction error of session-specific masked autoencoders, eliminating the need for task identities or a shared trainable router. To prevent unbounded model growth, it merges least-used experts via Git Re-Basin alignment followed by topology-aware pseudo-data distillation, achieving strong replay-free performance across four benchmarks.

Graph Neural Networks (GNNs) excel on graph-structured data but suffer catastrophic forgetting when classes arrive sequentially. In class-incremental graph learning, predictions are often dominated by recently learned classes, causing severe recency bias. Dynamic expansion can isolate knowledge across sessions, but many methods assume task-incremental settings and require task identities at test time to select branches or heads---information unavailable in class-incremental learning. Learnable routers offer an alternative, yet continual fine-tuning the router on new data can bias routing toward recent components. We propose MAEM, a replay-free, capacity-controlled dynamic mixture-of-experts framework for class-incremental graph learning. MAEM builds representations with Simplified Graph Convolution (SGC) and trains a dedicated expert for each new session, then freezes it to protect prior knowledge. For routing without task IDs or a shared trainable router, each expert uses a masked autoencoder (MAE); test nodes are assigned to the expert with the lowest reconstruction error, reducing routing-level recency bias. When the expert pool reaches a fixed capacity, MAEM merges the least-used experts via Git Re-Basin alignment and further distills them into a fused expert using topology-aware pseudo data synthesized from MAE distributions (DeepInversion-style), without storing or revisiting historical graph data. On four class-incremental graph benchmarks, MAEM mitigates recency bias, outperforms strong continual graph learning baselines, and keeps model capacity bounded under strict replay-free learning.
0
stat.MLcs.LG Eliot Beyler, Francis Bach · Oct 14, 2026

Fine-tuning pretrained diffusion and flow matching models for reward maximization tends to collapse modes or shift mass between distant regions. This paper introduces Weighted Adjoint Matching, which modifies the stochastic optimal control formulation of Adjoint Matching by placing a time-dependent weight $w(t)$ on the quadratic control cost---expensive early to preserve coarse semantics, cheap late to refine details and increase reward. The authors prove equivalence to maximizing reward under pathwise KL regularization, and demonstrate improved reward-diversity tradeoffs on Gaussian mixtures and MNIST.

We introduce Weighted Adjoint Matching, a method to finetune a pretrained diffusion or flow matching model given a differentiable reward. We derive it from Adjoint Matching, which reformulates finetuning as a stochastic optimal control problem with a quadratic control cost. Building on the observation that these models generate coarse features before fine details, we place a time-dependent weight on the control cost. A high cost early in the generative process preserves balance and diversity in the coarse features, which carry most of the semantics, while a low cost at the end lets the model increase the reward by refining details. We prove that our method is equivalent to maximizing the reward under time-varying Kullback--Leibler regularization on the generative paths. Experiments on Gaussian mixtures and on MNIST show a better tradeoff between reward and mode balance than Adjoint Matching.
0
cs.LGcs.CV Alan Baade, Kyle Sargent, Vincent Sitzmann et al. · Oct 14, 2026

Training good latent tokenizers for generative models remains poorly understood, especially how to scale the encoder itself. This paper proposes Relative Prediction Representations (RePR), which supervise patch features to factorize a pointwise mutual information matrix estimated by a conditional diffusion model. The result is a theoretically grounded, negative-sample-free objective that resists feature suppression and claims to be the first to demonstrate compute-optimal tokenizer encoder scaling, matching or beating pixel-space diffusion at equal FLOPs.

We introduce an approach for self-supervised representation learning designed to be used as a tokenizer for generative modeling. Our objective supervises a representation loss by factorizing the pointwise mutual information of the data, which we estimate by taking information differences between conditional predictions from a learned generative model. We justify our algorithm theoretically and empirically, demonstrating that our generative approach to representation learning avoids dimensional collapse and feature suppression in cases where commonly used methods like DINO and LeJEPA fail. Moreover, we show our representation learning technique is scalable; expending more compute leads to predictably better latent tokenization. For the first time, we perform a careful analysis of the combined flop efficiency of tokenizer and generative model and demonstrate that our approach is a compute-optimal improvement over direct diffusion for pixel-space generation, obtaining an FID-50K of 2.18 with JiT-L. Applied to ImageNet and the text-to-image dataset GPIC, we show tokenizer encoder scaling laws for diffusability, where extrapolating our approach achieves representations in 7x fewer FLOPs than DINOv2 on ImageNet and 8x fewer FLOPs than WebSSL on GPIC.
0
cs.LG Lukas Niehaus, Robin Rawiel, Ulf Krumnack · Oct 14, 2026

KANLab is a browser-based educational tool that adapts the TensorFlow Playground interface to Kolmogorov-Arnold Networks (KANs). It strips away modern performance-oriented additions like residual connections and learnable scaling so that displayed B-spline curves map directly to their trainable control points, letting users drag points and watch gradient descent correct them in real time. By making KAN internals tangible in a compact 2D setting, it aims to bridge the gap between novices and experts.

KANLab is a browser-based visualization tool designed to interactively teach the core concepts of Kolmogorov-Arnold Networks (KAN), forked from the widely successful TensorFlow Playground. To maximize educational clarity, KANLab omits residual connections and learnable scaling parameters found in modern KAN implementations, ensuring that displayed B-spline activation curves directly correspond to their underlying trainable control points. Users can inspect and modify activation functions, initialization schemes and observe gradient descent correcting manual disruptions. Furthermore, user can explore phenomena such as activation variance, interactive pruning, and re\-/training within a compact 2D machine learning setting. KANLab is open-source and available at https://kanlab.app/
0
cs.LG Daman Arora, Xuecheng Liu, Himanshu Gaurav Singh et al. · Oct 14, 2026

Standard reinforcement learning on verifiable rewards incentivizes language models to guess rather than admit uncertainty, because a wrong answer carries no penalty while abstention foregoes any chance of reward. This paper frames graceful failure as a three-outcome RL problem (Success \succ Fallback \succ Error), identifies why naive policy-gradient methods collapse to degenerate always-fallback policies, and proposes GracefulRL, a log-barrier-shaped objective that preserves gradient signal near saturation. The idea matters because coding and agentic mistakes can propagate into real harm, yet current RL pipelines train models to maximize success rate without regard for how they behave when success is unlikely.

Reinforcement learning makes language models better at solving tasks, but the same mechanism also rewards guessing. When success is the only way to earn reward, opting for a safe fallback is always worse than a long-shot attempt. When a wrong action can corrupt a database or ship broken code, that incentive can be costly. Therefore, we propose training models to fail gracefully: to preserve successful behavior whenever possible while learning to execute a safe fallback policy when acting would be unreliable. However, naively using policy gradients with intermediate reward for fallback behavior is brittle due to unfavorable optimization dynamics. We address this challenge with GracefulRL, which reshapes the optimization landscape to promote graceful failure. Importantly, this behavior emerges directly from outcome-level reward signals, without annotations specifying when to fall back or additional inference-time machinery such as a verifier. Across a range of tasks and scales, we train models that largely preserve their task-solving capability while selectively falling back when acting would be unreliable. More broadly, our results show that reinforcement learning can not only incentivize models to succeed, but also to reason about the limits of their capabilities and change course when they determine that success is out of reach.
0
cs.LGstat.ML Renjie Xie, Juncheng Yang, Aoting Hu et al. · Oct 14, 2026

Mixed-precision quantization for sparse MoE models needs more than expert rankings: NatScale introduces contrast calibration, a principled way to control how strongly sensitivity scores compete for storage. It applies a model-specific power transformation $w_i = g_i^{\beta_\tau}$ so that normalized squared weights achieve a target Jeffreys divergence $\mathcal{J}(q(\beta_\tau), u) = \tau$, preserving rank order while setting a common information-theoretic scale. Combined with joint gate/up/down error measurement and spectral structural priors, it allocates bit-widths under a global byte budget via LP relaxation and rounding.

Mixture-of-experts (MoE) language models activate only a few experts per token, yet all expert weights must be stored. Mixed-precision quantization reduces this storage cost by assigning higher bit-widths to sensitive experts and lower bit-widths to resilient ones, but expert rankings alone do not determine how strongly to weight their errors. To address this gap, we propose NatScale, a contrast-calibrated allocation framework for mixed-precision MoE quantization. It uses a model-specific power transformation to calibrate gradient-based expert weights to an information-theoretic contrast target while preserving rankings. We jointly evaluate each expert's gate, up and down configurations to estimate its output error. We then combine these errors and calibrated weights with an existing prior based on weight spectra, variance and precision decay to select configurations under a storage budget. Across three models, NatScale achieves the lowest native-to-quantized KL divergence and the highest top-token agreement in all nine settings against five allocation baselines. It also yields the lowest perplexity in seven settings. These findings highlight the importance of objective-aware precision allocation.