arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2201.00199cs.LGcs.AI

The GatedTabTransformer. An enhanced deep learning architecture for tabular modeling

High School of Mathematics(数学高中);Obecto(Obecto公司)

Radostin Cholakov, Todor Kolev

英文摘要:

There is an increasing interest in the application of deep learning architectures to tabular data. One of the state-of-the-art solutions is TabTransformer which incorporates an attention mechanism to better track relationships between categorical features and then makes use of a standard MLP to output its final logits. In this paper we propose multiple modifications to the original TabTransformer performing better on binary classification tasks for three separate datasets with more than 1% AUROC gains. Inspired by gated MLP, linear projections are implemented in the MLP block and multiple activation functions are tested. We also evaluate the importance of specific hyper parameters during training.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