Journal Article

MSWNet: A visual deep machine learning method adopting transfer learning based upon ResNet 50 for municipal solid waste sorting

Cui, Feifei, Lin, Kunsen, Wang, Lina, Zhou, Tao, Zhao, Youcai, Shi, Wenjie

2023

Overview

● MSWNet was proposed to classify municipal solid waste. ● Transfer learning could promote the performance of MSWNet. ● Cyclical learning rate was adopted to quickly tune hyperparameters. An intelligent and efficient methodology is needed owning to the continuous increase of global municipal solid waste (MSW). This is because the common methods of manual and semi-mechanical screenings not only consume large amount of manpower and material resources but also accelerate virus community transmission. As the categories of MSW are diverse considering their compositions, chemical reactions, and processing procedures, etc., resulting in low efficiencies in MSW sorting using the traditional methods. Deep machine learning can help MSW sorting becoming into a smarter and more efficient mode. This study for the first time applied MSWNet in MSW sorting, a ResNet-50 with transfer learning. The method of cyclical learning rate was taken to avoid blind finding, and tests were repeated until accidentally encountering a good value. Measures of visualization were also considered to make the MSWNet model more transparent and accountable. Results showed transfer learning enhanced the efficiency of training time (from 741 s to 598.5 s), and improved the accuracy of recognition performance (from 88.50% to 93.50%); MSWNet showed a better performance in MSW classsification in terms of sensitivity (93.50%), precision (93.40%), F1-score (93.40%), accuracy (93.50%) and AUC (92.00%). The findings of this study can be taken as a reference for building the model MSW classification by deep learning, quantifying a suitable learning rate, and changing the data from high dimensions to two dimensions.

• Publisher: Higher Education Press,Springer Nature B.V
• Subject: artificial intelligence / Chemical composition / Chemical reactions / Cyclic learning rate / Deep learning / Deep residual network / Earth and Environmental Science / Environment / Learning algorithms / Machine learning / Manpower / materials / Municipal solid waste / Municipal solid waste sorting / Municipal waste management / Research Article / screening / Solid waste management / Solid wastes / Transfer learning / virus transmission / Visual discrimination learning / Visualization / Waste sorting