OpenAI Blog

Implicit generation and generalization methods for energy-based models

We’ve made progress towards stable and scalable training of energy-based models (EBMs) resulting in better sample quality and generalization ability than existing models. Generation in EBMs spends more compute to continually refine its answers and doing so can generate samples competitive with GANs at low temperatures, while also having mode coverage guarantees of likelihood-based models.

Hugging Face Trending Papers
Jul 29

Equilibrium Training of Energy-Based Models with Parallel Trajectory Tempering

Energy-Based Models (EBMs) provide an interpretable framework for generative modeling of scientific data, but poor Markov Chain Monte Carlo mixing often limits their reliability. We introduce a training algorithm based on Parallel Trajectory Tempering (PTT), which exploits the continuity of the optimization path to maintain equilibrium sampling throughout learning.

arXiv AI
Jun 24

Catastrophic Compositional Generation: Why Vanilla Diffusion Models Fail to Extrapolate

arXiv:2606. 23920v1 Announce Type: cross Abstract: The task of compositional generation involves using a conditional generative model, trained only on a subset of the possible conditions, to produce samples from compositionally-defined target distributions such as a geometric combination of the source distributions.

By Duncan Soiffer, Chandler Squires, Yuan Guan, Jason Hartford, Pradeep Ravikumar
arXiv Machine Learning
Jul 2

Decision-Aware Training for Sample-Based Generative Models

arXiv:2607. 01171v1 Announce Type: new Abstract: Sample-based generative models are increasingly used for probabilistic forecasting in high-stakes decision settings, yet their training objectives are blind to the decision maker's cost structure.

By Kornelius Raeth, Nicole Ludwig
arXiv Machine Learning
Sep 11

Smoothing the Score Function to Enhance Generalization in Diffusion Models

The paper investigates memorization in diffusion models, showing that the empirical score function is a weighted sum of Gaussian score functions with sharp softmax weights, causing individual training samples to dominate and lead to sampling collapse. By approximating this function with a neural network, the authors obtain a smoother representation that generalizes better. They introduce two techniques—Noise Unconditioning and Temperature Smoothing—to further reduce single‑sample dominance, and demonstrate improved generalization across multiple datasets while preserving generation quality.

By Xinyu Zhou, Jiawei Zhang, Stephen J. Wright
arXiv Machine Learning
Aug 24

Learning Minimal-Deviation Corrections for Multi-Dimensional Mismodelling in HEP Simulations

The paper introduces a neural‑network method that learns a minimal‑deviation transformation of Monte Carlo simulated events to match one‑dimensional target distributions while preserving the multidimensional correlation structure of the original simulation. By operating under limited experimental information, the approach avoids the pitfalls of traditional one‑dimensional reweighting and the data‑hungry fully multidimensional corrections. Controlled pseudo‑data studies demonstrate improved agreement with target distributions and consistent multidimensional structure, making the method suitable for complex, high‑dimensional analyses where conventional techniques fall short.

By Matthias Schott, Lucie Flek
arXiv Machine Learning
2d ago

Manifold-Aware Perturbations for Constrained Generative Modeling

The paper introduces a new method for perturbing data distributions in a way that respects equality constraints, allowing generative models to better handle constrained data. By adjusting the distribution while preserving the manifold geometry, the approach ensures support matches the ambient space dimension. Experiments with diffusion models and normalizing flows demonstrate improved data recovery and stable sampling across several tasks.

By Katherine Keegan, Lars Ruthotto