Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

GANDALF: Gated Adaptive Network for Deep Automated Learning of Features

About

We propose a novel high-performance, interpretable, and parameter \& computationally efficient deep learning architecture for tabular data, Gated Adaptive Network for Deep Automated Learning of Features (GANDALF). GANDALF relies on a new tabular processing unit with a gating mechanism and in-built feature selection called Gated Feature Learning Unit (GFLU) as a feature representation learning unit. We demonstrate that GANDALF outperforms or stays at-par with SOTA approaches like XGBoost, SAINT, FT-Transformers, etc. by experiments on multiple established public benchmarks. We have made available the code at github.com/manujosephv/pytorch_tabular under MIT License.

Manu Joseph, Harsh Raj• 2022

Related benchmarks

TaskDatasetResultRank
RegressionCA Housing--
54
RegressionCHS
R^20.878
11
RegressionnHS
R-Squared0.869
11
RegressionHP Kaggle
R^20.864
11
RegressionnAbal
R^2 Score0.513
11
RegressionPMI Kaggle
R^20.845
11
RegressioncAbal
R^20.521
11
RegressionCAS
R^20.944
11
RegressionnElev
R^20.856
11
RegressioncSeat
R^20.157
11
Showing 10 of 15 rows

Other info

Follow for update