Exploration of Data Augmentation in Xception for Waste Classification

Ariza Ikhlas, Syafri Arlis

Abstract


The increasing volume of waste worldwide has led to significant challenges related to pollution, waste management, and recycling. These issues require innovative solutions to enhance the waste management ecosystem, such as the implementation of Smart Waste Management, which leverages information technology and artificial intelligence. This study aims to implement the Xception Convolutional Neural Network (CNN) model for waste classification, explore various data augmentation techniques, and identify optimal model configurations for this task. The research methodology consists of several stages, including data preparation, model building and training, model adaptation for classification tasks, model evaluation, iterative experimentation, and saving and reloading the trained model. The dataset used in this study is the TrashNet dataset obtained from Kaggle, consisting of 2,527 images across several classes: cardboard, glass, metal, paper, plastic, and trash. Based on the optimization process, the selected hyperparameters include a batch size of 32, 64 convolutional filters, the Adam optimizer (learning rate = 0.0001), and a dropout rate of 0.25. After training for 100 epochs, the model achieved a training accuracy of 99% with a loss of 0.7%, and a validation accuracy of 87% with a validation loss of 52%. Evaluation on the test dataset yielded an accuracy of 76%, precision of 79%, recall of 75%, and an F1-score of 75%. The application of data augmentation techniques—such as scaling, translation, and color space transformation—resulted in performance improvements, increasing accuracy by 13%, precision by 11%, recall by 13%, and F1-score by 12%. This study contributes by implementing the Xception model on the TrashNet dataset for waste classification and proposing several data augmentation methods that provide empirical evidence to support or challenge existing approaches. The findings offer practical insights for the development of Smart Waste Management systems, enrich the literature through experimental results, and provide a comparative analysis of data augmentation techniques suitable for the TrashNet dataset.

Keywords


artificial intelligence; computer vision; data augmentation; optimization; smart waste management; xception

Full Text:

PDF

References


KLHK, “Sistem Informasi Pengelolaan Sampah Nasional (SIPSN),” Kementrian Lingkungan Hidup dan Kehutanan.

W. C. Purnomo, Solusi Pengelolaan Sampah Kota. Yogyakarta: Gadja Mada University Press, 2023.

B. Fang et al., “Artificial Intelligence for Waste Management in Smart Cities: A Review,” Aug. 01, 2023, Springer Science and Business Media Deutschland GmbH. DOI: 10.1007/s10311-023-01604-3.

D. Pang et al., “A Mini Review on AI-Driven Thermal Treatment of Solid Waste: Emission Control and Process Optimization,” Jun. 01, 2025, Elsevier B.V. DOI: 10.1016/j.gerr.2025.100132.

J. Zhou, J. Liu, J. Ren, and C. He, “A Comprehensive Study of Machine Learning for Waste-to-Energy Process Modeling and Optimization,” Processes, Vol. 13, No. 9, p. 2691, Aug. 2025, DOI: 10.3390/pr13092691.

P. Lisbona, S. Pascual, and V. Pérez, “Waste to Energy: Trends and Perspectives,” Chemical Engineering Journal Advances, Vol. 14, May 2023, DOI: 10.1016/j.ceja.2023.100494.

M. Sharma Timilsina, Y. Chaudhary, P. Bhattarai, B. Uprety, and D. Khatiwada, “Optimizing Pyrolysis and Co-Pyrolysis of Plastic and Biomass using Artificial Intelligence,” Energy Conversion and Management: X, Vol. 24, Oct. 2024, DOI: 10.1016/j.ecmx.2024.100783.

D. A. Akinpelu, O. A. Adekoya, P. O. Oladoye, C. C. Ogbaga, and J. A. Okolie, “Machine Learning Applications in Biomass Pyrolysis: From Biorefinery to End-of-Life Product Management,” Sep. 01, 2023, Elsevier Ltd. DOI: 10.1016/j.dche.2023.100103.

E. R. Belden et al., “Machine Learning Predictions of Oil Yields Obtained by Plastic Pyrolysis and Application to Thermodynamic Analysis,” ACS Engineering Au, Vol. 3, No. 2, pp. 91–101, Apr. 2023, DOI: 10.1021/acsengineeringau.2c00038.

Y. Ma, P. He, F. Lü, and H. Zhang, “Improving the Method for Calculating Carbon Emissions from Waste Incineration: Confirmed with Carbon-14 Testing of Flue Gas,” Carbon Research, vol. 3, No. 1, Nov. 2024, DOI: 10.1007/s44246-024-00162-9.

