摘要:
本文提出了一个名为 INCE 的表格深度学习模型,其核心创新在于用 GNN 实现交互网络,用作表格特征的上下文嵌入器。模型将每个表格行转换为一个全连接图,其中每个节点对应一个特征的嵌入表示,并引入一个虚拟 CLS 节点。随后,一个 IN 层堆栈通过消息传递机制,动态地学习和更新图中所有节点(特征)之间连接的强度,从而建模复杂的特征交互。经过 IN 增强后的 CLS节点的最终表示,被用作整个表格行的上下文感知的全局嵌入,并送入一个 MLP 解码器进行预测。通过在公开数据集上与多个 baselines 的广泛对比实验,证明了 INCE 模型在深度学习模型中达到了最佳平均性能,并能与树模型形成有力竞争。 阅读全文