activity
20242026
collaborators
Showing stat.MLShow all

6 papers · 1 filter

stat.ML2026

Robustness and Structure Preservation in Flow-Based Generative Models via Wasserstein Path-Space Divergences

Ziyu Chen, Markos A. Katsoulakis, Benjamin J. Zhang

We introduce a novel Wasserstein-1 () path-space divergence for stochastic and deterministic dynamics and establish a Wasserstein Uncertainty Propagation (WUP) theorem that bo…

stat.ML2026

ISOMORPH: A Supply Chain Digital Twin for Simulation, Dataset Generation, and Forecasting Benchmarks

Zhizhen Zhang, Hyemin Gu, Benjamin J. Zhang +6

Open time-series forecasting (TSF) benchmarks cover retail, energy, weather, and traffic, but supply-chain logistics remains underserved. We introduce ISOMORPH, the first public di…

stat.ML2025

Probabilistic operator learning: generative modeling and uncertainty quantification for foundation models of differential equations

Benjamin J. Zhang, Siting Liu, Stanley J. Osher +1

In-context operator networks (ICON) are a class of operator learning methods based on the novel architectures of foundation models. Trained on a diverse set of datasets of initial…

stat.ML2025

Nonlinear denoising score matching for enhanced learning of structured distributions

Jeremiah Birrell, Markos A. Katsoulakis, Luc Rey-Bellet +2

We present a novel method for training score-based generative models which uses nonlinear noising dynamics to improve learning of structured distributions. Generalizing to a nonlin…

stat.ML2024

Combining Wasserstein-1 and Wasserstein-2 proximals: robust manifold learning via well-posed generative flows

Hyemin Gu, Markos A. Katsoulakis, Luc Rey-Bellet +1

We formulate well-posed continuous-time generative flows for learning distributions that are supported on low-dimensional manifolds through Wasserstein proximal regularizations of…

stat.ML2024

Score-based generative models are provably robust: an uncertainty quantification perspective

Nikiforos Mimikos-Stamatopoulos, Benjamin J. Zhang, Markos A. Katsoulakis

Through an uncertainty quantification (UQ) perspective, we show that score-based generative models (SGMs) are provably robust to the multiple sources of error in practical implemen…