• Laser & Optoelectronics Progress
  • Vol. 60, Issue 2, 0215002 (2023)
Lei Xiao* and Zongmiao Lan
Author Affiliations
  • College of Automation, Guangdong Polytechnic Normal University, Guangzhou 510665, Guangdong, China
  • show less
    DOI: 10.3788/LOP212628 Cite this Article Set citation alerts
    Lei Xiao, Zongmiao Lan. Identification of Sewage Microorganisms Using Attention Mechanism[J]. Laser & Optoelectronics Progress, 2023, 60(2): 0215002 Copy Citation Text show less

    Abstract

    To accurately identify microorganism species in the activated sludge of sewage treatment systems and modify the wastewater treatment process in real-time, using traditional machine learning methods is a challenge because of various complicated processes. In this study, a deep learning approach based on the integration of attention mechanism and transfer learning is proposed to accurately identify the species of microorganisms in sewage-activated sludge by overcoming the requirements of developing features manually, extracting features, designing classifiers, and other complicated processes. On the basis of transfer learning, the conventional VGG16 model is enhanced by including the attention module (SE-Net block) and modifying the output module, and the dataset is expanded using the data improvement approach. Experimental findings demonstrate that compared with the model before the enhancement, the enhanced model (T-SE-VGG16) can accurately recognize microorganisms in various types of sewage-activated sludge with a test accuracy of 98.21%, which enhances the recognition accuracy and reduces the training time. The model converges rapidly and has a strong generalization ability in terms of training time. Moreover, the T-SE-VGG16 model’s feasibility and reliability for the identification of microorganisms in sewage-activated sludge are verified.
    Lei Xiao, Zongmiao Lan. Identification of Sewage Microorganisms Using Attention Mechanism[J]. Laser & Optoelectronics Progress, 2023, 60(2): 0215002
    Download Citation