Towards artificially intelligent recycling Improving image processing
for waste classification
- URL: http://arxiv.org/abs/2108.06274v1
- Date: Mon, 9 Aug 2021 21:41:48 GMT
- Title: Towards artificially intelligent recycling Improving image processing
for waste classification
- Authors: Youpeng Yu and Ryan Grammenos
- Abstract summary: IBM's Wastenet project aims to improve recycling by using artificial intelligence for waste classification.
This paper builds on this project through the use of transfer learning and data augmentation techniques.
Results show that these augmentation techniques further improve the test accuracy of the final model to 95.40%.
- Score: 0.0
- License: http://arxiv.org/licenses/nonexclusive-distrib/1.0/
- Abstract: The ever-increasing amount of global refuse is overwhelming the waste and
recycling management industries. The need for smart systems for environmental
monitoring and the enhancement of recycling processes is thus greater than
ever. Amongst these efforts lies IBM's Wastenet project which aims to improve
recycling by using artificial intelligence for waste classification. The work
reported in this paper builds on this project through the use of transfer
learning and data augmentation techniques to ameliorate classification
accuracy. Starting with a convolutional neural network (CNN), a systematic
approach is followed for selecting appropriate splitting ratios and for tuning
multiple training parameters including learning rate schedulers, layers
freezing, batch sizes and loss functions, in the context of the given scenario
which requires classification of waste into different recycling types. Results
are compared and contrasted using 10-fold cross validation and demonstrate that
the model developed achieves a 91.21% test accuracy. Subsequently, a range of
data augmentation techniques are then incorporated into this work including
flipping, rotation, shearing, zooming, and brightness control. Results show
that these augmentation techniques further improve the test accuracy of the
final model to 95.40%. Unlike other work reported in the field, this paper
provides full details regarding the training of the model. Furthermore, the
code for this work has been made open-source and we have demonstrated that the
model can perform successful real-time classification of recycling waste items
using a standard computer webcam.
Related papers
- Advancing Recycling Efficiency: A Comparative Analysis of Deep Learning Models in Waste Classification [0.0]
The research tackles the pressing issue of waste classification for recycling by analyzing various deep learning models.
The results indicate the method significantly boosts accuracy in complex waste categories.
The research paves the way for future advancements in multi-category waste recycling.
arXiv Detail & Related papers (2024-11-05T03:44:54Z) - WasteGAN: Data Augmentation for Robotic Waste Sorting through Generative Adversarial Networks [7.775894876221921]
We introduce a data augmentation method based on a novel GAN architecture called wasteGAN.
The proposed method allows to increase the performance of semantic segmentation models, starting from a very limited bunch of labeled examples.
We then leverage the higher-quality segmentation masks predicted from models trained on the wasteGAN synthetic data to compute semantic-aware grasp poses.
arXiv Detail & Related papers (2024-09-25T15:04:21Z) - Image Recognition for Garbage Classification Based on Pixel Distribution Learning [0.0]
This study proposes a novel approach inspired by pixel distribution learning techniques to enhance automated garbage classification.
The method aims to address limitations of conventional convolutional neural network (CNN)-based approaches, including computational complexity and vulnerability to image variations.
We will conduct experiments using the Kaggle Garbage Classification dataset, comparing our approach with existing models to demonstrate the strength and efficiency of pixel distribution learning in automated garbage classification technologies.
arXiv Detail & Related papers (2024-09-05T21:22:48Z) - Efficient Transferability Assessment for Selection of Pre-trained Detectors [63.21514888618542]
This paper studies the efficient transferability assessment of pre-trained object detectors.
We build up a detector transferability benchmark which contains a large and diverse zoo of pre-trained detectors.
Experimental results demonstrate that our method outperforms other state-of-the-art approaches in assessing transferability.
arXiv Detail & Related papers (2024-03-14T14:23:23Z) - MWaste: A Deep Learning Approach to Manage Household Waste [0.0]
MWaste is a mobile application that uses computer vision and deep learning techniques to classify waste materials as trash, plastic, paper, metal, glass or cardboard.
Its effectiveness was tested on various neural network architectures and real-world images, achieving an average precision of 92% on the test set.
arXiv Detail & Related papers (2023-04-02T16:56:49Z) - A Memory Transformer Network for Incremental Learning [64.0410375349852]
We study class-incremental learning, a training setup in which new classes of data are observed over time for the model to learn from.
Despite the straightforward problem formulation, the naive application of classification models to class-incremental learning results in the "catastrophic forgetting" of previously seen classes.
One of the most successful existing methods has been the use of a memory of exemplars, which overcomes the issue of catastrophic forgetting by saving a subset of past data into a memory bank and utilizing it to prevent forgetting when training future tasks.
arXiv Detail & Related papers (2022-10-10T08:27:28Z) - Unsupervised Restoration of Weather-affected Images using Deep Gaussian
Process-based CycleGAN [92.15895515035795]
We describe an approach for supervising deep networks that are based on CycleGAN.
We introduce new losses for training CycleGAN that lead to more effective training, resulting in high-quality reconstructions.
We demonstrate that the proposed method can be effectively applied to different restoration tasks like de-raining, de-hazing and de-snowing.
arXiv Detail & Related papers (2022-04-23T01:30:47Z) - Improving the Deployment of Recycling Classification through Efficient
Hyper-Parameter Analysis [0.0]
This paper develops a more efficient variant of WasteNet, a collaborative recycling classification model.
The newly developed model scores a test-set accuracy of 95.8% with a real world accuracy of 95%, a 14% increase over the original.
Our acceleration pipeline boosted model throughput by 750% to 24 inferences per second on the Jetson Nano embedded device.
arXiv Detail & Related papers (2021-10-21T10:42:14Z) - Always Be Dreaming: A New Approach for Data-Free Class-Incremental
Learning [73.24988226158497]
We consider the high-impact problem of Data-Free Class-Incremental Learning (DFCIL)
We propose a novel incremental distillation strategy for DFCIL, contributing a modified cross-entropy training and importance-weighted feature distillation.
Our method results in up to a 25.1% increase in final task accuracy (absolute difference) compared to SOTA DFCIL methods for common class-incremental benchmarks.
arXiv Detail & Related papers (2021-06-17T17:56:08Z) - ZeroWaste Dataset: Towards Automated Waste Recycling [51.053682077915546]
We present the first in-the-wild industrial-grade waste detection and segmentation dataset, ZeroWaste.
This dataset contains over1800fully segmented video frames collected from a real waste sorting plant.
We show that state-of-the-art segmentation methods struggle to correctly detect and classify target objects.
arXiv Detail & Related papers (2021-06-04T22:17:09Z) - Automatic Recall Machines: Internal Replay, Continual Learning and the
Brain [104.38824285741248]
Replay in neural networks involves training on sequential data with memorized samples, which counteracts forgetting of previous behavior caused by non-stationarity.
We present a method where these auxiliary samples are generated on the fly, given only the model that is being trained for the assessed objective.
Instead the implicit memory of learned samples within the assessed model itself is exploited.
arXiv Detail & Related papers (2020-06-22T15:07:06Z)
This list is automatically generated from the titles and abstracts of the papers in this site.
This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.