Multi-Scale Feature Attention Network for Rapid and Non-Destructive Quantification of Aflatoxin B1 in Maize Using Hyperspectral Imaging.
Yichi Zhang, Kewei Huan, Xiaoxi Liu, Yuqing Fan, Xianwen Cao, Xueyan Han
Abstract
Open AccessMaize, a globally important crop, is highly susceptible to aflatoxin contamination, posing a serious threat. Therefore, accurate detection of aflatoxin levels in maize is of critical importance. In this study, the Multi-Scale Feature Network with Efficient Channel Attention (MSFNet-ECA) model, based on near-infrared hyperspectral imaging combined with deep learning techniques was developed to analyze the content of aflatoxin B1 (AFB1) in maize. Three data augmentation methods-multiplicative random scaling, bootstrap resampling, and Wasserstein generative adversarial networks (WGAN)-were compared with various preprocessing strategies to assess their impact on model performance. Multiplicative random scaling combined with second derivative (D2) preprocessing yielded the best predictive performance for the MSFNet-ECA model. Using this augmentation, the MSFNet-ECA model outperformed four conventional models (partial least squares regression (PLSR), support vector regression (SVR), extreme learning machine (ELM), and one-dimensional convolutional neural network (1D-CNN)), achieving a root mean square error of prediction (RMSEP) of 2.3 μg·kg-1, coefficient of determination for prediction (Rp2) of 0.99, and the residual predictive deviation (RPD) of 9, with accuracy improvements of 86.4%, 79.1%, 71.3%, and 42.5%, respectively. This finding demonstrates that applying data augmentation methods substantially improves the predictive performance of hyperspectral chemometric models driven by deep learning. Moreover, when combined with data augmentation techniques, the proposed MSFNet-ECA model can accurately predict AFB1 content in maize, offering an efficient and reliable tool for hyperspectral applications in food quality and safety monitoring.