基于注意力机制残差神经网络的近红外芒果种类定性建模方法

doi:10.3964/j.issn.1000-0593(2024)08-2262-06

Abstract
References
Related Citation (15)

Download: PDF (4115 KB)
Export: BibTeX | EndNote (RIS)

Abstract In recent years, the rapid development of modern spectral detection technology is closely related to deep learning. As an end-to-end model, the deep neural network can get more information from the spectra, thus improving the robustness of the model. A one-dimensional residual neural network (1D-AD-ResNet-18) model based on a convolutional block attention module was proposed to explore the feasibility of qualitative prediction of mango species by near-infrared spectroscopy combined with deep learning. Firstly, to reduce the interference of redundant information in the spectra, the CBAM convolution attention module is added to the traditional one-dimensional residual neural network, which can focus on the local useful information of the spectra. Secondly, to avoid the disappearance of gradient and the occurrence of overfitting, ResNet-18 is used to solve the problem of network “degradation”. For 186 mango samples, 70% of the samples were trained, and 30% were tested. Accuracy, Precision, Recall, F1-score, Macro-average, and weighted average were used as evaluation indexes of the model. Three comparison models were established, including traditional one-dimensional ResNet-18, SNV-SVM, and PCA-KNN. Compared with the above three methods, the established 1D-AD-ResNet-18 model obtained the optimal prediction results, and the accuracy of the four qualitative analysis models was 96.42%，80.35%，76.78% and 67.85%. The experimental results show that the 1D-AD-ResNet-18 model can accurately identify and classify mango species, which provides a new idea for the qualitative analysis of mango species by NIR spectroscopy.

Key words：Mango species identification; CBAM attention mechanism; Near-infrared spectroscopy; Residual network

Received: 2023-05-08 Accepted: 2023-11-07

ZTFLH:

O657.33

Corresponding Authors: WAN Jin-cong E-mail: wjcym@outlook.com

Cite this article:

WANG Shu-tao,WAN Jin-cong,LIU Shi-yu, et al. Qualitative Modeling Method of Mango Species in Near Infrared Based on Attention Mechanism Residual Neural Network[J]. SPECTROSCOPY AND SPECTRAL ANALYSIS, 2024, 44(08): 2262-2267.

URL:

https://www.gpxygpfx.com/EN/10.3964/j.issn.1000-0593(2024)08-2262-06 OR https://www.gpxygpfx.com/EN/Y2024/V44/I08/2262