Conditioning Tree-Based Diffusions and Flows for Probabilistic Tabular Regression
On the map: Diffusion models → Diffusion foundations and sampling
Abstract (arXiv)
Tree-based diffusion models fit flexible conditional predictive distributions for tabular regression without a neural density estimator, but they inherit their design defaults---noising path, parameterization, training distribution, features, sampler---from the neural setting. We show these defaults are the binding constraint: what a gradient-boosted ensemble actually solves is a supervised regression problem whose conditioning they determine. We present DiffGBM, which makes them explicit along two axes. First, a Gaussian-path flow-matching trainer for p(y mid x) that learns a velocity field directly and recovers the score algebraically, admitting few-step deterministic ODE sampling. Second, we expose the score-side recipe---residualization, EDM-style preconditioning, log-sigma time sampling, noise-level features, loss weighting, and histogram resolution---as jointly tunable axes over a shared LightGBM surface rather than one frozen bundle. This score-flex space represents the published recipe as a special case; across eleven tabular benchmarks under fold-0 tuning, folds-1--5 evaluation, and a matched 40-trial budget and sampler, the selected configurations beat that baseline on every dataset (paired Wilcoxon 11/0, p<10-3), with the best aggregate CRPS skill (0.725 vs. 0.699) of any row. The two rows are complementary: score-flex buys accuracy with a stochastic sampler and is the slowest row, while flow matching is the cheapest sampler (5.2× faster than the published baseline) and the best-calibrated DiffGBM row. Tuned non-diffusion baselines still win individual datasets, and stochastic (varepsilon>0) flow samplers do not Pareto-dominate the deterministic corner.
Key facts. No citing papers in the map yet (data through 7 Oct 2026). Placed in Diffusion foundations and sampling, one of 14 directions in Diffusion models. Builds on 17 of its 23 references that are in the map. First posted on arXiv 30 Jul 2026.
Builds on
17 of 23 references.
Closest ancestors · 5
Same direction, newest first
- 2024Treeffuser: Probabilistic Predictions via Conditional Diffusions with Gradient-Boosted Trees11 Jun
- 2023Efficient Diffusion Training via Min-SNR Weighting Strategy16 Mar
- 2022CARD: Classification and Regression Diffusion Models15 Jun
- 2022Elucidating the Design Space of Diffusion-Based Generative Models1 Jun
- 2020Score-Based Generative Modeling through Stochastic Differential Equations26 Nov
Other references · 12
- 2022Flow Matching for Generative Modelingcited by 5,426
- 2016Simple and Scalable Predictive Uncertainty Estimation using Deep Ensemblescited by 3,888
- 2022Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flowcited by 3,550
- 2023Stochastic Interpolants: A Unifying Framework for Flows and Diffusionscited by 897
- 2019Conformalized Quantile Regressioncited by 620
- 2022TabDDPM: Modelling Tabular Data with Diffusion Modelscited by 373
- 2019NGBoost: Natural Gradient Boosting for Probabilistic Predictioncited by 94
- 2023Generating and Imputing Tabular Data via Diffusion and Flow-based Gradient-Boosted Treescited by 60
- 2020Distributional Random Forests: Heterogeneity Adjustment and Multivariate Distributional Regressioncited by 36
- 2020SGLB: Stochastic Gradient Langevin Boostingcited by 12
- 2022Instance-Based Uncertainty Estimation for Gradient-Boosted Regression Treescited by 9
- 2018Boulevard: Regularized Stochastic Gradient Boosted Trees and Their Limiting Distributioncited by 7
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2023Zero-Shot Conditioning of Score-Based Diffusion Models by Neuro-Symbolic Constraints
- 2026XGenBoost: Synthesizing Small and Large Tabular Datasets with XGBoost
- 2024TFG: Unified Training-Free Guidance for Diffusion Models
- 2026CDMD: A Cross-Dataset Mixed-Type Diffusion Model for Tabular Data
- 2025TabRep: Training Tabular Diffusion Models with a Simple and Effective Continuous Representation
- 2023MissDiff: Training Diffusion Models on Tabular Data with Missing Values
- 2025Improved probabilistic regression using diffusion models
- 2025Boosting Predictive Performance on Tabular Data through Data Augmentation with Latent-Space Flow-Based Diffusion