• Laser & Optoelectronics Progress
  • Vol. 62, Issue 7, 0730004 (2025)
Xueling Li1, Jing Yu1, Haiyang Zhang2, Lu Dong3..., Zhengdong Zhang2, Ke Li2, Yaqin Yu2 and Qi Li2,*|Show fewer author(s)
Author Affiliations
  • 1College of Metrology Measurement and Instrument, China Jiliang University, Hangzhou 310018, Zhejiang , China
  • 2National Institute of Metrology, China, Beijing 100029, China
  • 3Liaoning Inspection, Examination & Certification Centre, Shenyang 110032, Liaoning , China
  • show less
    DOI: 10.3788/LOP242165 Cite this Article Set citation alerts
    Xueling Li, Jing Yu, Haiyang Zhang, Lu Dong, Zhengdong Zhang, Ke Li, Yaqin Yu, Qi Li. Improved Attention Mechanism MobileNetV2 Network for SERS Classification of Water Pollution[J]. Laser & Optoelectronics Progress, 2025, 62(7): 0730004 Copy Citation Text show less

    Abstract

    To address the necessity for rapid identification of pollutants in abrupt water-pollution incidents, a lightweight neural network algorithm suitable for portable devices is proposed. By obtaining surface-enhanced Raman spectroscopy (SERS) data of five common water pollutants and performing preprocessing, a two-dimensional Morlet wavelet transform is applied to separate high- and low-frequency signals, thus enhancing feature representation. To improve the model's feature-extraction capability, a multipooling strategy is introduced, and the efficient channel attention (ECA) mechanism is modified to develop a multipooling attention ECA (MP_ECA) module. This module is integrated with the MobileNetV2 network to construct the MobileNetV2_MP_ECA model for wavelet image classification and recognition. The gradient-weighted class activation mapping (Grad-CAM) technique is utilized to generate heatmaps, which further verifies the effectiveness of wavelet transform in enhancing feature extraction and classification accuracy. Experimental results show that the proposed model achieves a classification accuracy of 97.83%, thus outperforming other attention mechanism models, conventional convolutional neural networks, and common machine-learning methods. Additionally, the model size of proposed model is only 6.11 MB and incurs a floating-point computation of 230.20 MFLOPs, thus rendering it suitable for resource-constrained mobile-device applications. This study provides a novel strategy and approach for efficiently detecting pollutants in real-world abrupt water-pollution scenarios.
    Xueling Li, Jing Yu, Haiyang Zhang, Lu Dong, Zhengdong Zhang, Ke Li, Yaqin Yu, Qi Li. Improved Attention Mechanism MobileNetV2 Network for SERS Classification of Water Pollution[J]. Laser & Optoelectronics Progress, 2025, 62(7): 0730004
    Download Citation