Machine learning

Thursday, October 158 papers

arXiv

Are We Overthinking One-Step Generation?

Woobin Park, Jaeyeon Kim, Yilun Du et al.

Diffusion and flow matching are now dominant frameworks for generative modeling in continuous domains, but they typically require multiple network evaluations at…Diffusion and flow matching are now dominant frameworks for generative modeling in continuous domains, but they typically require multiple network evaluations at inference.

Preprint: new model
arXiv

Learning to Compute Is Not Learning to Retain or Reuse in Looped Transformers

Peilong Zhou, Hongtao Zhang, Yulong Huang et al.

Looped transformers compute by applying one block repeatedly, so they can run for more loops than they were trained with.Looped transformers compute by applying one block repeatedly, so they can run for more loops than they were trained with. Recurrent reasoners are known to overthink: iterating past the trained depth can turn a correct answer into…

Preprint: empirical study
arXiv

One-Step Prediction Fidelity Is a Poor Proxy for Planning Performance in JEPA World Models

Joshua Cheung, Zhang He

Latent world models are routinely selected, checkpointed and ablated on one-step prediction loss, on the assumption that a model which predicts the next latent better is…Latent world models are routinely selected, checkpointed and ablated on one-step prediction loss, on the assumption that a model which predicts the next latent better is a better model.

Preprint: empirical study
arXiv

Verifier-Grounded Agentic RL for Provably Correct Code

George Ma, Yuhao Zhang, Bing He et al.

Reinforcement learning for code usually rewards passing tests, which a model can do while remaining wrong on the inputs the tests do not cover.Reinforcement learning for code usually rewards passing tests, which a model can do while remaining wrong on the inputs the tests do not cover.

Preprint: new model
arXiv

Dimension-Free Scaling Laws for Size Generalization: An Exponential Separation

Behrooz Tahmasebi, Melanie Weber

Modern machine learning models increasingly operate on variable-size structured data, such as sets and graphs.Modern machine learning models increasingly operate on variable-size structured data, such as sets and graphs. A central requirement is size generalization: the ability of a model trained on inputs of one size to generalize to…

Preprint: theory
arXiv

Does Pretraining Teach Chemical Language Models Chemistry?

Leif Sieben, Yoel Zimmermann

Chemical language models (CLMs) are pretrained on large corpora of molecules with the aim of learning representations that transfer to diverse downstream tasks.Chemical language models (CLMs) are pretrained on large corpora of molecules with the aim of learning representations that transfer to diverse downstream tasks.

Preprint: empirical study
arXiv

AI Models Can Provably Hide Arbitrary Capabilities

Andis Draguns, Sumeet Ramesh Motwani, Raymond Douglas et al.

Capability evaluations aim to surface dangerous model behaviors before deployment, but their reliability depends on the assumption that hidden capabilities can be…Capability evaluations aim to surface dangerous model behaviors before deployment, but their reliability depends on the assumption that hidden capabilities can be elicited.

Preprint: method
arXiv

Do LLMs Learn from Rewards in Context? : Rethinking the role of reward in In-Context RL

Minchan Kwon, Seunghee Koh, Sunghyun Baek et al.

LLM agents increasingly improve at inference time by accumulating experience in context rather than by updating parameters.LLM agents increasingly improve at inference time by accumulating experience in context rather than by updating parameters. This process is often described as in-context reinforcement learning (ICRL).

Preprint: empirical study

Wednesday, October 1410 papers

arXiv

FlashMask-3: Efficient and Expressive Mask-Aware Distributed Attention

Guoxia Wang, Qianyue He, Siming Wu et al.

Long-context training of large language and multimodal models couples structured attention masks (document, prefix-LM, share-question, sliding-window, block-sparse) with…Long-context training of large language and multimodal models couples structured attention masks (document, prefix-LM, share-question, sliding-window, block-sparse) with multi-node context parallelism, making the mask a dominant…

Preprint: system
arXiv

Revisiting the 16x Pretraining Compute Gap between Diffusion and Autoregressive Language Models

