KnowsTFM:知识引导的小型表格基础模型微调

KnowsTFM: Knowledge-Informed Fine-Tuning of Small Tabular Foundation Models

精选理由

这篇论文教你用小知识图谱提升小表格模型在专业领域的效果,比普通微调更靠谱,还揭示了持续微调的风险。

AI 摘要

论文针对TabPFN和TabICL等纳米级表格基础模型,提出KnowsTFM方法,通过知识图谱的结构注意力先验和参数高效低秩更新进行微调。在数据稀缺、高维且分布偏移的专业领域,KnowsTFM相比原始变体取得显著提升,但在通用任务上增益微小。研究还发现,对前沿模型持续微调可能导致预训练知识及机制崩溃。

原文 · arXiv cs.LG

KnowsTFM: Knowledge-Informed Fine-Tuning of Small Tabular Foundation Models

Tabular foundation models have advanced deep learning for tabular data by delivering strong default performance across many small and medium tasks. Yet in niche domains, where data is scarce, high-dimensional, and shifted from the pretraining distribution, they may still fail to outperform carefully designed domain-specific methods. Many such domains also provide curated relational knowledge in the form of knowledge graphs and knowledge banks, but how to use this knowledge to improve and steer \textit{small} specialist tabular foundation models remains unclear. We address this problem through \textbf{Know}ledge-informed fine-tuning of \textbf{s}mall \textbf{T}abular \textbf{F}oundation \textbf{M}odels (\modelname). Specifically, we study nanoscale TabPFN- and TabICL-style variants, pretrained under controlled synthetic prior families and adapted using two complementary mechanisms: structural attention priors derived from knowledge graphs and parameter-efficient low-rank updates. We show that injecting domain-specific structural knowledge during fine-tuning yields meaningful gains over vanilla variants in specialist settings, whereas gains on general-domain tasks are marginal. We further observe that continual fine-tuning of frontier models can trigger collapse of pretrained knowledge and mechanisms.