English

Constant-Target Energy Matching: A Unified Framework for Continuous and Discrete Density Estimation

Artificial Intelligence 2026-05-12 v1 Probability

Abstract

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.

Keywords

Cite

@article{arxiv.2605.09085,
  title  = {Constant-Target Energy Matching: A Unified Framework for Continuous and Discrete Density Estimation},
  author = {Zhijun Zeng and Yixuan Jiang and Pipi Hu and Zuoqiang Shi},
  journal= {arXiv preprint arXiv:2605.09085},
  year   = {2026}
}
R2 v1 2026-07-01T13:00:15.634Z