arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2107.08784cs.LGstat.ML
收藏

Boost-R:递归数据的梯度提升树

Boost-R: Gradient Boosted Trees for Recurrence Data

University of Arkansas(阿肯色大学);Arizona State University(亚利桑那州立大学)

Xiao Liu, Rong Pan

摘要:

复发数据来自跨可靠性,网络安全,医疗保健,在线零售等多学科领域。本文研究了一种基于加法树的方法,称为Boost-R(复发数据的提升),用于具有静态和动态特征的复发事件数据。Boost-R构建梯度提升的加性树的集合来估计复发事件过程的累积强度函数,其中通过最小化观测和预测累积强度之间的正则化L2距离来将新树添加到集合。与传统的回归树不同,Boost-R在每个树叶上构造了一个时间依赖函数。来自多个树的这些函数的和产生累积强度的集合估计。基于树的方法的分而治之的性质是有吸引力的,当隐藏的子群体存在于一个异质的人口。回归树的非参数性质有助于避免对事件过程和特征之间的复杂交互进行参数假设。Boost-R的关键见解和优势进行了研究,通过全面的数值例子。Boost-R的数据集和计算机代码可在GitHub上获得。据我们所知,Boost-R是第一个基于梯度提升加法树的方法,用于建模具有静态和动态特征信息的大规模复发事件数据。

英文摘要:

Recurrence data arise from multi-disciplinary domains spanning reliability, cyber security, healthcare, online retailing, etc. This paper investigates an additive-tree-based approach, known as Boost-R (Boosting for Recurrence Data), for recurrent event data with both static and dynamic features. Boost-R constructs an ensemble of gradient boosted additive trees to estimate the cumulative intensity function of the recurrent event process, where a new tree is added to the ensemble by minimizing the regularized L2 distance between the observed and predicted cumulative intensity. Unlike conventional regression trees, a time-dependent function is constructed by Boost-R on each tree leaf. The sum of these functions, from multiple trees, yields the ensemble estimator of the cumulative intensity. The divide-and-conquer nature of tree-based methods is appealing when hidden sub-populations exist within a heterogeneous population. The non-parametric nature of regression trees helps to avoid parametric assumptions on the complex interactions between event processes and features. Critical insights and advantages of Boost-R are investigated through comprehensive numerical examples. Datasets and computer code of Boost-R are made available on GitHub. To our best knowledge, Boost-R is the first gradient boosted additive-tree-based approach for modeling large-scale recurrent event data with both static and dynamic feature information.

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

↑