K. Mizohata, K. Helariutta, J. Kauppinen, M. Lappalainen, T. Pellikka, and M. Oinonen, “Accelerator Mass Spectrometry for Tracing Fossil Carbon Content of Combusted Fuel,” Applied Radiation and Isotopes, Vol. 225, Nov. 2025, DOI: 10.1016/j.apradiso.2025.111996.

Y. Kwon, K. Choi, and Y. C. Jang, “Greenhouse Gas Emissions from Incineration of Municipal Solid Waste in Seoul, South Korea,” Energies (Basel), Vol. 16, No. 12, Jun. 2023, DOI: 10.3390/en16124791.

H. Jerbi, V. G. A. Gnana Vincy, S. Ben Aoun, R. Abbassi, and M. Kchaou, “Optimizing Waste Management in Smart Cities: An IoT-based Approach using Dynamic Bald Eagle Search Optimization Algorithm (DBESO) and Machine Learning,” Journal of Urban Management, 2025, DOI: 10.1016/j.jum.2025.05.015.

H. Fakhouri, A. Hudaib, F. Hamad, S. Fakhouri, N. Halalsheh, and M. S. Alkhalaileh, “Vehicle Route Optimizer for Waste Collection and Routing Optimization Problem,” Intelligent Systems with Applications, Vol. 27, Sep. 2025, DOI: 10.1016/j.iswa.2025.200521.

J. Cui, Y. Yan, L. Jiang, L. Zhang, and W. Xu, “Research on Optimization of Waste Sorting and Transportation Network in Smart Cities based on Garbage Volume Prediction,” Discover Computing, Vol. 28, No. 1, Dec. 2025, DOI: 10.1007/s10791-025-09537-x.

X. Sun et al., “Revealing Influencing Factors on Global Waste Distribution Via Deep-Learning based Dumpsite Detection from Satellite Imagery,” Nat Commun, Vol. 14, No. 1, Dec. 2023, DOI: 10.1038/s41467-023-37136-1.

A. D. Sakti et al., “Identification of Illegally Dumped Plastic Waste in a Highly Polluted River in Indonesia using Sentinel-2 Satellite Imagery,” SCI Rep, Vol. 13, No. 1, Dec. 2023, DOI: 10.1038/s41598-023-32087-5.

R. N. Torres and P. Fraternali, “AerialWaste Dataset for Landfill Discovery in Aerial and Satellite Images,” SCI Data, Vol. 10, No. 1, Dec. 2023, DOI: 10.1038/s41597-023-01976-9.

J.-S. Lee and D.-C. Shin, “Prediction of Waste Generation using Machine Learning: A Regional Study in Korea,” Urban Science, Vol. 9, No. 8, p. 297, Jul. 2025, DOI: 10.3390/urbansci9080297.

S. Alam, M. Rokonuzzaman, K. S. Rahman, and W. S. Tan, “A Comparative Analysis of Forecasting Algorithms for Predicting Municipal Solid Waste Generation in Chittagong City,” International Journal of Environmental Science and Technology, 2025, DOI: 10.1007/s13762-025-06488-0.

R. Awad, C. Budayan, and A. P. Gurgun, “Construction and Demolition Waste Generation Prediction by using Artificial Neural Networks and Metaheuristic Algorithms,” Buildings, Vol. 14, No. 11, Nov. 2024, DOI: 10.3390/buildings14113695.

T. Cheng, D. Kojima, H. Hu, H. Onoda, and A. H. Pandyaswargo, “Optimizing Waste Sorting for Sustainability: An AI-Powered Robotic Solution for Beverage Container Recycling,” Sustainability (Switzerland), Vol. 16, No. 23, Dec. 2024, DOI: 10.3390/su162310155.

Y. Goorun, S. Marais, and F. Nicolls, “Robotic Waste Sorting using Deep Learning,” MATEC Web of Conferences, Vol. 406, p. 10003, 2024, Doi: 10.1051/matecconf/202440610003.

A. M. Vukicevic et al., “Versatile Waste Sorting in Small Batch and Flexible Manufacturing Industries using Deep Learning Techniques,” SCI Rep, Vol. 15, No. 1, Dec. 2025, DOI: 10.1038/s41598-025-87226-x.

H. T. N. Le and H. Q. T. Ngo, “Application of the Vision-based Deep Learning Technique for Waste Classification using the Robotic Manipulation System,” International Journal of Cognitive Computing in Engineering, Vol. 6, pp. 391–400, Dec. 2025, DOI: 10.1016/j.ijcce.2025.02.005.

A. Pollak, A. Gupta, and D. Gohlich, “Optimized Operation Management with Predicted Filling Levels of the Litter Bins for a Fleet of Autonomous Urban Service Robots,” IEEE Access, Vol. 12, pp. 7689–7703, 2024, DOI: 10.1109/ACCESS.2024.3352436.

