Attention ConvMixer Model and Application for Fish Species Classification

Abstract

Exploring the ocean has always been one of the foremost challenges for humankind, and fish classification is one of the crucial tasks in this endeavor. Manual fish classification methods, although accurate, consume significant time, money, and effort, while computer-based methods such as image processing and traditional machine learning often fall short of achieving high accuracy. Recently, deep convolutional neural networks have demonstrated their capability to ensure both time efficiency and accuracy in this task. However, deep convolutional networks typically have a large number of parameters, requiring substantial training time, and the convolutional operations lack attentional mechanisms. Therefore, in this paper, we propose the AttentionConvMixer neural network with Priority Channel Attention (PCA) and Priority Spatial Attention (PSA). The proposed approach exhibits good performance across all three fish classification datasets without introducing any additional parameters, thus demonstrating the effectiveness of our proposed method.