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

基于主动特征获取的高效在线决策树学习

Efficient Online Decision Tree Learning with Active Feature Acquisition

Chalmers University of Technology(查尔姆斯理工大学);University of Chicago(芝加哥大学)

Arman Rahbar, Ziyu Ye, Yuxin Chen, Morteza Haghir Chehreghani

AI总结:

针对特征和标签需成本获取的在线决策树构建问题,提出嵌入主动规划预言机的在线学习框架,用自适应次模代理函数低成本主动查特征、后验采样保低遗憾,经实验验证高效有效且适配概念漂移场景。

摘要:

在线构建决策树是机器学习领域的经典问题。现有研究通常假设每个流入的数据点的特征都是现成可用的。但在许多实际应用中,特征值和标签事先都是未知的,且只能通过付出成本获取。例如在医疗诊断场景中,医生必须选择对患者进行哪些检查(即执行有成本的特征查询),才能做出诊断决策(即预测标签)。我们为应对这一实际挑战提出了全新视角。我们的框架包含一个嵌入在线学习方案中的主动规划预言机,针对该框架我们研究了多种信息获取函数。具体而言,我们采用基于自适应次模性的代理信息获取函数,以最低成本主动查询特征值,同时使用后验采样方案保持在线预测的低遗憾度。我们在多个真实世界数据集上开展了大量实验,验证了该框架的效率与有效性。我们的框架还能自然适配带概念漂移的高难度在线学习场景,实验表明其性能与基线模型相当,且灵活性更强。

英文摘要:

Constructing decision trees online is a classical machine learning problem. Existing works often assume that features are readily available for each incoming data point. However, in many real world applications, both feature values and the labels are unknown a priori and can only be obtained at a cost. For example, in medical diagnosis, doctors have to choose which tests to perform (i.e., making costly feature queries) on a patient in order to make a diagnosis decision (i.e., predicting labels). We provide a fresh perspective to tackle this practical challenge. Our framework consists of an active planning oracle embedded in an online learning scheme for which we investigate several information acquisition functions. Specifically, we employ a surrogate information acquisition function based on adaptive submodularity to actively query feature values with a minimal cost, while using a posterior sampling scheme to maintain a low regret for online prediction. We demonstrate the efficiency and effectiveness of our framework via extensive experiments on various real-world datasets. Our framework also naturally adapts to the challenging setting of online learning with concept drift and is shown to be competitive with baseline models while being more flexible.

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

↑