Performance Enhancement of Kernel Extreme Learning Machine Using Whale Optimization Algorithm in Fruit Image Classification
Hengren Xu · 2023
This paper presents an evaluation of the Extreme Learning Machine (ELM), Kernel Extreme Learning Machine (KELM), and the Kernel Extreme Learning Machine optimized by the Whale Optimization Algorithm (WOA-KELM) in the context of fruit image classification. Fruits play a crucial role in everyday life, and the application of machine learning for fruit classification can significantly improve efficiency. The ELM, a single hidden layer feedforward neural network, is the foundation for this comparison. The KELM, an evolution of the ELM, incorporates a kernel function to map input data into a high-dimensional feature space, thereby enhancing performance. The development of WOA-KELM involves optimizing the regularization parameter in KELM and the width parameter of the radial basis function through the Whale Optimization Algorithm, further enhancing its capabilities. In classifying twelve types of fruit images, the ELM achieved a classification accuracy of 77%. In contrast, the WOA-KELM, as proposed in this paper, significantly outperforms with a remarkable classification accuracy of 92%.