Constant-Target Energy Matching: A Unified Framework for Continuous and Discrete Density Estimation
Not on the map. Why?
Abstract (arXiv)
Density estimation is a central primitive in probabilistic modeling, yet continuous, discrete, and mixed-variable domains are often treated by separate objectives, limiting the ability to exploit a common statistical structure across data types. Continuous score-based methods rely on log-density gradients, while discrete extensions typically use concrete score whose unbounded targets become unstable near low-probability states. We introduce Constant-Target Energy Matching (CTEM), a unified energy-based framework for density estimation on general state spaces. CTEM replaces ordinary density-ratio regression with a bounded energy-difference transform and derives from it a sample-only training objective with the constant target 1. The learned scalar potential recovers log p without partition-function estimation or explicit unbounded ratio regression. Across continuous, discrete, and mixed-variable benchmarks, CTEM substantially improves density estimation over competitive baselines and yields higher-quality samples under standard sampling procedures.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 24 of its 34 references that are in the map. First posted on arXiv 9 May 2026.
Builds on
24 of 34 references.
References · 24
- 2020Denoising Diffusion Probabilistic Modelscited by 21,076
- 2020Score-Based Generative Modeling through Stochastic Differential Equationscited by 9,122
- 2022Flow Matching for Generative Modelingcited by 5,702
- 2019Generative Modeling by Estimating Gradients of the Data Distributioncited by 4,101
- 2022Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flowcited by 3,700
- 2022Elucidating the Design Space of Diffusion-Based Generative Modelscited by 3,045
- 2015Variational Inference with Normalizing Flowscited by 2,170
- 2016Density estimation using Real NVPcited by 2,042
- 2023Consistency Modelscited by 1,822
- 2021Structured Denoising Diffusion Models in Discrete State-Spacescited by 1,719
- 2018Glow: Generative Flow with Invertible 1x1 Convolutionscited by 1,545
- 2024Simple and Effective Masked Diffusion Language Modelscited by 848
12 more
- 2023Discrete Diffusion Modeling by Estimating the Ratios of the Data Distributioncited by 745
- 2017Masked Autoregressive Flow for Density Estimationcited by 684
- 2024Simplified and Generalized Masked Diffusion for Discrete Datacited by 626
- 2019Neural Spline Flowscited by 596
- 2018FFJORD: Free-form Continuous Dynamics for Scalable Reversible Generative Modelscited by 594
- 2022A Continuous Time Framework for Discrete Denoising Modelscited by 451
- 2019Sliced Score Matching: A Scalable Approach to Density and Score Estimationcited by 384
- 2024Your Absorbing Discrete Diffusion Secretly Models the Conditional Distributions of Clean Datacited by 381
- 2024Discrete Flow Matchingcited by 317
- 2022Concrete Score Matching: Generalized Score Matching for Discrete Datacited by 148
- 2025The Diffusion Dualitycited by 135
- 2025SD-KDE: Score-Debiased Kernel Density Estimationcited by 2
Built on by
Not cited yet.