Probing 1D convolutional neural network adapted to near-infrared spectroscopy for efficient classification of mixed fish

Abstract

Salmon and Cod are economically significant world-class fish that have high economic value. It is difficult to accurately sort and process them by appearance during harvest and transportation. Conventional chemical detection means are time-consuming and costly, which greatly affects the cost and efficiency of Fishery production. Therefore, there is an urgent need for smart Fisheries methods which use for the classification of mixed fish. In this paper, near-infrared spectroscopy (NIRS) was used to assess salmon and cod samples. This study aims to evaluate feasibility of a back-propagation neural network (BPNN) and a convolutional neural network (CNN) for identifying different species of fishes by the corresponding spectra in comparison to traditional chemometrics Partial Least Squares. After comparing the effects of different batch sizes, number of convolutional kernels, number of convolutional layers, and number of pooling layers on the classification of NIRS spectra comparing different structures of one-dimensional (1D)–CNN, we propose the 1D-CNN-8 model that is most suitable for the classification of mixed fish. Compared with the results of traditional chemometrics methods and BPNN, the prediction model of the 1D-CNN model can reach 98.00% Accuracy and the parameters are significantly better than others. Meanwhile, the parameters and floating-point operations of the optimal model are both small. Therefore, the improved CNN model based on the NIRS can effectively and quickly identify different kinds of fish samples and contribute to realizing edge computing at the same time.