Timezone: »
Denoising diffusion probabilistic models are becoming the leading generative modeling paradigm for many important data modalities. Being the most prevalent in the computer vision community, diffusion models have recently gained some attention in other domains, including speech, NLP, and graph-like data. In this work, we investigate if the framework of diffusion models can be advantageous for general tabular problems, where data points are typically represented by vectors of heterogeneous features. The inherent heterogeneity of tabular data makes it quite challenging for accurate modeling since the individual features can be of a completely different nature, i.e., some of them can be continuous and some can be discrete. To address such data types, we introduce TabDDPM --- a diffusion model that can be universally applied to any tabular dataset and handles any feature types. We extensively evaluate TabDDPM on a wide set of benchmarks and demonstrate its superiority over existing GAN/VAE alternatives, which is consistent with the advantage of diffusion models in other fields.
Author Information
Akim Kotelnikov (HSE, Yandex)
Dmitry Baranchuk (Yandex)
Ivan Rubachev (Higher School of Economics)
Artem Babenko (Yandex)
More from the Same Authors
-
2021 : Discovering Weight Initializers with Meta Learning »
Dmitry Baranchuk · Artem Babenko -
2021 Poster: Object Segmentation Without Labels with Large-Scale Generative Models »
Andrey Voynov · Stanislav Morozov · Artem Babenko -
2021 Spotlight: Object Segmentation Without Labels with Large-Scale Generative Models »
Andrey Voynov · Stanislav Morozov · Artem Babenko -
2021 Poster: Functional Space Analysis of Local GAN Convergence »
Valentin Khrulkov · Artem Babenko · Ivan Oseledets -
2021 Spotlight: Functional Space Analysis of Local GAN Convergence »
Valentin Khrulkov · Artem Babenko · Ivan Oseledets -
2020 Poster: Unsupervised Discovery of Interpretable Directions in the GAN Latent Space »
Andrey Voynov · Artem Babenko -
2019 Poster: Learning to Route in Similarity Graphs »
Dmitry Baranchuk · Dmitry Persiyanov · Anton Sinitsin · Artem Babenko -
2019 Oral: Learning to Route in Similarity Graphs »
Dmitry Baranchuk · Dmitry Persiyanov · Anton Sinitsin · Artem Babenko