arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2404.06403cs.LG
收藏

基于汤普森采样的决策树在线学习

Online Learning of Decision Trees with Thompson Sampling

LIX, Ecole Polytechnique, IP Paris(巴黎综合理工学院LIX实验室,巴黎理工学院);AI Institute, University of Waikato(怀卡托大学人工智能研究所);LTCI, Télécom Paris, IP Paris(巴黎高等电信学院LTCI实验室,巴黎理工学院)

Ayman Chaouki, Jesse Read, Albert Bifet

AI总结:

针对现有批量决策树算法无法适配在线流数据且贪心策略难保证全局最优的问题,提出基于汤普森采样的在线决策树算法TSDT,证明其几乎必然收敛到最优树,且在多个基准上性能优于现有算法。

摘要:

决策树是可解释机器学习领域的主流预测模型,目前已得到深入研究,但大多围绕固定标记数据集的批量场景展开,由此催生了C4.5、ID3、CART等主流算法。遗憾的是,这些方法本质上属于启发式方法,依赖贪心分裂策略,无法保证全局最优性,还往往会生成结构过于复杂、难以解释的决策树。近期的研究突破解决了批量场景下的这种次优性问题,但尚无相关工作关注数据以流形式到达的在线场景。针对这一空白,我们设计了一种全新的蒙特卡洛树搜索算法——汤普森采样决策树(TSDT),能够在在线场景下生成最优决策树。我们对该算法进行了理论分析,证明其几乎必然收敛到最优树。此外,我们开展了大量实验以实证验证上述结论。所提出的TSDT在多个基准测试上的表现优于现有算法,同时还具备专为在线场景设计的实用优势。

英文摘要:

Decision Trees are prominent prediction models for interpretable Machine Learning. They have been thoroughly researched, mostly in the batch setting with a fixed labelled dataset, leading to popular algorithms such as C4.5, ID3 and CART. Unfortunately, these methods are of heuristic nature, they rely on greedy splits offering no guarantees of global optimality and often leading to unnecessarily complex and hard-to-interpret Decision Trees. Recent breakthroughs addressed this suboptimality issue in the batch setting, but no such work has considered the online setting with data arriving in a stream. To this end, we devise a new Monte Carlo Tree Search algorithm, Thompson Sampling Decision Trees (TSDT), able to produce optimal Decision Trees in an online setting. We analyse our algorithm and prove its almost sure convergence to the optimal tree. Furthermore, we conduct extensive experiments to validate our findings empirically. The proposed TSDT outperforms existing algorithms on several benchmarks, all while presenting the practical advantage of being tailored to the online setting.

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

↑