MMODELYST
Papers/Constant-Target Energy Matching: A Unified Framework for Continuous and Discrete Density Estimation
PAP

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

May 9, 2026

arXiv
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.

Select text to highlight · click a highlight to remove · saved in this browser only
Authors
Zhijun Zeng, Yixuan Jiang, Pipi Hu, Zuoqiang Shi
Your notes (browser-local)
saved
arXiv:2605.09085