Better by Default: Strong Pre-Tuned MLPs and Boosted Trees on Tabular Data

David Holzmüller,Léo Grinsztajn,Ingo Steinwart
2024-10-31
Abstract:For classification and regression on tabular data, the dominance of gradient-boosted decision trees (GBDTs) has recently been challenged by often much slower deep learning methods with extensive hyperparameter tuning. We address this discrepancy by introducing (a) RealMLP, an improved multilayer perceptron (MLP), and (b) strong meta-tuned default parameters for GBDTs and RealMLP. We tune RealMLP and the default parameters on a meta-train benchmark with 118 datasets and compare them to hyperparameter-optimized versions on a disjoint meta-test benchmark with 90 datasets, as well as the GBDT-friendly benchmark by Grinsztajn et al. (2022). Our benchmark results on medium-to-large tabular datasets (1K--500K samples) show that RealMLP offers a favorable time-accuracy tradeoff compared to other neural baselines and is competitive with GBDTs in terms of benchmark scores. Moreover, a combination of RealMLP and GBDTs with improved default parameters can achieve excellent results without hyperparameter tuning. Finally, we demonstrate that some of RealMLP's improvements can also considerably improve the performance of TabR with default parameters.
Machine Learning
What problem does this paper attempt to address?
The main problem that this paper attempts to solve is that in classification and regression tasks on tabular data, although deep - learning methods have potential in terms of accuracy, they usually require a large amount of hyper - parameter tuning, which leads to their inefficiency in practical applications. In contrast, Gradient - Boosting Decision Trees (GBDTs) perform well on these tasks, but their advantages are being challenged by deep - learning methods. For this reason, the paper makes the following two main contributions: 1. **Introduction of RealMLP**: This is an improved Multi - Layer Perceptron (MLP). Through a series of optimization techniques and better default parameter settings, its performance on tabular data is improved. These optimizations include, but are not limited to, pre - processing using robust scaling and smooth clipping, new numerical embedding variants, diagonal weight layers, new scheduling methods, and different initialization methods, etc. 2. **Proposing default parameters for strong meta - tuning**: These are for not only RealMLP but also GBDTs. These parameters can achieve excellent performance without being tuned on individual datasets. The author verifies the effectiveness of these default parameters by tuning these models on a meta - training benchmark containing 118 datasets and evaluating them on an independent meta - testing benchmark containing 90 datasets. The goal of the paper is to make neural - network - based methods less dependent on hyper - parameter tuning without sacrificing too much performance through these improvements, thereby improving their practicality and efficiency. In addition, the paper also explores the selection and integration strategies of different models and shows how to use these improved default parameters to achieve a better time - performance trade - off.