Authors
Sercan Ö Arik, Tomas Pfister
Publication date
2021/5/18
Journal
Proceedings of the AAAI conference on artificial intelligence
Volume
35
Issue
8
Pages
6679-6687
Description
We propose a novel high-performance and interpretable canonical deep tabular data learning architecture, TabNet. TabNet uses sequential attention to choose which features to reason from at each decision step, enabling interpretability and more efficient learning as the learning capacity is used for the most salient features. We demonstrate that TabNet outperforms other variants on a wide range of non-performance-saturated tabular datasets and yields interpretable feature attributions plus insights into its global behavior. Finally, we demonstrate self-supervised learning for tabular data, significantly improving performance when unlabeled data is abundant.
Total citations
2020202120222023202435114268455325
Scholar articles
SÖ Arik, T Pfister - Proceedings of the AAAI conference on artificial …, 2021