6 CONCLUSIONS
This research performed a comparative study of
machine learning models for waste classification
using the TrashNet dataset, aiming to determine the
most efficient method of automation in waste
separation. The models being tested included a
personalized convolutional neural network (CNN), a
support vector machine (SVM) with features
extracted from MobileNetV2, and two transfer
learning models: MobileNetV2 and ResNet 50. The
analysis results revealed the following major
findings: Among the models considered, it was
ResNet50 that performed best with 93.45% accuracy,
featuring higher precision, recall, and F1 scores
across all six waste types. Very deep architecture
along with pre-trained weights made ResNet50 learn
features from the problem domain very effectively
and this makes the network highly suitable for waste
classification tasks. It was MobileNetV2 that
performed comparably less with 91.67% accuracy;
however, it proved to be lightweight and
computationally efficient, giving the additional
advantage of faster training and inference, thus ideal
for deployment in resource-constrained environments
like smart bins or IoT devices. SVM with
MobileNetV2 features performed well (88.90%
accuracy), surpassing the CNN trained from scratch.
Thus, it emphasizes the importance of strong feature
extraction to enhance traditional classifiers. The CNN
model reached an acceptable accuracy level of
85.34%, but it could be recognized as weaker than
transfer learning approaches due to limited depth and
lack of pre-trained feature representations.
REFERENCES
A. Mazloumian, M. Rosenthal, and H. Gelke, "Deep
Learning for Classifying Food Waste," arXiv, 2020.
A. Thompson et al., "Image-Based Waste Sorting Using
Convolutional Neural Networks," IEEE Transactions
on Environmental Engineering, 2022.
B. Lee et al., "Integrating SVM with CNN Features for
Enhanced Waste Classification," Machine Vision
Applications, 2020.
C. Zhao et al., "Transfer Learning in Environmental Data:
Applications in Waste Sorting," Sustainable
Computing, 2022.
C. Zhao et al., "Transfer Learning in Environmental Data:
Applications in Waste Sorting," Sustainable
Computing, 2022.
D. Gyawali et al., "Comparative Analysis of Multiple Deep
CNN Models for Waste Classification," arXiv, 2020.
D. K. Sharma, U. Bharti, and T. Pandey, "Deep Learning
Approaches for Automated Waste Classification and
Sorting," International Journal for Research in Applied
Science and Engineering Technology, 2024.
J. Singh et al., "Convolutional Neural Networks for Image-
Based Waste Classification," Journal of AI in
Environmental Engineering, 2022.
J. Shah and S. Kamat, "A Method for Waste Segregation
using Convolutional Neural Networks," arXiv, 2022.
J. Singh et al., "Convolutional Neural Networks for Image-
Based Waste Classification," Journal of AI in
Environmental Engineering, 2022.
K. Srivatsan, S. Dhiman, and A. Jain, "Waste Classification
using Transfer Learning with Convolutional Neural
Networks," IOP Conference Series: Earth and
Environmental Science, 2021.
M. S. Patel et al., "Transfer Learning for Sustainable Waste
Classification: A Case Study Using MobileNetV2,"
Journal of Sustainable AI Research, 2020.
M. S. Nafiz et al., "ConvoWaste: An Automatic Waste
Segregation Machine Using Deep Learning," arXiv,
2023.
N. Yadav et al., "Waste Image Classification Using Deep
Learning: A Comparative Study," Computational
Ecology and Environmental Science, 2023.
N. Yadav et al., "Waste Image Classification Using Deep
Learning: A Comparative Study," Computational
Ecology and Environmental Science, 2023.
P. Li et al., "Enhancing Waste Classification with SVM-
CNN Hybrid Models," International Journal of
Environmental AI Research, 2023.
P. Li et al., "Enhancing Waste Classification with SVM-
CNN Hybrid Models," International Journal of
Environmental AI Research, 2023.
R. Kumar et al., "Artificial Intelligence in Waste
Management: Current Trends and Future Directions,"
Environmental AI Applications, 2021.
R. Kumar et al., "Artificial Intelligence in Waste
Management: Current Trends and Future Directions,"
Environmental AI Applications, 2021.
S. Khana et al., "Automated Waste Classification with Deep
Learning: Enhancing Recycling Efficiency," Journal of
Environmental Science & Technology, 2021.
S. Kunwar, B. R. Owabumoye, and A. S. Alade, "Plastic
Waste Classification Using Deep Learning: Insights
from the WaDaBa Dataset," arXiv, 2024.
World Bank, "What a Waste 2.0: A Global Snapshot of
Solid Waste Management to 2050," World Bank
Group, 2019.
Y. Narayan, "DeepWaste: Applying Deep Learning to
Waste Classification for a Sustainable Planet," arXiv,
2021.
Z. Yuan and J. Liu, "A Hybrid Deep Learning Model for
Trash Classification Based on Deep Transfer
Learning," Journal of Electrical and Computer
Engineering, 2022.
Z. Qiao, "Advancing Recycling Efficiency: A Comparative
Analysis of Deep Learning Models in Waste
Classification," arXiv, 2024.