Multi-spectral fusion and self-attention mechanisms for Gentiana origin identification via near-infrared spectroscopy

Abstract

Gentiana is rich in Gentiopicroside and strychnine acid with medicinal value. However, the active ingredients of Gentiana from different origins are different, so identifying Gentian's origin is significant. Currently, neural networks such as CNN and GRU are widely used for spectral data analysis, but the modeling effect is easily affected by the spectral preprocessing method, and the long region and many features of spectral data make it difficult for CNN models to capture the long-term dependence of spectra, while GRU modeling has a large number of parameters, high computational complexity, and low efficiency. Therefore, a Gentian Root Data Fusion Module (GL) for sequence data is proposed to achieve the fusion between spectral data under different pre-processing by assigning weights to multiple pre-processing data and all features of pre-processing data respectively, making full use of the advantages of different pre-processing methods. Aiming at the characteristics of the long spectral data region, the joint architecture of convolutional neural network (CNN) and gated neural network (GRU) is adopted to achieve the extraction of features and the capture of long-term dependencies, while reducing the model complexity. Finally, GL is integrated with CNN and GRU to craft the advanced collaborative framework known as CCRN. The experimental findings demonstrate that CCRN outperforms CNN + GRU, CNN, PLS-DA, and SVM in terms of accuracy and loss function performance. Notably, CCRN exhibits superior Accuracy, Recall, and F1-score, surpassing the CNN + GRU model by 2.4 %, 2.1 %, and 2.1 %, respectively. These results validate the efficacy of the GL module in seamlessly integrating various preprocessing methods. In addition, the model CCRN still performs best when tested on public datasets, proving that CCRN has good Portability and scalability.