为表格数据设计图结构位置编码,提升Transformer模型泛化能力
Tab-PET: Graph-Based Positional Encodings for Tabular Transformers
- 基于图结构估计位置编码,融入表格Transformer模型
- 在50个数据集上显著提升3T模型性能,最高增益达12.3%
- 关联性构建的图编码效果更稳定,适合实际应用
表格数据的监督学习面临数据量小、缺乏结构线索及特征异质性等挑战。与视觉和语言任务不同,表格数据无天然的位置结构,削弱了自注意力机制的效果。尽管近期基于Transformer的模型(如TabTransformer、SAINT、FT-Transformer,统称3T)在表格数据上表现良好,但通常未利用位置编码(PEs),因缺乏先验结构信息。本文从理论和实证两方面发现,位置编码能有效降低特征的有效秩(内在维度),简化任务并提升泛化能力。为此,提出Tab-PET框架,通过图结构估计并嵌入位置编码。受图拓扑启发,探索了基于关联性和因果性的两种图构建范式。实验证明,图导位置编码在50个分类与回归数据集上显著提升3T性能,其中关联性图编码效果更优且更稳定。研究揭示了位置编码在表格Transformer中的意外作用,展示了其提升泛化能力的潜力。
原文摘要 · Abstract (English)
Supervised learning with tabular data presents unique challenges, including low data sizes, the absence of structural cues, and heterogeneous features spanning both categorical and continuous domains. Unlike vision and language tasks, where models can exploit inductive biases in the data, tabular data lacks inherent positional structure, hindering the effectiveness of self-attention mechanisms. While recent transformer-based models like TabTransformer, SAINT, and FT-Transformer (which we refer to as 3T) have shown promise on tabular data, they typically operate without leveraging structural cues such as positional encodings (PEs), as no prior structural information is usually available. In this work, we find both theoretically and empirically that structural cues, specifically PEs can be a useful tool to improve generalization performance for tabular transformers. We find that PEs impart the ability to reduce the effective rank (a form of intrinsic dimensionality) of the features, effectively simplifying the task by reducing the dimensionality of the problem, yielding improved generalization. To that end, we propose Tab-PET (PEs for Tabular Transformers), a graph-based framework for estimating and inculcating PEs into embeddings. Inspired by approaches that derive PEs from graph topology, we explore two paradigms for graph estimation: association-based and causality-based. We empirically demonstrate that graph-derived PEs significantly improve performance across 50 classification and regression datasets for 3T. Notably, association-based graphs consistently yield more stable and pronounced gains compared to causality-driven ones. Our work highlights an unexpected role of PEs in tabular transformers, revealing how they can be harnessed to improve generalization.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。