JOIV : International Journal on Informatics Visualization
Vol 8, No 2 (2024)

A Hybrid Model for Dry Waste Classification using Transfer Learning and Dimensionality Reduction

Santoso, Hadi (Unknown)
Hanif, Ilham (Unknown)
Magdalena, Hilyah (Unknown)
Afiyati, Afiyati (Unknown)



Article Info

Publish Date
31 May 2024

Abstract

The categorization of waste plays a crucial part in efficient waste management, facilitating the recognition and segregation of various waste types to ensure appropriate disposal, recycling, or repurposing. With the growing concern for environmental sustainability, accurate waste classification systems are in high demand. Traditional waste classification methods often rely on manual sorting, which is time-consuming, labor-intensive, and prone to errors. Hence, there is a need for automated and efficient waste classification systems that can accurately categorize waste materials. In this research, we introduce an innovative waste classification system that merges feature extraction from a pretrained EfficientNet model with Principal Component Analysis (PCA) to reduce dimensionality. The methodology involves two main stages: (1) transfer learning using the EfficientNet-CNN architecture for feature extraction, and (2) dimensionality reduction using PCA to reduce the feature vector dimensionality. The features extracted from both the average pooling and convolutional layers are combined by concatenation, and subsequently, classification is performed using a fully connected layer. Extensive experiments were conducted on a waste dataset, and the proposed system achieved a remarkable accuracy of 99.07%. This outperformed the state-of-the-art waste classification systems, demonstrating the effectiveness of the combined approach. Further research can explore the application of the proposed waste classification system on larger and diverse datasets, optimize the dimensionality reduction technique, consider real-time implementation, investigate advanced techniques like ensemble learning and deep learning, and assess its effectiveness in industrial waste management systems.

Copyrights © 2024






Journal Info

Abbrev

joiv

Publisher

Subject

Computer Science & IT

Description

JOIV : International Journal on Informatics Visualization is an international peer-reviewed journal dedicated to interchange for the results of high quality research in all aspect of Computer Science, Computer Engineering, Information Technology and Visualization. The journal publishes state-of-art ...