Skip to content

Machine Learning project focused on improving robustness of image classification

Notifications You must be signed in to change notification settings

stockeh/augmented-image-classification

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Strategies for Robust Image Classification

Abstract—In this work we evaluate the impact of digitally altered images on the performance of artificial neural networks. We explore factors that negatively affect the ability of an image classification model to produce consistent and accurate results. A model’s ability to classify is negatively influenced by alterations to images as a result of digital abnormalities or changes in the physical environment. The focus of this paper is to discover and replicate scenarios that modify the appearance of an image and evaluate them on state-of-the-art machine learning models. Our contributions present various training techniques that enhance a model’s ability to generalize and improve robustness against these alterations.

Find our paper on arXiv [ https://arxiv.org/abs/2004.03452 ]

srs/

  • location of our neural network classifier code
  • ml utilities for reading datasets, perturbing images, and visualizing results
  • source of scripts to automate the training of various models

notebooks/

  • experimental interactive python notebooks
  • location of notebooks/media with all PDF visuals

TeX/

  • LaTeX and BibTeX information to create our final paper

Contributors

Jason D. Stock - stock@colostate.edu - stockeh
Andy A. Dolan - adolan5@colostate.edu - adolan5
Tom E. Cavey - tomcavey@colostate.edu - tc30

About

Machine Learning project focused on improving robustness of image classification

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published