Improved Weighted Random Forest for Classification Problems

Autor: Mohsen Shahhosseini, Guiping Hu
Rok vydání: 2021
Předmět:
Zdroj: Advances in Intelligent Systems and Computing ISBN: 9783030665005
DOI: 10.1007/978-3-030-66501-2_4
Popis: Several studies have shown that combining machine learning models in an appropriate way will introduce improvements in the individual predictions made by the base models. The key in making well-performing ensemble model is in the diversity of the base models. Of the most common solutions for introducing diversity into the decision trees are bagging and random forest. Bagging enhances the diversity by sampling with replacement and generating many training data sets, while random forest adds selecting random number of features as well. This has made random forest a winning candidate for many machine learning applications. However, assuming equal weights for all base decision trees does not seem reasonable as the randomization of sampling and input feature selection may lead to different levels of decision-making abilities across base decision trees. Therefore, we propose several algorithms that intend to modify the weighting strategy of regular random forest and consequently make better predictions. The designed weighting frameworks include optimal weighted random forest based on accuracy, optimal weighted random forest based on area under the curve (AUC), performance-based weighted random forest, and several stacking-based weighted random forest models. The numerical results show that the proposed models are able to introduce significant improvements compared to regular random forest.
Databáze: OpenAIRE