Unified Multi-Color-model-learning-based Deep Support Vector Machine for Underwater Image Classification
Weidong Zhang,Baiqiang Yu,Guohou Li,Peixian Zhuang,Zheng Liang,Wenyi Zhao
DOI: https://doi.org/10.1016/j.engappai.2024.109437
IF: 8
2024-01-01
Engineering Applications of Artificial Intelligence
Abstract:Underwater images face various quality degradation issues due to the complex underwater physicalization environment. However, most existing underwater image clarification methods cannot effectively enhance all degradation types of underwater images, which is significant for designing dedicated degradation classes of underwater image clarification methods. To effectively solve the classification issue of underwater images driving the design of specialized underwater image clarification methods, we present a U nified multi-color- model-learning-based D eep S upport V ector M achine (UDSVM) for underwater image classification. Specifically, we propose a multi-color model feature encode strategy, combining features from different color models into a unified feature model to enrich the diversity of feature representation capabilities. Whereafter, we design a deep support vector machine considering the deep network structure with better nonlinear modeling performance. Meanwhile, we embed the unified multi-color model into the deep support vector machine with gradient descent backward optimization, which solves the issues of insufficient learning features and low classification accuracy of the single-color model. Besides, we built a large-scale underwater image classification dataset (UICD) including 6630 underwater images, which have the characteristics of different degradation classes and different degradation levels of the same degradation class. Our UDSVM is compared with NNCO, CSDBN, AlexNet, GoogleNet, ResNet, EfficientNet, VGG19, CoatNet, DAMNet, and MCANet, experiments on UICD demonstrate that our UDSVM achieves the highest classification accuracy of 99.21% . 21% and 98.25% . 25% for rough and fine classifications, respectively. Additionally, our method outperforms the ten compared methods.