We develop a principled framework for analyzing and designing noise schedules in diffusion models. We show that one can recast this design problem as an optimal control problem, whose state is the Fisher information of the diffusion process, which evolves according to an ODE, and whose control input is the noise schedu...
Seo Taek Kong, Weina Wang, R. Srikant· 0 citations
This work addresses the problem of learning directed acyclic graphs (DAGs) from nodal observations adhering to a linear structural equation model. DAG structure learning is a central task in signal processing, machine learning, and causal inference, but it remains challenging because acyclicity is a global combinatoria...
Samuel Rey, Madeline navarro, Santiago Segarra et al.· 0 citations
The rapid advancement toward long-context reasoning and multi-modal intelligence has made the memory footprint of the Key-Value (KV) cache a dominant memory bottleneck for efficient deployment. While the established per-channel quantization effectively accommodates intrinsic channel-wise outliers in Key tensors, its ef...
Zunhai Su, Rui Yang, Chao Zhang et al.· 0 citations
Test-time scaling is an important mechanism for improving large language models, especially on tasks with deterministic verifiers. Code translation is a canonical example: the source program constrains valid outputs, while compilers, type checkers, and behavioral checks provide exact pass/fail feedback. Outcome-only ap...
Tianyang Zhou, Somesh Jha, Mihai Christodorescu et al.· 0 citations
Reach audiences
Advertise in front of researchers, engineers, and readers.
In partially observable reinforcement learning (RL), a decision often depends on a cue seen many steps earlier, and training at such horizons is costly. Ideally, a policy trained on short horizons would remain optimal at arbitrarily long ones, making the training horizon a matter of convenience. We formalize this prope...
Asad Bakija, Florent De Geeter, Julien Brandoit et al.· 0 citations
Large language model agents increasingly rely on external skills to solve complex tasks, where skills act as modular units that extend their capabilities beyond what parametric memory alone supports. Existing skill-based RL methods assume external skills either accumulate as persistent guidance or internalized into the...
Junhao Shen, Teng Zhang, Xiaoyan Zhao et al.· 0 citations
We propose a novel doubly robust neural mean-embedding framework for proxy causal learning (PCL) for continuous and structured treatments. PCL uses proxy variables and bridge functions to identify causal effects under unobserved confounding: either an outcome bridge or a treatment bridge can identify the target, while...
Bariscan Bozkurt, Alexandre Galashov, Dimitri Meunier et al.· 0 citations
A distilled student can copy its teacher's mistakes along with its skills. On-policy distillation (OPD), used in recent post-training recipes, has the teacher grade every token of the student's own outputs. Which teacher errors does it pass on? We build ErrOPD, a testbed whose teachers are fine-tuned to close a JSON li...
We study Slowly Annealed Langevin Dynamics (SALD), a sampler for tracking a path of moving target distributions and approximating the terminal target through time slowdown. We establish non-asymptotic convergence guarantees via a KL differential inequality, showing that slowdown improves tracking through contraction of...
Atsushi Nitanda, Dake Bu, Yueming Lyu et al.· 0 citations
PACEvolve++ is introduced, an advisor-model reinforcement learning framework for test-time policy adaptation in evolutionary search agents, which outperforms the state-of-the-art evolutionary search framework with frontier models, achieving faster convergence and stabilizing test-time training during evolutionary searc...
Minghao Yan, Bo Peng, Benjamin Coleman et al.· arXiv.org· 4 citations
Diffusion-based posterior samplers draw from measurement- or reward-conditioned posteriors using pretrained diffusion priors. Though widely used for inverse problems, their output is biased even with exact prior scores. First, we quantify this bias for the DPS algorithm, the base method for a family of samplers, by int...
William Porteous, Advait Parulekar, Sebastien Motsch et al.· 0 citations
Subgraph pattern detection is a fundamental graph mining task central to anti-money laundering (AML) in finance. However, state-of-the-art graph neural network (GNN)-based methods assume centralized graph access, impractical when transaction data cannot be shared across institutions. In practice, graphs are distributed...
A machine-learning approach called CheMeleon provides a practical way to develop new drugs or other molecules with much less detailed, expensive experimentation.
Research Manager Mike Walker reflects on a career spanning hacker culture, cybersecurity, and AI, sharing lessons about trust, teamwork, leadership, and why solving difficult problems requires curiosity, judgment, agility—and an understanding of the rules. The post Learn the rules: The hacker mindset appeared first on Microsoft Research.
Learn how Google Antigravity and our Cheminformatics Skills accelerate drug discovery workflows, search compounds, and propose novel rigidified macrocycles.
We use cookies to run the site and, with your consent, for analytics and to show ads.
See our Cookie Policy.