HSoMLSDP: A Hybrid Swarm-Optimized Machine Learning Framework for Software Defect Prediction

Abstract

This research aims to design a hybrid swarm-optimized machine learning software defect prediction (HSoMLSDP) framework to predict software defects. We strive to do this by designing a swarm-optimized machine learning defect prediction (SoMLDP) model within the HSoMLSDP framework. In pursuit of enhancing the defect prediction accuracy of the SoMLDP model, this paper designed two novel hybrid swarm-optimization algorithms (SOAs) referred to as gravitational force grasshopper optimization algorithm-artificial bee colony (GFGOA-ABC), and levy flight grasshopper optimization algorithm-artificial bee colony (LFGOA-ABC) algorithms. By combining the enhanced exploration features of LFGOA and GFGOA with the robust exploitation capacity of the artificial bee colony (ABC), the LFGOA-ABC and GFGOA-ABC algorithms are proposed. Prior to validating the HSoMLSDP framework, the LFGOA-ABC and GFGOA-ABC algorithm’s efficacy is first confirmed by experimenting on 19 benchmark functions (BFs) to assess their mean, standard deviation (SD) of optimal values, convergence rate, and convergence rate improvements. Following BFs verification, the second experiment tunes the hyperparameters of the ML models (artificial neural network, XGBOOST) to improve the defect accuracy of the SoMLDP model. The outcomes of the experiments justify a more rapid convergence rate for BFs and notable enhancements of 0.01-0.28 in software defect prediction (SDP) accuracy for NASA defect datasets when compared with state-of-the-art methods. As an enhancement of accuracy justifies the correctness of the SoMLDP model, thus validating the HSoMLSDP framework. © 2013 IEEE.