Team up GBDTs and DNNs: Advancing Efficient and Effective Tabular Prediction with Tree-hybrid MLPs

• https://arxiv.org/pdf/2407.09790，KDD 2024.

核心思想

融合GBDT和DNN优势，提出高效表格数据预测框架T-MLP。

论文概述

虽然深度模型在各个领域大放光彩，但在表格数据领域，树模型（GBDT）和深度模型（DNN）谁更胜一筹仍在上演龙虎之争！特别是在Kaggle等数据科学竞赛平台上，树模型仍常位居榜首。工业界，深度模型在各个岗位上发光发热，但仍少不了一些树模型在某些角落打螺丝。

GBDT和DNN各有其优势和缺点。

针对这些问题，论文提出了Tree-hybrid MLP（T-MLP）方法，其核心思想是结合GBDT的特征选择和模型集成优势与DNN的高维特征空间和光滑优化特性，通过张量化GBDT特征门、DNN架构剪枝和反向传播协同训练MLP模型，以实现高效、有效的表数据预测。

方法介绍

具体来说，T-MLP包括以下几个关键步骤或模块：

1. 【GBDT特征门控（GBDT Feature Gate，GFG）】
   先用GBDT在训练数据上学习一遍得到一个树模型，再利用这个树模型获得特征频率来选择对当前样本最重要的特征，通过张量化操作将GBDT的决策过程融入到DNN中，实现样本特定的特征选择。
   

2. 纯MLP基本块（Pure MLP Basic Block） ：采用简化版的MLP结构，通过空间门控单元（SGU）实现特征间的交互，保持模型的紧凑性。

3. DNN架构剪枝（DNN Architecture Pruning） ：借鉴GBDT的预剪枝思想，通过DNN剪枝技术减少模型参数，提高模型的泛化能力和训练效率。

实验分析

论文中也对比了不同模块的重要性，特地对比了使用神经网络门控，依然有一定效果。

另外一个有趣的实验是分析了不同模型的决策边界（FT-T表示FT-Transformer）。通过在两个数据集上的例子可以看出树模型和深度模型决策边界的明显区别：树模型决策边界通常是垂直的，深度模型是很光滑的（这也是深度模型容易过拟合的原因——拟合的太好了）。T-MLP看起来兼具了两者的特点，既有大致的垂直也有较光滑的边界。

结论

内核感觉就是特征权重，只是权重来源于树模型，有点像经典的GBDT+LR的方法。

不失为打比赛时的一种选择哦~