Skip to content

Category

data science

3,093 papers

SHAKE-GNN: Scalable Hierarchical Kirchhoff-Forest Graph Neural Network

This work introduces SHAKE-GNN, a novel scalable graph-level GNN framework based on a hierarchy of Kirchhoff Forests, a class of random spanning forests used to construct stochastic multi-resolution decompositions of graphs, enabling flexible trade-offs between efficiency and performance.

Zhipu Cui, J. Lutzeyer · 0 citations

Integrating attention into explanation frameworks for language and vision transformers

The empirical evaluations on standard benchmarks and in a comparison study with widely used explanation methods show that attention weights can be meaningfully incorporated into the studied XAI frameworks, highlighting their value in enriching transformer explainability.

Marte Eggen, Jacob Lysnæs-Larsen, Inga Strümke · 1 citation

Adaptive teachers for amortized samplers

The teacher, an auxiliary behavior model, is trained to sample high-loss regions of the student and can generalize across unexplored modes, thereby enhancing mode coverage by providing an efficient training curriculum.

Minsu Kim, Sanghyeok Choi, Taeyoung Yun et al. · 27 citations · ⚡6

Understanding Deep Learning via Notions of Rank

It is established that gradient-based training can induce an implicit regularization towards low rank for several neural network architectures, and it is demonstrated empirically that this phenomenon may facilitate an explanation of generalization over natural data.

Noam Razin · 1 citation
#machine learning Open access Oct 2023

Delta-AI: Local objectives for amortized inference in sparse graphical models

A new algorithm for amortized inference in sparse probabilistic graphical models (PGMs) is presented that enables off-policy training but avoids the need to instantiate all the random variables for each parameter update, thus speeding up training considerably.

J. Falet, Haebeom Lee, Esmeralda S. Whitammer et al. · 9 citations
#machine learning Open access Oct 2022

GFlowNets and variational inference

This paper builds bridges between two families of probabilistic algorithms: (hierarchical) variational inference (VI), which is typically used to model distributions over continuous spaces, and generative flow networks (GFlowNets), which have been used for distributions over discrete structures such as graphs. We demon...

Esmeralda S. Whitammer, S. Lahlou, T. Deleu et al. · 120 citations · ⚡9
#machine learning Preprint Aug 2026

Implementing neural network mixed-effects models in Template Model Builder (TMB)

A general framework for implementing NMMs using Template Model Builder (TMB), which automatically integrates out random effects and evaluates the marginal objective function alongside its exact gradients, eliminating the need for manual derivations or ad hoc approximations.

Nan Zheng, H. Cheung, Vibhu Sharma et al. · 0 citations
#machine learning Preprint Aug 2026

Learning the Geometry of Admissible Hypotheses through Inductive Bias in Training Distributions

This work presents a framework for learning continuous latent representations of admissible partial differential equations by embedding a scientific inductive bias directly into the training distribution, and shows that embedding a scientific inductive bias in the training distribution enables the learning of compact a...

J. Crowley, Faez Ahmed, A. van Beek · 0 citations
#machine learning Preprint Aug 2026

Informative Label Missingness in Multiclass Classification Information Geometry and Excess Risk

A classification-weighted generalized-eigenvalue criterion is developed under which informative partial classification may have smaller asymptotic classification risk without globally dominating complete classification in Fisher information.

Fariborz Setoudehtazang, Geoffrey J. McLachlan · 0 citations
#machine learning Preprint Aug 2026

Learning Representations through Token Prediction: Geometry, Approximation, and Downstream Guarantees

A statistical framework connecting token prediction with representation geometry, encoder approximation, and downstream performance is developed, introducing a self-consistency principle showing that repeated applications of a shared representation block can progressively refine the contextual representation without in...

Shu-Lei Wang · 0 citations

From tech blogs

See all →
Microsoft Research Blog Oct 15, 2026

Proof-based general-purpose computing with Magnetar

Zero-knowledge proofs can verify computing results without rerunning workloads or exposing sensitive data. Magnetar helps bring them into the mainstream with support for standard programming languages, major speed gains, and quantum-resistant security. The post Proof-based general-purpose computing with Magnetar 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.