Spectroscopy and Spectral Analysis, Volume. 44, Issue 8, 2262(2024)

Qualitative Modeling Method of Mango Species in Near Infrared Based on Attention Mechanism Residual Neural Network

WANG Shu-tao1... WAN Jin-cong1,*, LIU Shi-yu2, ZHANG Jin-qing1 and WANG Yu-tian1 |Show fewer author(s)
Author Affiliations
  • 1[in Chinese]
  • 2[in Chinese]
  • show less

    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.

    Tools

    Get Citation

    Copy Citation Text

    WANG Shu-tao, WAN Jin-cong, LIU Shi-yu, ZHANG Jin-qing, WANG Yu-tian. Qualitative Modeling Method of Mango Species in Near Infrared Based on Attention Mechanism Residual Neural Network[J]. Spectroscopy and Spectral Analysis, 2024, 44(8): 2262

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Received: May. 8, 2023

    Accepted: --

    Published Online: Oct. 11, 2024

    The Author Email: Jin-cong WAN (wjcym@outlook.com)

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

    Topics