SamsungSAILMontreal / ForestDiffusion

Generating and Imputing Tabular Data via Diffusion and Flow XGBoost Models
142 stars 13 forks source link

Density estimation #7

Closed calvinmccarter closed 11 months ago

calvinmccarter commented 11 months ago

This method worked really well on my data! Is it possible to perform density estimation with ForestDiffusion?

AlexiaJM commented 11 months ago

Hi Calvin,

Our method does not perform density estimation. It could be possible to implement it following the approach used in the zero-shot classification, but it would be computationally intensive (n_t=10, n_z=10, 250 values = 250 x 10 x 10 xgboost predictions) and I'm not sure what would be the advantage over using standard kernel-based density estimation.

Alexia

calvinmccarter commented 11 months ago

Hi Alexia, thanks -- makes sense!