Laser & Optoelectronics Progress, Volume. 60, Issue 2, 0215002(2023)

Identification of Sewage Microorganisms Using Attention Mechanism

Lei Xiao* and Zongmiao Lan
Author Affiliations
  • College of Automation, Guangdong Polytechnic Normal University, Guangzhou 510665, Guangdong, China
  • show less

    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.

    Tools

    Get Citation

    Copy Citation Text

    Lei Xiao, Zongmiao Lan. Identification of Sewage Microorganisms Using Attention Mechanism[J]. Laser & Optoelectronics Progress, 2023, 60(2): 0215002

    Download Citation

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

    Category: Machine Vision

    Received: Sep. 28, 2021

    Accepted: Nov. 8, 2021

    Published Online: Jan. 6, 2023

    The Author Email: Xiao Lei (2650782969@qq.com)

    DOI:10.3788/LOP212628

    Topics