arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2201.12682stat.MLcs.LGstat.APstat.ME

Geometry- and Accuracy-Preserving Random Forest Proximities

Idaho State University(爱达荷州立大学);Utah State University(犹他州立大学)

Jake S. Rhodes, Adele Cutler, Kevin R. Moon

英文摘要:

Random forests are considered one of the best out-of-the-box classification and regression algorithms due to their high level of predictive performance with relatively little tuning. Pairwise proximities can be computed from a trained random forest and measure the similarity between data points relative to the supervised task. Random forest proximities have been used in many applications including the identification of variable importance, data imputation, outlier detection, and data visualization. However, existing definitions of random forest proximities do not accurately reflect the data geometry learned by the random forest. In this paper, we introduce a novel definition of random forest proximities called Random Forest-Geometry- and Accuracy-Preserving proximities (RF-GAP). We prove that the proximity-weighted sum (regression) or majority vote (classification) using RF-GAP exactly matches the out-of-bag random forest prediction, thus capturing the data geometry learned by the random forest. We empirically show that this improved geometric representation outperforms traditional random forest proximities in tasks such as data imputation and provides outlier detection and visualization results consistent with the learned data geometry.

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

↑