Skip to content

Category

machine learning

15,178 papers

#machine learning Preprint Sep 2026

ID Balancing: Stable Training of Extremely Sparse MoE via PID-Based Load Control

Scaling Large Language Models (LLMs) via Mixture-of-Experts (MoE) enables massive parameter growth with nearly constant per-token computation. However, further scaling the parameter count requires increasingly sparse routing, where expert load imbalance becomes more severe. This imbalance reduces parameter utilization...

Peng Jin, Zi-Han Qiu, Ze-Kun Wang et al. · 0 citations
#machine learning Preprint Sep 2026

Characterizing High Bandwidth Flash for LLM Serving

Large language model (LLM) serving requires substantial memory to store model weights and KV caches. As models grow larger and contexts become longer, memory capacity and bandwidth increasingly become bottlenecks for serving performance. Agentic workloads compound this pressure through repeated interactions over growin...

Zack Yu, Chloe Wong, C. Hooper et al. · 0 citations
#machine learning Preprint Open access Oct 2026

CDMD: A Cross-Dataset Mixed-Type Diffusion Model for Tabular Data

Generative models for tabular data are typically trained separately for each dataset, limiting knowledge transfer and requiring the storage of many specialized models. In this paper, we introduce CDMD, a tabular diffusion model trained jointly across heterogeneous datasets with different schemas and variable numbers of...

Mohamed Amine Ketata, Maximilian Schambach, Stephan G\"unnemann · 0 citations
#machine learning Preprint Sep 2026

Prequential E-Values for Selected-GP Near-Optimality Certificates

When optimizing an expensive black-box function sequentially, as in hyperparameter optimization, we may want to stop once the best evaluated value is certified within $\varepsilon$ of the global optimum. Such a certificate needs two ingredients: a lower confidence bound for the selected value and an upper confidence en...

A. Tavory, Noa Cohen · 0 citations
#machine learning Preprint Open access Oct 2026

The Row Normalization Puzzle in Muon

This paper examines how row-wise renormalization affects Muon, focusing on the gap between NorMuon's worst-case guarantees and its practical performance (Li et al.). Despite its growing adoption and promising performance in large language model (LLM) pretraining, NorMuon's worst-case guarantees remain poorly understood...

Jiayu Zhang, Tianyi Lin · 0 citations
#machine learning Preprint Sep 2026

Learning Infinite-Horizon Average-Reward CMDPs via State Augmentation

We study infinite-horizon average-reward constrained Markov decision processes (CMDPs) under the weakly communicating assumption. Existing high-probability guarantees for this setting either require computationally inefficient algorithms or have suboptimal dependence on the number of interactions $T$. We propose, to th...

Kihyun Yu, Seoungbin Bae, Dabeen Lee · 0 citations
#machine learning Preprint Sep 2026

Shared Phase and Retention Control for Efficient Adaptive Spectral Recurrence

As new evidence arrives, a sequence model must update what it remembers and how memory influences predictions. While Transformers incur computation and cache costs scaling with context length, fixed-state recurrent models offer constant-memory inference. However, linear and spectral recurrences traditionally rely on st...

Wen-Tao Wang, Heng-Yu Zhong, Yun-Han Jiang et al. · 0 citations
#machine learning Preprint Sep 2026

Parameter symmetries determine representational geometry in overparameterized nonlinear networks

Representations are routinely used across machine learning, psychology, and neuroscience to draw inferences about the computations of biological and artificial systems. Such inferences presume a meaningful link between representational geometry and the computation being performed. For artificial neural networks, howeve...

Marvin Theiss, Lukas Braun, Andrew M. Saxe et al. · 0 citations
#machine learning Preprint Sep 2026

SparseEngine: Sparse-First Inference Engine

Long-context LLM agents accumulate interaction histories that strain KV-cache memory and attention computation. Although sparse attention reduces these costs, heterogeneous cache representations and workflows hinder integration with existing inference engines, while prior sparse-serving abstractions support only specif...

Jitai Hao, Quan-Sheng Gu, Qiang Huang et al. · 0 citations
#machine learning Preprint Open access Oct 2026

Argus: A Real-EKS Study of When Predicting Spot Interruptions Beats Simple Checkpointing

Elastic Compute Cloud (EC2) Spot is 60% to 90% cheaper than On-Demand but can be reclaimed on just a 2-minute notice; for expensive multi-node training this loss can be severe, with one reclaim costing hours of synchronous progress. We build Argus, a Kubernetes operator, and ask empirically, on a CIFAR-10 testbed, when...

Angshuman Chakravertty, MD Rayyan · 0 citations
#machine learning Preprint Sep 2026

The Missing Coefficients: Bayesian Pairwise Merging for Model Personalization

How can we personalize a shared expert library from a user's pairwise choices? Prior work can realize different reward trade-offs by merging reward-specialized experts, given a vector of trade-off weights. In practice, users can more naturally choose between outputs than specify numerical weights. The challenge is ther...

Ya-Ling Shen, Tong-Tong Wu, Si-Yuan Yan et al. · 0 citations
#machine learning Preprint Sep 2026

A Rank Graduation metric for Algorithmic fairness

Fairness assessment in algorithmic decisions that affect individuals, such as credit scoring, often relies on parity measures calculated at the aggregate group level. Such measures may not reveal which individuals experience unfairness or which explanatory factors contribute to it. In this paper, we propose a rank-base...

Dalia Atif, Paolo Giudici · 0 citations

From tech blogs

See all →
Microsoft Research Blog Oct 12, 2026

Simulating student learning for better AI tutoring

StudentSim is a research system for evaluating and improving AI tutoring tools before they are tested with learners. It models how individuals may engage with tutors, giving feedback to improve tutor performance across subjects like chess, writing, and mathematics. The post Simulating student learning for better AI tutoring appeared first on Microsoft Research.

We use cookies to run the site and, with your consent, for analytics and to show ads. See our Cookie Policy.