arXiv:2602.14622cs.AIcs.DB2026-02

用表格大模型直接学关联规则,不依赖传统频繁项集挖掘。

Tabular Foundation Models Can Learn Association Rules

  • 基于条件概率模型,从预训练表格模型中提取规则。
  • 低数据场景下仍能生成高质量、简洁的关联规则。
  • 无需微调,适合快速部署在小样本或新领域任务中。

关联规则挖掘(ARM)是表格数据知识发现的核心任务,广泛应用于高风险决策场景。传统方法依赖频繁项集挖掘,易导致规则爆炸且可扩展性差;近期神经方法虽缓解此问题,但在低数据环境下性能下降。表格基础模型(TFMs)通过在多样化表格数据上预训练,具备强大的上下文泛化能力,为解决上述挑战提供可能。本文提出一种与模型无关的关联规则学习框架,可从任意表格数据的条件概率模型中提取规则,从而利用TFMs。我们进一步提出TabProbe,作为该框架的具体实现,直接使用TFMs作为条件概率估计器,无需频繁项集挖掘即可开箱即用地学习关联规则。我们在不同规模的表格数据集上评估了该方法在标准ARM规则质量指标及下游分类性能的表现。结果表明,TFMs能持续生成简洁、高质量的关联规则,并在低数据设置下保持鲁棒性,且无需特定任务训练。源代码已公开于https://github.com/DiTEC-project/tabprobe。

原文摘要 · Abstract (English)

Association Rule Mining (ARM) is a fundamental task for knowledge discovery in tabular data and is widely used in high-stakes decision-making. Classical ARM methods rely on frequent itemset mining, leading to rule explosion and poor scalability, while recent neural approaches mitigate these issues but suffer from degraded performance in low-data regimes. Tabular foundation models (TFMs), pretrained on diverse tabular data with strong in-context generalization, provide a basis for addressing these limitations. We introduce a model-agnostic association rule learning framework that extracts association rules from any conditional probabilistic model over tabular data, enabling us to leverage TFMs. We then introduce TabProbe, an instantiation of our framework that utilizes TFMs as conditional probability estimators to learn association rules out-of-the-box without frequent itemset mining. We evaluate our approach on tabular datasets of varying sizes based on standard ARM rule quality metrics and downstream classification performance. The results show that TFMs consistently produce concise, high-quality association rules with strong predictive performance and remain robust in low-data settings without task-specific training. Source code is available at https://github.com/DiTEC-project/tabprobe.

关联规则表格模型低数据无监督

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。