arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2602.14622cs.AIcs.DBcs.LG
收藏

表格基础模型可以学习关联规则

Tabular Foundation Models Can Learn Association Rules

Amsterdam UMC, location VUmc, Vrije Universiteit Amsterdam, the Netherlands(阿姆斯特丹大学医学中心,位于VUmc,荷兰代尔夫特自由大学);University of Amsterdam, The Netherlands(阿姆斯特丹大学,荷兰);Vrije Universiteit Amsterdam, The Netherlands(荷兰阿姆斯特丹自由大学)

Erkan Karabulut, Daniel Daza, Paul Groth, Martijn C. Schut, Victoria Degeler

AI总结:

本文提出基于表格基础模型的关联规则学习框架,通过TabProbe无需频繁项集挖掘即可学习高质量关联规则,适用于低数据场景。

摘要:

关联规则挖掘(ARM)是表格数据中的知识发现基本任务,并广泛应用于高风险决策。经典ARM方法依赖于频繁项集挖掘,导致规则爆炸和较差的可扩展性,而近期的神经方法缓解了这些问题,但在低数据情况下性能下降。表格基础模型(TFMs),在多样化的表格数据上预训练,具有强大的上下文泛化能力,为解决这些限制提供了基础。我们介绍了一个模型无关的关联规则学习框架,该框架可以从任何关于表格数据的条件概率模型中提取关联规则,使我们能够利用TFMs。然后我们介绍了TabProbe,这是我们框架的一个实例,利用TFMs作为条件概率估计器来学习关联规则,无需频繁项集挖掘。我们基于标准ARM规则质量度量和下游分类性能,在不同大小的表格数据集上评估我们的方法。结果表明,TFMs能够持续产生简洁、高质量的关联规则,具有强大的预测性能,并且在低数据情况下保持稳健,而无需特定任务训练。源代码可在https://github.com/DiTEC-project/tabprobe上获得。

英文摘要:

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.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