Noise-Assisted Data Enhancement Promoting Image Classification of Municipal Solid Waste

Chao Chen,Rui Liang,Mengyao Song,Ziyi Zhang,Junyu Tao,Beibei Yan,Zhanjun Cheng,Guanyi Chen
DOI: https://doi.org/10.1016/j.resconrec.2024.107790
IF: 13.716
2024-01-01
Resources Conservation and Recycling
Abstract:Data enhancement is an emerging approach to solve the limited sample size for computer vision-based municipal solid waste (MSW) classification. To overcome the unmatched features between real MSW pictures and those generated by data enhancement, this paper examined the potential of noise-assisted data enhancement methods on two representative MSW classification models: Convolutional Neural Networks (CNN) and Deep Residual Shrinkage Networks (DRSN). The experimental results showed that data enhancement significantly improved the model's accuracy compared with no data enhancement. Specifically, CNN model's test accuracy increased from 64.56 % to 81.78 %, while DRSN model rose from 70.30 % to 90.32 %. Notably, DRSN demonstrated exceptional robustness under noise conditions, achieving over 90 % accuracy in training and test sets. Analysis of the confusion matrix revealed the accuracy significantly improved in identifying glass, plastic, and metal under noise enhancement. These findings emphasized the importance of enhancement strategies in noisy industrial scenarios, offering guidance for MSW management.
What problem does this paper attempt to address?