NAISS
SUPR
NAISS Projects
SUPR
Diffusion-Based Methods for Constrained Generative Modeling and Statistical Sampling
Dnr:

NAISS 2026/4-640

Type:

NAISS Small

Principal Investigator:

Marcus Häggbom

Affiliation:

Kungliga Tekniska högskolan

Start Date:

2026-04-13

End Date:

2027-05-01

Primary Classification:

10106: Probability Theory and Statistics (Statistics with medical aspects at 30118 and with social aspects at 50907)

Webpage:

Allocation

Abstract

This project develops generative modeling methods for scientific domains where physical or mathematical constraints must be respected. We focus on Schrödinger Bridge matching models, a class of diffusion-like generative models, and study how to incorporate domain constraints in a way that provides theoretical guarantees. In parallel, we adapt ideas from diffusion models to improve sampling of complex probability distributions arising in statistical mechanics, with the aim of obtaining more accurate and efficient approximations of stationary ergodic stochastic processes. More broadly, the project explores how modern generative-modeling frameworks can be extended beyond mainstream AI applications to mathematically grounded scientific settings. Main supervisor: Joakim Andén, Dept. of Mathematics, KTH