Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

A CDF-First Framework for Free-Form Density Estimation

About

Conditional density estimation (CDE) is a fundamental task in machine learning that aims to model the full conditional law $\mathbb{P}(\mathbf{y} \mid \mathbf{x})$, beyond mere point prediction (e.g., mean, mode). A core challenge is free-form density estimation, capturing distributions that exhibit multimodality, asymmetry, or topological complexity without restrictive assumptions. However, prevailing methods typically estimate the probability density function (PDF) directly, which is mathematically ill-posed: differentiating the empirical distribution amplifies random fluctuations inherent in finite datasets, necessitating strong inductive biases that limit expressivity and fail when violated. We propose a CDF-first framework that circumvents this issue by estimating the cumulative distribution function (CDF), a stable and well-posed target, and then recovering the PDF via differentiation of the learned smooth CDF. Parameterizing the CDF with a Smooth Min-Max (SMM) network, our framework guarantees valid PDFs by construction, enables tractable approximate likelihood training, and preserves complex distributional shapes. For multivariate outputs, we use an autoregressive decomposition with SMM factors. Experiments demonstrate our approach outperforms state-of-the-art density estimators on a range of univariate and multivariate tasks.

Chenglong Song, Mazharul Islam, Lin Wang, Bing Chen, Bo Yang• 2026

Related benchmarks

TaskDatasetResultRank
Conditional Density EstimationFish 1D (test)
Negative Log-Likelihood0.89
6
Conditional Density EstimationConcrete 1D (test)
Negative Log-Likelihood0.44
6
Conditional Density EstimationEnergy 2D (test)
Negative Log-Likelihood-2.03
6
Conditional Density EstimationTemperature 2D (test)
Negative Log-Likelihood (NLL)0.7
6
Conditional Density EstimationSkillcraft 4D (test)
Negative Log-Likelihood (NLL)1.65
6
Conditional Density EstimationParkinsons 2D (test)
Negative Log-Likelihood0.73
6
Conditional Density EstimationAir 3D (test)
Negative Log-Likelihood1.06
6
Showing 7 of 7 rows

Other info

Follow for update