Exponential Family Variational Flow Matching for Tabular Data Generation
Andrés Guzmán-Cordero, Floor Eijkelboom, Jan-Willem van de Meent
Abstract
While denoising diffusion and flow matching have driven major advances in generative modeling, their application to tabular data remains limited, despite its ubiquity in real-world applications. To this end, we develop TabbyFlow, a variational Flow Matching (VFM) method for tabular data generation. To apply VFM to data with mixed continuous and discrete features, we introduce Exponential Family Variational Flow Matching (EF-VFM), which represents heterogeneous data types using a general exponential family distribution. We hereby obtain an efficient, data-driven objective based on moment matching, enabling principled learning of probability paths over mixed continuous and discrete variables. We also establish a connection between variational flow matching and generalized flow matching objectives based on Bregman divergences. Evaluation on tabular data benchmarks demonstrates state-of-the-art performance compared to baselines.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext f0f50707-2b2e-4e53-ba39-6f4cf2a3f4e3Cited by top-tier papers6
- Categorical Flow MapsDaan Roos, Oscar Davis, Floor Eijkelboom, Michael Bronstein et al.ICML 2026 · 23 citations
- Riemannian Variational Flow Matching for Material and Protein DesignOlga Zaghen, Floor Eijkelboom, Alison Pouplin, Cong Liu et al.ICLR 2026 · 10 citations
- Purrception: Variational Flow Matching for Vector-Quantized Image GenerationRazvan-Andrei Matisan, Vincent Tao Hu, Grigory Bartosh, Björn Ommer et al.ICLR 2026 · 4 citations
- Controlled Generation with Equivariant Variational Flow MatchingFloor Eijkelboom, Heiko Zimmermann, Sharvaree Vadgama, Erik J. Bekkers et al.ICML 2025
- TabGeoFlow: A Geometric Flow Matching Model for Tabular Data SynthesisJong In ChoiAAAI 2026
Builds on16
- High-Resolution Image Synthesis with Latent Diffusion ModelsRobin Rombach, Andreas Blattmann, Dominik Lorenz, Patrick Esser et al.CVPR 2022 · 13,123 citations
- Elucidating the Design Space of Diffusion-Based Generative ModelsTero Karras, Miika Aittala, Timo Aila, Samuli LaineNeurIPS 2022 · 3,959 citations
- Structured Denoising Diffusion Models in Discrete State-SpacesJacob Austin, Daniel D. Johnson, Jonathan Ho, Daniel Tarlow et al.NeurIPS 2021 · 2,256 citations
- Maximum Likelihood Training of Score-Based Diffusion ModelsYang Song, Conor Durkan, Iain Murray, Stefano ErmonNeurIPS 2021 · 958 citations
- Discrete Flow MatchingItai Gat, Tal Remez, Neta Shaul, Felix Kreuk et al.NeurIPS 2024 · 363 citations
Related papers
- Cascaded Flow Matching for Heterogeneous Tabular Data with Mixed-Type FeaturesMarkus Mueller, Kathrin Gruber, Dennis FokICML 2026 · 5 citations
- TabDDPM: Modelling Tabular Data with Diffusion ModelsAkim Kotelnikov, Dmitry Baranchuk, Ivan Rubachev, Artem BabenkoICML 2023 · 518 citations
- Continuous Diffusion for Mixed-Type Tabular DataMarkus Mueller, Kathrin Gruber, Dennis FokICLR 2025
- TabDiff: a Mixed-type Diffusion Model for Tabular Data GenerationJuntong Shi, Minkai Xu, Harper Hua, Hengrui Zhang et al.ICLR 2025
- Variational Flow Matching for Graph GenerationFloor Eijkelboom, Grigory Bartosh, Christian Andersson Naesseth, Max Welling et al.NeurIPS 2024 · 96 citations
