Random Shapley Forests: Cooperative Game-Based Random Forests With Consistency
- 23 March 2020
- journal article
- research article
- Published by Institute of Electrical and Electronics Engineers (IEEE) in IEEE Transactions on Cybernetics
- Vol. 52 (1) , 205-214
Abstract
The original random forests (RFs) algorithm has been widely used and has achieved excellent performance for the classification and regression tasks. However, the research on the theory of RFs lags far behind its applications. In this article, to narrow the gap between the applications and the theory of RFs, we propose a new RFs algorithm, called random Shapley forests (RSFs), based on the Shapley value. The Shapley value is one of the well-known solutions in the cooperative game, which can fairly assess the power of each player in a game. In the construction of RSFs, RSFs use the Shapley value to evaluate the importance of each feature at each tree node by computing the dependency among the possible feature coalitions. In particular, inspired by the existing consistency theory, we have proved the consistency of the proposed RFs algorithm. Moreover, to verify the effectiveness of the proposed algorithm, experiments on eight UCI benchmark datasets and four real-world datasets have been conducted. The results show that RSFs perform better than or at least comparable with the existing consistent RFs, the original RFs, and a classic classifier, support vector machines.Keywords
Funding Information
- Major Project for New Generation of AI (2018AAA0100400)
- National Natural Science Foundation of China (41706010)
- Science and Technology Program of Qingdao (17-3-3-20-nsh)
- CERNET Innovation Project (NGII20170416)
- Joint Fund of the Equipments Preresearch and Ministry of Education of China (6141A020337)
- Open Fund of Engineering Research Center for Medical Data Mining and Application of Fujian Province (MDM2018007)
- Fundamental Research Funds for the Central Universities of China
This publication has 40 references indexed in Scilit:
- Using cooperative game theory to optimize the feature selection problemNeurocomputing, 2012
- Variance reduction in purely random forestsJournal of Nonparametric Statistics, 2012
- Computational Aspects of Cooperative Game TheorySynthesis Lectures on Artificial Intelligence and Machine Learning, 2011
- Classification and regression treesWIREs Data Mining and Knowledge Discovery, 2011
- Consistency of random survival forestsStatistics & Probability Letters, 2010
- Ensemble-based classifiersArtificial Intelligence Review, 2009
- The random subspace method for constructing decision forestsIEEE Transactions on Pattern Analysis and Machine Intelligence, 1998
- C4.5: Programs for Machine Learning by J. Ross Quinlan. Morgan Kaufmann Publishers, Inc., 1993Machine Learning, 1994
- Elements of Information TheoryJournal of Modern Optics, 1992
- Neural network ensemblesIEEE Transactions on Pattern Analysis and Machine Intelligence, 1990