Revisiting Deep Learning Models for Tabular Data

By Yu. V. Gorishniy, Ivan Rubachev, Valentin Khrulkov, Artem Babenko. Neural Information Processing Systems, 2021

1,551 citations recorded by the source corpus.

The existing literature on deep learning for tabular data proposes a wide range of novel architectures and reports competitive results on various datasets. However, the proposed models are usually not properly compared to each other and existing works often use different benchmarks and experiment protocols. As a result, it is unclear for both researchers and practitioners what models perform best. Additionally, the field still lacks effective baselines, that is, the easy-to-use models that provide competitive performance across different problems. In this work, we perform an overview of the main families of DL architectures for tabular data and raise the bar of baselines in tabular DL by identifying two simple and powerful deep architectures. The first one is a ResNet-like architecture which turns out to be a strong baseline that is often missing in prior works. The second model is our simple adaptation of the Transformer architecture for tabular data, which outperforms other solutions on most tasks. Both models are compared to many existing architectures on a diverse set of tasks under the same training and tuning protocols. We also compare the best DL models with Gradient Boosted Decision Trees and conclude that there is still no universally superior solution.

Subfields: Artificial Intelligence, Computer Vision and Pattern Recognition

View Paper: https://www.semanticscholar.org/paper/5fa06d856ba6ae9cd1366888f8134d7fd0db75b9

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Colin White
  2. Edouard Oyallon
  3. Frank Hutter
  4. Samuel G. Müller
  5. Noah Hollmann
  6. Gaël Varoquaux
  7. Léo Grinsztajn
  8. C. Yuhua

In Computer Vision and Pattern Recognition

  1. Luc Van Gool
  2. Edouard Oyallon
  3. Yi Yang
  4. Ming-Hsuan Yang
  5. Dacheng Tao
  6. Trevor Darrell
  7. Andrea Vedaldi
  8. Jitendra Malik

Citations in the corpus