Author(s)

Qingqing He¹, Chao Qin²

¹School of Business Administration, Shandong University of Finance and Economics, Jinan, China.
²School of Computer Science and Technology, Shandong University of Finance and Economics, Jinan, China.

In order to solve the problem that, the hyper-parameters of the existing random forest-based classification prediction model depend on empirical settings, which leads to unsatisfactory model performance. We propose a based on adaptive particle swarm optimization algorithm random forest model to optimize data classification and an adaptive particle swarm algorithm for optimizing hyper-parameters in the random forest to ensure that the model can better predict unbalanced data. Aiming at the premature convergence problem in the particle swarm optimization algorithm, the population is adaptively divided according to the fitness of the population, and an adaptive update strategy is introduced to enhance the ability of particles to jump out of the local optimum. The main steps of the model are as follows: Normalize the data set, initialize the model on the training set, and then use the particle swarm optimization algorithm to optimize the modeling process to establish a classification model. Experimental results show that our proposed algorithm is better than traditional algorithms, especially in terms of F1-Measure and ACC evaluation standards. The results of the six-keel imbalanced data set demonstrate the advantages of our proposed algorithm.

Random Forest, APSO, Unbalanced Data, Parameter Optimization

He, Q. and Qin, C. (2021) Adaptive Optimization Swarm Algorithm Ensemble Model Applied to the Classification of Unbalanced Data. Intelligent Information Management, 13, 251-267. doi: 10.4236/iim.2021.135014.