Guanghan Wang, Bing He, Zhan Shi et al.

Diffusion language models (DLMs) have attracted wide attention for their fast, parallel decoding, but they are believed to be far less efficient to train than…Diffusion language models (DLMs) have attracted wide attention for their fast, parallel decoding, but they are believed to be far less efficient to train than autoregressive language models (ARMs): a widely cited study reports…

Preprint: empirical study
arXiv

HarnessTTT: Learn to Optimize Harness at Test-Time for Scientific Discovery

Yingzi Ma, Yulong Cao, Nanxi Li et al.

Scientific discovery with language models requires finding the best valid solution within a finite computational budget.Scientific discovery with language models requires finding the best valid solution within a finite computational budget. We study how feedback from this search can improve the procedure used to conduct it.

Preprint: new model
arXiv

The Surprising Expressivity of a Single Neuron with Autoregressive Chain-of-Thought

Roey Magen, Nirmit Joshi, Gal Vardi

The success of autoregressive language models trained with Chain-of-Thought~(CoT) supervision raises a fundamental question: How simple can a next-token predictor be…The success of autoregressive language models trained with Chain-of-Thought~(CoT) supervision raises a fundamental question: How simple can a next-token predictor be while still enabling the efficient simulation of general…

Preprint: theory
arXiv

Davinci-Tool: Scaling Agentic Tool-Use Supervision with Verifiable Executable Environments

Xiaoyang Yuan, Ji Zhao, Xingyu Hu et al.

Reliable agentic tool use requires training data that pair diverse multi-step workflows with executable evidence that the requested outcomes were achieved.Reliable agentic tool use requires training data that pair diverse multi-step workflows with executable evidence that the requested outcomes were achieved.

Preprint: new model
arXiv

Mean Accuracy Is the Wrong Estimand for a Release Decision

Jaewon Choi, Jihye Kang, Namhyuk Ahn

A release decision about a fine-tuned model is an operational commitment about the next rerun of its training recipe.A release decision about a fine-tuned model is an operational commitment about the next rerun of its training recipe. Yet standard benchmark reporting gives only the mean accuracy of past seeds.

Preprint: method
arXiv

What Do Sparse Autoencoder Feature Discovery Scores Identify? From Activations to Interventions

Yifeng Guo, Yanran Li, Xiang Chen

Sparse autoencoder (SAE) features are typically discovered from how their activations co-vary with behaviour, yet they are often used through interventions such as…Sparse autoencoder (SAE) features are typically discovered from how their activations co-vary with behaviour, yet they are often used through interventions such as steering or ablation.

Preprint: empirical study
arXiv

Beyond Adjacent Layers: Graph-Guided Layer Fusion for Compressing Large Language Models

Muhammad Usama, Yeoreum Jung

Existing depth-compression methods for large language models (LLMs) score layers individually or group only adjacent or contiguous layers, implicitly assuming that…Existing depth-compression methods for large language models (LLMs) score layers individually or group only adjacent or contiguous layers, implicitly assuming that functional redundancy is local.

Preprint: method
arXiv

Swarm Telepathy: Prefill-Free KV-Cache Transfer Across Families

Abdulrahman Alabdulkareem, Irene Tenison, Yang Zhang et al.

LLM multi-agent systems split work across agents that communicate through text. An agent summarizes its findings, another agent receives the text.LLM multi-agent systems split work across agents that communicate through text. An agent summarizes its findings, another agent receives the text.

Preprint: method
arXiv

PATTERN Can Be Solved from Memory: Generator-Aware Baselines for Synthetic Node Classification

Amin Najafgholizadeh, Mohsen Afsharchi

PATTERN, a synthetic node-classification benchmark still used to rank graph networks, plants a random subgraph in a random host graph and asks which nodes belong to it.PATTERN, a synthetic node-classification benchmark still used to rank graph networks, plants a random subgraph in a random host graph and asks which nodes belong to it.

Preprint: empirical study

Monday, October 126 papers

arXiv

