Skip to main content
Springer Nature Link
Log in

Comparison of Deep Learning, Data Augmentation and Bag of-Visual-Words for Classification of Imbalanced Image Datasets

  • Conference paper
  • First Online:
Recent Trends in Image Processing and Pattern Recognition (RTIP2R 2018)

Part of the book series: Communications in Computer and Information Science ((CCIS,volume 1035))

  • 809 Accesses

Abstract

Image classification is a supervised machine learning task to classify images into different categories. As most real-world datasets are imbalanced in nature, instances are not equally distributed in all classes which often results in biased classification. So considering this objective, we are dealing with an imbalanced dataset. We have created a small size of an imbalanced dataset consisting of Ant and Plane Images from Caltech-101 dataset. This paper illustrates the comparison between Deep learning, Data Augmentation and the Bag-of-Visual Words (BOVW) for classification of imbalanced image datasets. According to the experimental results, it was found that deep learning results in higher accuracy in comparison to bag-of-visual-words (BOVW) and data augmentation.

This is a preview of subscription content, log in via an institution to check access.

Access this chapter

Log in via an institution

Subscribe and save

Springer+ Basic
$34.99 /Month
  • Get 10 units per month
  • Download Article/Chapter or eBook
  • 1 Unit = 1 Article or 1 Chapter
  • Cancel anytime
Subscribe now

Buy Now

Chapter
USD 29.95
Price excludes VAT (USA)
  • Available as PDF
  • Read on any device
  • Instant download
  • Own it forever

Tax calculation will be finalised at checkout

Purchases are for personal use only

Institutional subscriptions

Similar content being viewed by others

