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.
1579 papers
Trending mixes fresh papers with community signal.
0
cs.LG Siddhant Setia, Junichi Suzuki, Tadashi Nakano · Oct 14, 2026

Molecular communication simulators are computationally expensive, so machine learning models are often trained to estimate performance metrics like round-trip time, but protocol evaluation proceeds sequentially from simple to complex settings and causes catastrophic forgetting when models are fine-tuned. This paper proposes Affine Memory Replay (AMR), a continual learning method that stores past simulation samples and synthesizes additional replay samples through affine interpolation and extrapolation to regularize a small neural network. AMR aims to retain knowledge from earlier experimental settings while adapting to new ones, offering a practical way to amortize simulation costs in molecular communication research.

This paper considers reliable in-sequence molecular communication, where bio-nanomachines exchange molecule-encoded information in a specified order, and investigates a new simulation-based method to incrementally evaluate communication protocols by gradually increasing the complexity of experimental settings. The proposed method, called Affine Memory Replay (AMR), builds a continual learning (CL) model to estimate communication performance between bio-nanomachines. AMR stores past simulation samples and synthesizes additional replay samples through interpolation and extrapolation between past samples. Through incremental performance estimation tasks, it replays both past and synthetic samples to regularize model updates, helping the model adapt to new experimental settings while retaining knowledge from earlier ones. Experimental results show that AMR significantly outperforms existing CL methods in maintaining overall estimation accuracy across sequentially learned estimation tasks. It effectively balances memory stability and learning plasticity under moderate training cost.
0
cs.LGcs.SYeess.SY Samuela Rokocakau, Panagiotis Kakosimos, Giulia Tresca et al. · Oct 14, 2026

This paper tackles offline parameter identification for a non-ideal DC-DC buck converter using Physics-Informed Neural Networks (PINNs). The core idea is a staged training schedule that progressively unfreezes physical parameters—inductance, capacitance, and parasitic resistances—to improve convergence and interpretability. The work matters because accurate component-level estimates enable condition monitoring and predictive maintenance without intrusive frequency-response tests.

A staged physics-informed neural network (PINN) is proposed for identifying the inductance, capacitance and parasitic resistances of a DC buck converter from time-domain data. Gradients are released in the order of L to C to rL to rC, a sequence derived from sensitivity analysis. The five-stage schedule recovers L and C to within 5% and the series ESR to 10%, while a three-stage variant attains the same accuracy 11% faster. An 84 factorial sweep confirms that the trained PINN remains accurate for simultaneous 8% perturbations of all four components, achieving a waveform RMSE of 2.6 x 10-2 over 5 ms-an order-of-magnitude better than a simultaneous-training baseline.
0
cs.LG Weixin Wang, Wei Deng, Anderson Schneider et al. · Oct 14, 2026

Training-free inference-time scaling aims to improve LLM reasoning without updating model weights or relying on external reward models. Existing reward-free methods sharpen the full sequence distribution, which can suppress diverse reasoning paths that support the same correct answer, and suffer from path collapse in Sequential Monte Carlo samplers. This paper proposes Decoupled Marginal Sharpening (DMS), a two-stage framework that separately controls reasoning-path concentration and answer-marginal consensus via independent exponents, paired with a novel Fully Adapted Annealed SMC sampler. The authors report the highest mean pass@1 across three base models and four benchmarks, including mathematics, science, and code tasks.

We study how to improve LLM reasoning at inference time without updating model weights or using external reward models. Recent reward-free power-sampling methods define the sequence-level power target $\pi_\alpha(y\mid x)\propto p_0(y\mid x)^\alpha$, where $\alpha>1$, which increases the relative probability of completions that are already likely under the base model. Existing methods use Metropolis--Hastings or Sequential Monte Carlo (SMC) to approximately sample from this target. However, two challenges arise at different levels. At the target level, full-sequence sharpening can concentrate probability on a few exact reasoning--answer trajectories, suppressing alternative paths that support the same answer. At the sampler level, finite-particle SMC can suffer from path collapse when resampling repeatedly duplicates a few high-weight prefixes. To address these challenges, we propose Decoupled Marginal Sharpening (DMS), which separately controls how strongly the model favors particular reasoning paths and how strongly it favors final answers after aggregating support across those paths. At the target level, DMS first samples from a sharpened reasoning distribution, then marginalizes the model's answer probabilities over the resulting weighted reasoning population and sharpens the induced answer marginal. Separate exponents allow mild reasoning sharpening to encourage exploration of diverse paths, followed by stronger answer-marginal sharpening to concentrate probability on answers with greater aggregate support across those paths. At the sampler level, we develop Fully Adapted Annealed SMC, which combines an annealed sequence of power targets with locally fully adapted proposals and predictive ancestor resampling. Experiments on MATH500, GPQA-Diamond, HumanEval, and AIME24/25 across three base models show that DMS achieves the highest mean pass@1 among the compared methods in all tasks.
0
cs.LG Jon Sporring · Oct 14, 2026

