arXiv:2506.18155cs.LG2025-06被引 2

用概率与强化学习挖掘关联规则,更好发现稀有模式。

Probabilistic and reinforced mining of association rules

  • 引入高斯过程、贝叶斯等方法建模物品共现与不确定性
  • 在小数据和复杂模式上表现优于传统算法
  • 适合医疗、金融等需风险感知的场景

本文提出四种新型概率与强化驱动的关联规则挖掘方法:基于高斯过程的关联规则挖掘(GPAR)、贝叶斯关联规则挖掘(BARM)、基于多臂老虎机的关联规则挖掘(MAB-ARM)以及基于强化学习的关联规则挖掘(RLAR)。这些方法突破传统频率驱动算法(如Apriori、FP-Growth、Eclat)的局限,支持先验知识融入、不确定性建模、物品依赖关系刻画、概率推理及自适应搜索策略。GPAR利用高斯过程通过特征表示建模物品共现,实现可解释的推理、不确定性量化,并无需重新训练即可泛化至未见项集。BARM采用贝叶斯框架,结合先验与可选相关结构,通过完整后验分布实现对物品出现概率的稳健不确定性估计。MAB-ARM及其蒙特卡洛树搜索(MCTS)变体使用上置信界(UCB)策略,高效自适应探索项集空间;RLAR则应用深度Q网络(DQN)学习可泛化的高质量规则识别策略。实验证明,这些方法在合成与真实数据集上均有效提升挖掘灵活性与鲁棒性,尤其擅长发现稀有或复杂模式,在小样本场景下表现更优。同时揭示了计算复杂度与可解释性间的权衡。该研究标志着从静态频率范式向融合先验、依赖关系、不确定性感知或可扩展框架的重大转变,适用于零售、地理、金融、医疗诊断及风险敏感场景。

原文摘要 · Abstract (English)

This work introduces 4 novel probabilistic and reinforcement-driven methods for association rule mining (ARM): Gaussian process-based association rule mining (GPAR), Bayesian ARM (BARM), multi-armed bandit based ARM (MAB-ARM), and reinforcement learning based association rule mining (RLAR). These methods depart fundamentally from traditional frequency-based algorithms such as Apriori, FP-Growth, and Eclat, offering enhanced capabilities for incorporating prior knowledge, modeling uncertainty, item dependencies, probabilistic inference and adaptive search strategies. GPAR employs Gaussian processes to model item co-occurrence via feature representations, enabling principled inference, uncertainty quantification, and efficient generalization to unseen itemsets without retraining. BARM adopts a Bayesian framework with priors and optional correlation structures, yielding robust uncertainty quantification through full posterior distributions over item presence probabilities. MAB-ARM, including its Monte Carlo tree search (MCTS) companion, utilizes an upper confidence bound (UCB) strategy for efficient and adaptive exploration of the itemset space, while RLAR applies a deep Q-network (DQN) to learn a generalizable policy for identifying high-quality rules. Collectively, these approaches improve the flexibility and robustness of ARM, particularly for discovering rare or complex patterns and operating on small datasets. Empirical results on synthetic and real-world datasets demonstrate their effectiveness, while also highlighting trade-offs in computational complexity and interpretability. These innovations mark a significant shift from static, frequency-driven paradigms, offering some prior and dependency-informed, uncertainty-aware or scalable ARM frameworks for diverse application domains such as retail, geography, finance, medical diagnostics, and risk-sensitive scenarios.

关联规则概率建模强化学习小样本

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