用字符串编码表结构语义,实现高效可复用的表格预训练模型
Table Foundation Models: on knowledge pre-training for tabular learning
- 通过字符串融合列名与数值,捕捉表格上下文语义
- 预训练后只需轻量微调,即可在多个任务上达到领先性能
- 支持跨任务复用和与其他模型组合,适合领域定制化应用
表格基础模型为数据科学带来新希望:通过在表格数据上预训练,注入知识或先验信息,以辅助下游表格任务。一个核心挑战是数据语义:数值的意义依赖于上下文,如列名。近期联合建模列名与表格内容的神经网络显著提升了预测准确率。然而,这些模型缺乏文本或视觉领域主流基础模型的便捷性——需微调才能获益,计算开销大,且难以复用或与其他架构结合。本文提出TARTE,一种将表格转化为知识增强向量表示的基础模型,利用字符串捕捉语义。该模型在大规模关系型数据上预训练,生成的表示能以极低额外成本促进后续学习。这些表示可微调或与其他学习器结合,构建出性能领先的模型,同时优化预测与计算效率的权衡。针对特定任务或领域定制时,TARTE可生成领域特异性表示,进一步提升学习效果。本研究验证了表格学习中知识预训练的有效路径。
原文摘要 · Abstract (English)
Table foundation models bring high hopes to data science: pre-trained on tabular data to embark knowledge or priors, they should facilitate downstream tasks on tables. One specific challenge is that of data semantics: numerical entries take their meaning from context, e.g., column name. Pre-trained neural networks that jointly model column names and table entries have recently boosted prediction accuracy. While these models outline the promises of world knowledge to interpret table values, they lack the convenience of popular foundation models in text or vision. Indeed, they must be fine-tuned to bring benefits, come with sizeable computation costs, and cannot easily be reused or combined with other architectures. Here we introduce TARTE, a foundation model that transforms tables to knowledge-enhanced vector representations using the string to capture semantics. Pre-trained on large relational data, TARTE yields representations that facilitate subsequent learning with little additional cost. These representations can be fine-tuned or combined with other learners, giving models that push the state-of-the-art prediction performance and improve the prediction/computation performance trade-off. Specialized to a task or a domain, TARTE gives domain-specific representations that facilitate further learning. Our study demonstrates an effective approach to knowledge pre-training for tabular learning.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。