Enhancing trash classification in smart cities using federated deep learning

• Published in: Scientific reports Vol. 14; no. 1; p. 11816
• Main Authors: Ahmed Khan, Haroon, Naqvi, Syed Saud, Alharbi, Abeer A. K., Alotaibi, Salihah, Alkhathami, Mohammed
• Format: Journal Article
• Language: English
• Published: London Nature Publishing Group UK 23.05.2024; Nature Publishing Group; Nature Portfolio
• Subjects: 639/705/117; 639/705/258; Classification; Comparative analysis; Convolutional neural network; Deep learning; Deep neural network; Humanities and Social Sciences; multidisciplinary; Neural networks; Recycling; Science; Science (multidisciplinary); Smart cities; Solid waste management; Solid wastes; Training; Urban environments; Waste management; Deep neural network; Recycling; Solid waste management; Convolutional neural network; Classification
• ISSN: 2045-2322; 2045-2322
• DOI: 10.1038/s41598-024-62003-4

Efficient Waste management plays a crucial role to ensure clean and green environment in the smart cities. This study investigates the critical role of efficient trash classification in achieving sustainable solid waste management within smart city environments. We conduct a comparative analysis of various trash classification methods utilizing deep learning models built on convolutional neural networks (CNNs). Leveraging the PyTorch open-source framework and the TrashBox dataset, we perform experiments involving ten unique deep neural network models. Our approach aims to maximize training accuracy. Through extensive experimentation, we observe the consistent superiority of the ResNext-101 model compared to others, achieving exceptional training, validation, and test accuracies. These findings illuminate the potential of CNN-based techniques in significantly advancing trash classification for optimized solid waste management within smart city initiatives. Lastly, this study presents a distributed framework based on federated learning that can be used to optimize the performance of a combination of CNN models for trash detection.