Deep convolutional neural networks for construction and demolition waste classification: VGGNet structures, cyclical learning rate, and knowledge transfer
Kunsen Lin,Tao Zhou,Xiaofeng Gao,Zongshen Li,Huabo Duan,Huanyu Wu,Guanyou Lu,Youcai Zhao
DOI: https://doi.org/10.1016/j.jenvman.2022.115501
IF: 8.7
2022-09-15
Journal of Environmental Management
Abstract:The sorting of Construction and Demolition (C&D) waste is a critical step to linking the recycling system and to the macro prediction, which helps to promote the development of the circular economy. Moreover, the effective classification and automated separation process will also help to stop the spreading of pathogenic organisms, such as virus and bacteria, by minimizing human intervention in the sorting process, while also helping to prevent further contamination by COVID-19 virus. This study aims to develop an efficient method to sort C&D waste through deep learning combined with knowledge transfer approach. In this paper, CVGGNet models, that is four VGG structures (VGGNet-11, VGGNet-13, VGGNet-16, and VGGNet-19), based on knowledge transfer combined with the technology of data augmentation and cyclical learning rate, are proposed to classify ten types of C&D waste images. Results show that 2.5 × 10−4, 1.8 × 10−4, 0.8 × 10−4, and 1.0 × 10−4 are the optimum learning rate for CVGGNet-11, CVGGNet-13, CVGGNet-16, and CVGGNet-19, respectively. Knowledge transfer helped shorten the training time from 1039.45 s to 991.05 s, and while it improved the performance of the CVGGNet-11 model in training, validation, and test datasets. The average training time increases as the number of the layers in the CVGGNet architecture rises: CVGGNet-11 (991.05 s) ˂ CVGGNet-13 (1025.76 s) ˂ CVGGNet-16 (1090.48 s) ˂ CVGGNet-19 (1337.81 s). Compared to other CVGGNet models, CVGGNet-16 showed an excellent performance in various C&D waste types, in terms of accuracy (76.6%), weighted average precision (76.8%), weighted average recall (76.6%), weighted average F1-score (76.6%) and micro average ROC (87.0%). In addition, the t-distributed Stochastic Neighbor Embedding (t-SNE) approach can reduce the dataset to a lower dimension and distinctly separate each type of C&D waste. This study demonstrates the good performance of CVGGNet models that can be used to automatically sort most of the C&D waste, paving the way for better C&D waste management.
environmental sciences