This critical review reframes interactive segmentation, active learning, and machine teaching through a shared information-theoretic lens. It separates target information gain from predictive distortion reduction and human effort, exposing why predicting a teacher's clicks does not by itself improve reusable learning. By formalizing corrective interactions as nested memory records, the paper shows that current-mask refinement and reusable-model training are distinct objectives requiring different evaluation assumptions.

When does a human correction become useful learning, and what establishes better segmentation at matched human effort? This critical narrative review follows feedback through its selection, learner update, and evaluation. It compares active learning, machine teaching, interactive segmentation, and sequential predictive evaluation using a common mathematical account of target information, predictive error, response predictability, and human work. An exact Bayesian update converts evaluation-label information into expected log-loss improvement; an excess-risk decomposition explains what an approximate learner can lose. Conditional information decompositions separate the roles of selected locations, labels, and displayed context. Empirical comparisons show why response protocols, update mechanisms, and effort denominators matter, distinguishing current-mask refinement from reusable learning and simulated interaction from human evidence. We identify what logs can reveal, what requires independent evaluation, and what remains model-dependent. The resulting research priorities are to test what selection reveals, whether using it improves assimilation, and whether the resulting interaction policy benefits segmentation at matched effort. These are distinct claims: predicting a teacher's responses need not identify human intentions or establish better learning. No new acquisition algorithm or empirical superiority is claimed.
0
cs.LG Shigeyasu Uno · Oct 14, 2026

CktHGVAE tackles analog/mixed-signal CMOS circuit automation by learning a joint latent manifold of design parameters and transistor-level operating states. The core idea is a symmetric heterogeneous graph VAE that treats SPICE-derived state variables and design variables equally as inputs and outputs, enabling closed-loop design verification without a separate surrogate model. This is significant because prior graph learning approaches for analog circuits largely rely on macromodel abstractions or topology generation, leaving fine-grained operating-state modeling underexplored.

We propose CktHGVAE, a heterogeneous graph variational autoencoder for transistor-level analog/mixedsignal CMOS circuits that takes not only design variables but also state variables as its inputs and outputs. With a single hyperparameter configuration without topology-specific tuning, comparison between model-generated and SPICE-simulated circuit operational states gives median R2 values above 0.9 for typical single-ended voltage amplifiers and the CMOS inverter, and 0.81-0.96 for differential amplifiers with larger datasets. Latent-space optimization in the PCA-reduced latent space with Bayesian optimization, evolutionary search, random search, and quasi-exhaustive search is demonstrated for a common-source voltage amplifier with NMOS input and resistive load, where the objective function is defined as a penalty for deviation from required circuit specifications. For each search strategy, 93.3 - 100% of all design candidates suggested by the model satisfy the required circuit specifications at the decoder output, and SPICE simulations verify that 90 - 100% of them indeed satisfy the requirements. The model pretrained on a batch of topologies can be transferred to unseen topologies without retraining at all, with zero-shot reconstruction median R2 values of 0.72-0.90 in three of four scenarios, including transfer from single-ended to differential topologies, although the transfer fidelity depends on the choice of pretraining topologies. Our demonstrations suggest that the proposed model can potentially serve as a basis for AMS CMOS circuit design automation.
0
cs.LG Yaomengxi Han, Debarghya Ghoshdastidar · Oct 14, 2026

This paper asks how shallow transformers learn on unbounded-length inputs and whether layer-wise training can be decoupled. It proves that a one-step attention-then-output schedule fails to learn the occurrence language, while a three-phase output-attention-output schedule succeeds in both population and finite-sample settings. The finding is significant because it formally explains why attention weights often lag behind the output layer during training, challenging the feedforward intuition that feature learning must come first.

The success of transformers is attributed to several factors, including their capacity to express formal languages over variable-length inputs, and ability of the attention mechanism to focus on relevant input features. However, there is limited understanding of the training dynamics of shallow transformers, specifically how to decouple the learning of the attention and the output (feedforward) layer. A natural extension of the theory of one-step gradient feature learning in feedforward networks suggests that one could first learn the attention weights with a single gradient step, and then train the output layer. We show that this fails for a shallow transformer trained to recognize a star-free language from unbounded-length input. We prove that the transformer can be trained to recognize the language with a three-phase gradient descent: the output layer is updated first, followed by an attention update, and the output layer is fine-tuned in the third phase. In contrast, the language cannot be learned by training only the attention or output layer, or by a single attention-layer update followed by finetuning the output layer. We empirically show that ``attention learning lagging output layer learning" is beneficial in other language recognition, regression, and in-context learning tasks as well.
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.