arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2011.07640stat.MLcs.LG

精确率-召回率曲线(PRC)分类树

Precision-Recall Curve (PRC) Classification Trees

Stony Brook University(石溪大学)

Jiaju Miao, Wei Zhu

AI总结:

本文提出基于AUPRC的PRC分类树及PRC随机森林,通过改进变量与阈值选择提升不平衡数据分类性能,并结合ROC树增强少数类识别。

摘要:

不平衡数据的分类给大多数知名分类算法带来了重大挑战,因为这些算法通常是为类别分布相对平衡的数据而设计的。然而,偏斜的类别分布是现实世界问题中的一个常见特征。它在某些对机器学习和更好的预测分析有巨大需求的应用领域尤为普遍,例如疾病诊断、欺诈检测、破产预测和嫌疑人识别。在本文中,我们提出了一种基于精确率-召回率曲线下面积(AUPRC)的新型树基算法,用于分类场景中的变量选择。我们的算法被命名为“Precision-Recall Curve classification tree(精确率-召回率曲线分类树)”,或简称为“PRC classification tree(PRC分类树)”,它修改了树构建过程中的两个关键阶段。第一个阶段是在节点变量选择中最大化精确率-召回率曲线下面积。第二个阶段是在阈值选择中最大化召回率和精确率的调和平均数(F-measure)。我们发现,所提出的PRC分类树及其后续扩展PRC random forest(PRC随机森林)尤其适用于类别不平衡数据集。我们已经证明,对于合成数据和真实数据,我们的方法都优于其经典对应方法,即通常的CART和随机森林。此外,我们团队先前提出的ROC分类树在不平衡数据中表现出良好性能。它们的组合,即PRC-ROC tree,在识别少数类方面也显示出巨大前景。

英文摘要:

The classification of imbalanced data has presented a significant challenge for most well-known classification algorithms that were often designed for data with relatively balanced class distributions. Nevertheless skewed class distribution is a common feature in real world problems. It is especially prevalent in certain application domains with great need for machine learning and better predictive analysis such as disease diagnosis, fraud detection, bankruptcy prediction, and suspect identification. In this paper, we propose a novel tree-based algorithm based on the area under the precision-recall curve (AUPRC) for variable selection in the classification context. Our algorithm, named as the "Precision-Recall Curve classification tree", or simply the "PRC classification tree" modifies two crucial stages in tree building. The first stage is to maximize the area under the precision-recall curve in node variable selection. The second stage is to maximize the harmonic mean of recall and precision (F-measure) for threshold selection. We found the proposed PRC classification tree, and its subsequent extension, the PRC random forest, work well especially for class-imbalanced data sets. We have demonstrated that our methods outperform their classic counterparts, the usual CART and random forest for both synthetic and real data. Furthermore, the ROC classification tree proposed by our group previously has shown good performance in imbalanced data. The combination of them, the PRC-ROC tree, also shows great promise in identifying the minority class.

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

↑