Tabular Foundation Models Are Effectively Shallow

Irene Cannistraci, Julia E. Vogt

In-context-learning Tabular Foundation Models (TFMs) are uniquely suited to scenarios with limited resources or privacy constraints, as they solve new tasks without…In-context-learning Tabular Foundation Models (TFMs) are uniquely suited to scenarios with limited resources or privacy constraints, as they solve new tasks without additional training.

Preprint: method
arXiv

Generalized Proximal Policy Optimization: Unifying LLM RL Algorithms from the Perspective of Divergence

Zhicheng Cai, Xinyuan Guo, Hanlin Wu et al.

Reinforcement learning has become the foundational paradigm for enhancing the reasoning capabilities of LLMs.Reinforcement learning has become the foundational paradigm for enhancing the reasoning capabilities of LLMs. Yet the dominant approach, Proximal Policy Optimization (PPO), relies on a geometrically degenerate clipping proxy that…

Preprint: method
arXiv

Critical Mass: Structure-Disjoint Splits Drift With the Libraries They Are Built On

Tomo Oga, Devesh Shah, Cailum M. K. Stienstra et al.

Molecular machine learning evaluates out-of-distribution performance with structure disjoint splits, a construction rule fixed once at the scale its library happened to…Molecular machine learning evaluates out-of-distribution performance with structure disjoint splits, a construction rule fixed once at the scale its library happened to have: threshold a pairwise distance, then assign whole…

Preprint: method
arXiv

Kronecker Product Attention: Memory-Efficient KV Compression via Structured Factorization

Bahareh Nikpour, Hossein Rajabzadeh, Maryam Dialameh et al.

Efficient autoregressive generation is increasingly limited by the memory footprint of the key-value (KV) cache, which grows linearly with sequence length, batch size…Efficient autoregressive generation is increasingly limited by the memory footprint of the key-value (KV) cache, which grows linearly with sequence length, batch size, and model depth.

Preprint: method

Friday, October 95 papers

arXiv

Realistic delays shrink gains from adapting forecast models

Accounting for realistic delays left published adaptation methods improving by at most half a percent over frozen models across four of five benchmarks.Accounting for realistic delays left published adaptation methods improving by at most half a percent over frozen models across four of five benchmarks.

Drift overview of the pretraining/validation/online splits: rolling one-week mean and standard deviation per dataset.
Preprint: empirical study
arXiv

Early Signatures of Memorization in Diffusion Models via Basin Geometry and Cyclic Denoising

Nikhil Verma, Siddharthan Dileep, Anoop Singh et al.

Diffusion models generalize early in training and later reproduce individual training samples. Standard tests detect memorization only once one-shot generation produces…Diffusion models generalize early in training and later reproduce individual training samples. Standard tests detect memorization only once one-shot generation produces near-copies, leaving a released model unaudited until its…

Preprint: empirical study
arXiv

NanoProof: Open and Efficient Automated Theorem Proving in Lean 4

Matěj Kripner, Milan Straka

We introduce NanoProof, to our knowledge the first factorized execution-guided theorem prover in Lean 4 whose training data, extraction tooling, training pipeline, and…We introduce NanoProof, to our knowledge the first factorized execution-guided theorem prover in Lean 4 whose training data, extraction tooling, training pipeline, and weights are all released, making it end-to-end reproducible…

Preprint: new model
arXiv

When Routing Reveals Membership: Privacy Leakage from MoE Router Telemetry

Yixin Tan, Jiayang Liu, Lu Sun et al.

Mixture-of-Experts (MoE) language models produce routing information during inference that may be logged or exposed for monitoring, debugging, load analysis, and safety…Mixture-of-Experts (MoE) language models produce routing information during inference that may be logged or exposed for monitoring, debugging, load analysis, and safety auditing.

Preprint: method

Thursday, October 89 papers

arXiv

RSIGym: A Flexible Environment for Recursive Self-Improvement

Fanqing Meng, Lingxiao Du, Haocheng Lu et al.