References

  1. Frome, A., Singer, Y., Malik, J.: Image retrieval and classification using local distance functions. NIPS (2006)

    Google Scholar 

  2. Villar, P., Fernandez, A., Carrasco, R.A., Herrera, F.: Feature selection and granularity learning in genetic fuzzy rule-based classification systems for highly imbalanced data-sets. Int. J. Uncertainty Fuzziness Knowl. Based Syst. 20(3), 369–397 (2012)

    Article  Google Scholar 

  3. https://beamandrew.github.io/deeplearning/2017/06/04/deep_learning_works.html

  4. Zhao, W.: Research on the deep learning of the small sample data based on transfer learning. AIP Conf. Proc. 1864, 020018 (2017)

    Article  Google Scholar 

  5. Krizhevsky, A., Sutskever, I., Hinton, G.E.: Imagenet classification with deep convolutional neural networks. In: Advances in Neural Information Processing Systems, pp. 1097–1105 (2012)

    Google Scholar 

  6. Jia, Y., et al.: Caffe: convolutional architecture for fast feature embedding. In: Proceedings of the 22nd ACM International Conference on Multimedia, pp. 675–678. ACM (2014)

    Google Scholar 

  7. Bjerrum, E.J.: SMILES enumeration as data augmentation for neural network modeling of molecules. ArXiv e-prints, March (2017)

    Google Scholar 

  8. Yang, J., Jiang, Y.-G., Hauptmann, A.G., Ngo, C.-W.: Evaluating bag-of-visual-words representations in scene classification. In: Proceedings of the International Workshop on Multimedia Information Retrieval, pp. 197–206. ACM (2007)

    Google Scholar 

  9. Csurka, G., Dance, C., Fan, L., Willamowski, J., Bray, C.: Visual categorization with bags of keypoints. In: Workshop on Statistical Learning in Computer Vision, ECCV, Prague (2004)

    Google Scholar 

  10. Lowe, D.G.: Distinctive image features from scale-invariant keypoints. Int. J. Comput. Vision 60(2), 91–110 (2004)

    Article  Google Scholar 

  11. Platt, J.: Sequential minimal optimization: a fast algorith for training support vector machines. Microsoft Research Tech Rep MSR-TR-98-14 (1998)

    Google Scholar 

  12. Bunkhumpornpat, C., Sinapiromsaran, K., Lursinsap, C.: DBSMOTE: density-based synthetic minority over-sampling technique. Appl. Intell. 36(3), 664–684 (2012)

    Article  Google Scholar 

  13. LeCun, Y., Kavukcuoglu, K., Farabet, C.: Convolutional networks and applications in vision. In: Proceedings of 2010 IEEE International Symposium on Circuits and Systems (ISCAS), pp. 253–256. IEEE (2010)

    Google Scholar 

  14. Xie, L., Tian, Q., Wang, M., Zhang, B.: Spatial pooling of heterogeneous features for image classification. IEEE Trans. Image Process. 23(5), 1994–2008 (2014)

    Article  MathSciNet  Google Scholar 

  15. http://www.vision.caltech.edu/Image_Datasets/Caltech101/

  16. Szegedy, C., et al.: Going deeper with convolutions. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 1–12 (2015)

    Google Scholar 

  17. Lin, D., Shen, X., Lu, C., Jia, J.: Deep LAC: deep localization, alignment and classification for fine-grained recognition. In: The IEEE Conference on Computer Vision and Pattern Recognition (CVPR) (2015)

    Google Scholar 

  18. Reed, S., Lee, H., Anguelov, D., Szegedy, C., Erhan, D., Rabinovich, A.: Training deep neural networks on noisy labels with bootstrapping. arXiv preprint arXiv:1412.6596 (2014)

  19. Zhou, B., Lapedriza, A., Xiao, J., Torralba, A., Oliva, A.: Learning deep features for scene recognition using places database. In: Advances in Neural Information Processing Systems, pp. 487–495 (2014)

    Google Scholar 

  20. Yosinski, J., Clune, J., Bengio, Y., Lipson, H.: How transferable are features in deep neural networks? In: Advances in Neural Information Processing Systems, pp. 3320–3328 (2014)

    Google Scholar 

  21. Vajda, S., Santosh, K.C.: A fast k-nearest neighbor classifier using unsupervised clustering. In: Santosh, K.C., Hangarge, M., Bevilacqua, V., Negi, A. (eds.) RTIP2R 2016. CCIS, vol. 709, pp. 185–193. Springer, Singapore (2017). https://doi.org/10.1007/978-981-10-4859-3_17

    Chapter  Google Scholar 

  22. Ukil, S., Ghosh, S., Md-Obaidullah, S.K., Santosh, K.C., Roy, K., Das, N.: Deep learning for word-level handwritten Indic script identification. arXiv:1801.01627 (2018)

  23. Buda, M., Maki, A., Mazurowski, M.A.: A systematic study of the class imbalance problem in convolutional neural networks. arXivpreprint arXiv:1710.05381 (2017)

  24. Dong, Q., Gong, S., Zhu, X.: Imbalanced deep learning by minority class incremental rectification. IEEE Trans. Pattern Anal. Mach. Intell. 1–1 (2018)

    Google Scholar 

  25. Suh, H.K., Hofstee, J.W., IJsselmuiden, J.M.M., van Henten, E.J.: Sugar beet volunteer potato classification using Bag-of-Visual-words model, scale-invariant feature transform, or speeded up robust feature descriptors and crop row information. Bio Systems Engineering. Elsevier (2017)

    Google Scholar 

  26. Zhu, X., Liu, Y., Li, J., Wan, T., Qin, Z.: Emotion classification with data augmentation using generative adversarial networks. In: Phung, D., Tseng, V.S., Webb, G.I., Ho, B., Ganji, M., Rashidi, L. (eds.) PAKDD 2018. LNCS (LNAI), vol. 10939, pp. 349–360. Springer, Cham (2018). https://doi.org/10.1007/978-3-319-93040-4_28

    Chapter  Google Scholar 

  27. Harriet, S., Jonathan, V.: Handling unbalanced data in deep image segmentation (2017)

    Google Scholar 

  28. Wang, Q., Luo, Z.H., Huang, J.C., Feng, Y.H., Liu, Z.: A novel ensemble method for imbalanced data learning: bagging of extrapolation-SMOTE SVM. Hindawi Publishing Corporation Computational Intelligence and Neuroscience (2017)

    Google Scholar 

  29. Susan, S., Jain, A., Sharma, A., Verma, S., Jain, S.: Fuzzy match index for scale-invariant feature transform (SIFT) features with application to face recognition with weak supervision. IET Image Proc. 9(11), 951–958 (2015)

    Article  Google Scholar 

  30. Suh, H.K., Hofstee, J.W., Jsselmuiden, J.I., van Henten, E.J.: Sugar beet volunteer potato classification using Bag-of-Visual-words model, scale-invariant feature transform, or speeded up robust feature descriptors and crop row information. Bio Syst. Eng. 166, 210–226 (2018)

    Article  Google Scholar 

Download references

Author information

Authors and Affiliations

  1. Delhi Technological University, Delhi, India

    Manisha Saini & Seba Susan

Authors
  1. Manisha Saini
    View author publications

    You can also search for this author in PubMed Google Scholar

  2. Seba Susan
    View author publications

    You can also search for this author in PubMed Google Scholar

Corresponding author

Correspondence to Manisha Saini .

Editor information

Editors and Affiliations

  1. Department of Computer Science, University of South Dakota, Vermillion, SD, USA

    K. C. Santosh

  2. Solapur University, Solapur, India

    Ravindra S. Hegadi

Rights and permissions

Reprints and permissions

Copyright information

© 2019 Springer Nature Singapore Pte Ltd.

About this paper

Check for updates. Verify currency and authenticity via CrossMark

Cite this paper

Saini, M., Susan, S. (2019). Comparison of Deep Learning, Data Augmentation and Bag of-Visual-Words for Classification of Imbalanced Image Datasets. In: Santosh, K., Hegadi, R. (eds) Recent Trends in Image Processing and Pattern Recognition. RTIP2R 2018. Communications in Computer and Information Science, vol 1035. Springer, Singapore. https://doi.org/10.1007/978-981-13-9181-1_49

Download citation

  • DOI: https://doi.org/10.1007/978-981-13-9181-1_49

  • Published:

  • Publisher Name: Springer, Singapore

  • Print ISBN: 978-981-13-9180-4

  • Online ISBN: 978-981-13-9181-1

  • eBook Packages: Computer ScienceComputer Science (R0)

Publish with us

Policies and ethics