Underwater Image Classification Algorithm Based on Convolutional Neural Network and Optimized Extreme Learning Machine

Junyi Yang, Mudan Cai, Xingfan Yang, Zhiyu Zhou

In order to deal with the target recognition in the complex underwater environment, we carried out experimental research. This includes filtering noise in the feature extraction stage of underwater images rich in noise, or with complex backgrounds, and improving the accuracy of target classification in the recognition process. This paper discusses our contribution to improving the accuracy of underwater target classification. This paper proposes an underwater target classification algorithm based on the improved flow direction algorithm (FDA) and search agent strategy, which can simultaneously optimize the weight parameters, bias parameters, and super parameters of the extreme learning machine (ELM). As a new underwater target classifier, it replaces the full connection layer in the traditional classification network to build a classification network. In the first stage of the network, the DenseNet201 network pre-trained by ImageNet is used to extract features and reduce dimensions of underwater images. In the second stage, the optimized ELM classifier is trained and predicted. In order to weaken the uncertainty caused by the random input weight and offset of the introduced ELM, the fuzzy logic, chaos initialization, and multi population strategy-based flow direction algorithm (FCMFDA) is used to adjust the input weight and offset of the ELM and optimize the super parameters with the search agent strategy at the same time. We tested and verified the FCMFDA-ELM classifier on Fish4Knowledge and underwater robot professional competition 2018 (URPC 2018) datasets, and achieved 99.4% and 97.5% accuracy, respectively. The experimental analysis shows that the FCMFDA-ELM underwater image classifier proposed in this paper has a greater improvement in classification accuracy, stronger stability, and faster convergence. Finally, it can be embedded in the recognition process of underwater targets to improve the recognition performance and efficiency.