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
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| Regression | CA Housing | -- | 54 | |
| Regression | CHS | R^20.878 | 11 | |
| Regression | nHS | R-Squared0.869 | 11 | |
| Regression | HP Kaggle | R^20.864 | 11 | |
| Regression | nAbal | R^2 Score0.513 | 11 | |
| Regression | PMI Kaggle | R^20.845 | 11 | |
| Regression | cAbal | R^20.521 | 11 | |
| Regression | CAS | R^20.944 | 11 | |
| Regression | nElev | R^20.856 | 11 | |
| Regression | cSeat | R^20.157 | 11 |
Showing 10 of 15 rows