M. P. Arthur, S. Shoba, and A. Pandey, “A Survey of Smart Dustbin Systems using the IoT and Deep Learning,” Artif Intell Rev, Vol. 57, No. 3, Mar. 2024, DOI: 10.1007/s10462-023-10646-6.

H. Husnawati, A. B. J. Malyan, R. R. Putra, and S. Agustri, “Smart Trash Box Technology of Computer Vision to Support Ecogreen Campus,” 2024, pp. 220–229. DOI: 10.2991/978-94-6463-386-3_25.

A. Ishaq, S. J. Mohammad, A.-A. D. Bello, S. A. Wada, A. Adebayo, and Z. T. Jagun, “Smart Waste Bin Monitoring using IoT for Sustainable Biomedical Waste Management,” Environmental Science and Pollution Research, Oct. 2023, DOI: 10.1007/s11356-023-30240-1.

A. Pandey, R. Raja, M. Gupta, F. A. Alenizi, P. Suanpang, and A. Nanthaamornphong, “Deep Learning Enabled Garbage Classification and Detection by Visual Context for Aerial Images,” Applied Computational Intelligence and Soft Computing, Vol. 2025, No. 1, 2025, DOI: 10.1155/acis/9106130.

M. Chhabra, B. Sharan, M. Elbarachi, and M. Kumar, “Intelligent Waste Classification Approach based on Improved Multi-Layered Convolutional Neural Network,” Multimed Tools Appl, Vol. 83, No. 36, pp. 84095–84120, Nov. 2024, DOI: 10.1007/s11042-024-18939-w.

M. M. Abo-Zahhad and M. Abo-Zahhad, “Real Time Intelligent Garbage Monitoring and Efficient Collection using Yolov8 and Yolov5 Deep Learning Models for Environmental Sustainability,” SCI Rep, Vol. 15, No. 1, Dec. 2025, DOI: 10.1038/s41598-025-99885-x.

D. Zheng, R. Wang, Y. Duan, P. C. I. Pang, and T. Tan, “Focus-RCNet: A Lightweight Recyclable Waste Classification Algorithm based on Focus and Knowledge Distillation,” Vis Comput Ind Biomed Art, Vol. 6, No. 1, Dec. 2023, DOI: 10.1186/s42492-023-00146-3.

Z. Lun, Y. Pan, S. Wang, Z. Abbas, M. S. Islam, and S. Yin, “Skip-YOLO: Domestic Garbage Detection using Deep Learning Method in Complex Multi-Scenes,” International Journal of Computational Intelligence Systems, Vol. 16, No. 1, Dec. 2023, DOI: 10.1007/s44196-023-00314-6.

M. H. Moktar, H. Mohamed, S. S. H. Hajjaj, and M. Z. Baharuddin, “Medical Waste Sorting Machine Development with IoT and YOLO Model Utilization,” Journal of Engineering and Applied Science, Vol. 72, No. 1, Dec. 2025, DOI: 10.1186/s44147-025-00661-5.

A. Ettalibi, A. Elouadi, and A. Mansour, “AI and Computer Vision-based Real-Time Quality Control: A Review of Industrial Applications,” in Procedia Computer Science, Elsevier B.V., 2024, pp. 212–220. DOI: 10.1016/j.procs.2023.12.195.

Y. Zhou et al., “Optimization of Automated Garbage Recognition Model based on ResNet-50 and Weakly Supervised CNN for Sustainable Urban Development,” Alexandria Engineering Journal, Vol. 108, pp. 415–427, Dec. 2024, DOI: 10.1016/j.aej.2024.07.066.

F. Chollet, “Xception: Deep Learning with Depthwise Separable Convolutions.”

Z. Xu and N. Hajari, “Garbage Classification from Visual Footprints: using Transfer Learning Strategy,” in Proceedings of the International Joint Conference on Computer Vision, Imaging and Computer Graphics Theory and Applications, Science and Technology Publications, Lda, 2025, pp. 403–409. DOI: 10.5220/0013151200003912.

S. Sathyanarayanan, “Confusion Matrix-based Performance Evaluation Metrics,” African Journal of Biomedical Research, pp. 4023–4031, Nov. 2024, DOI: 10.53555/ajbr.v27i4s.4345.




DOI: https://doi.org/10.32520/stmsi.v15i4.5709

Article Metrics

Abstract view : 1 times
PDF - 0 times

Refbacks

  • There are currently no refbacks.


Creative Commons License
This work is licensed under a Creative Commons Attribution-ShareAlike 4.0 International License.