Recursive self-improvement requires carrying accepted changes into later improvement cycles, while studying agent-proposed changes also requires substantial research…Recursive self-improvement requires carrying accepted changes into later improvement cycles, while studying agent-proposed changes also requires substantial research infrastructure.

Preprint: system
arXiv

DSReg: Provably Recovering Individual World Latents without Reconstruction

Yujia Zheng, David Klindt, Randall Balestriero et al.

Methods that recover individual latent variables of the world, from nonlinear ICA to dictionary learning and causal representation learning, anchor the latents to…Methods that recover individual latent variables of the world, from nonlinear ICA to dictionary learning and causal representation learning, anchor the latents to observations through reconstruction, auxiliary supervision, or…

Preprint: method
arXiv

LayerRoPE: Dynamic Depth-wise Magnitude & Angular Superposition

Shikhar Srivastava, Christopher Kanan

As data propagates through a Transformer, the norm of its hidden states grows by orders of magnitude with depth, a phenomenon framed as 'curse of depth' and nearly…As data propagates through a Transformer, the norm of its hidden states grows by orders of magnitude with depth, a phenomenon framed as 'curse of depth' and nearly universally treated as a pathology to be suppressed.

Preprint: method
arXiv

Breaking Adversarial Transferability in Fine-Tuned Speech Recognition

Mojtaba Nafez, Aref Mousavi, Mohammad Ebrahim Mahdavi et al.

Many organizations fine-tune publicly available pretrained Automatic Speech Recognition (ASR) models and deploy them in black-box settings, assuming limited access…Many organizations fine-tune publicly available pretrained Automatic Speech Recognition (ASR) models and deploy them in black-box settings, assuming limited access provides protection.

Preprint: method
arXiv

Are We Really Benchmarking Forecasting Models? The Impact of Preprocessing on Time Series Performance

Guilherme Afonso Galindo Padilha, Paulo Salgado Gomes de Mattos Neto, Rafael Menelau Oliveira e Cruz

While established literature underscores the pivotal role of preprocessing in forecasting accuracy, this stage remains largely overlooked in current research.While established literature underscores the pivotal role of preprocessing in forecasting accuracy, this stage remains largely overlooked in current research.

Preprint: empirical study
arXiv

Removing Information Content Does Not Certify Tamper Resistance in Open-Weight Models

Domenic Rosati, Alessa Carbo, Ali Dadsetan et al.

Does removing harmful information make open-weight models resistant to fine-tuning attacks? We show that mutual information at release alone cannot universally certify…Does removing harmful information make open-weight models resistant to fine-tuning attacks? We show that mutual information at release alone cannot universally certify slow recovery.

Preprint: theory
arXiv

Sample-Optimal Estimation of the Fr'echet Inception Distance

Ziyun Chen, Jerry Li, Kevin Tian et al.

The Fr'echet Inception Distance (FID) is widely used to evaluate generative models, but its empirical plug-in estimator suffers from finite-sample bias [BSAG18, CF20].The Fr'echet Inception Distance (FID) is widely used to evaluate generative models, but its empirical plug-in estimator suffers from finite-sample bias [BSAG18, CF20].

Preprint: method

Wednesday, October 72 papers

arXiv

How Many Independent Samples Does a Satellite Image Contain? Generalization Bounds for Spatially Dependent Data

Robin Young

Machine learning classifiers for remote sensing imagery are typically evaluated as though every pixel were an independent sample.Machine learning classifiers for remote sensing imagery are typically evaluated as though every pixel were an independent sample. Spatial autocorrelation violates this assumption, since neighboring pixels carry redundant…

Preprint: theory
arXiv

DecepEval: A Benchmark for Evaluating Deception in LLM Agents

Yiming Xu, Hongyue Yu, Beihua Yang et al.

As large language model (LLM) agents become increasingly autonomous, they may pursue task performance through deception, raising concerns about their reliable deployment.As large language model (LLM) agents become increasingly autonomous, they may pursue task performance through deception, raising concerns about their reliable deployment.

Preprint: benchmark or dataset