Have a personal or library account? Click to login
Improvement of Classification Results of Convolutional Neural Networks Using Various Gan-Based Augmentation Techniques Cover

Improvement of Classification Results of Convolutional Neural Networks Using Various Gan-Based Augmentation Techniques

Open Access
|Nov 2024

Abstract

Generative Adversarial Network (GAN) is an exciting innovation in machine learning within the neural network field. These models are able to generate a realistic image, video or even voice output. One of the useful applications is its possibility to enrich data sets for better learning of neural network models. In the presented work, we focus on image augmentation with the use of several variations of GAN to improve the classification of convolutional neural network. Accordingly, to prove the advantage of GAN-based image augmentation in comparison with methods of classical augmentation, we used specifically three different degrees of image rotation and compared classification results of convolutional neural network that use images from these augmentation methods. Mentioned methods of image augmentation are applied to five datasets belonging to three different domains, specifically medical, astronomical and geological domain. The architecture and settings of the convolutional neural network are the same for all datasets. To evaluate classification results, we used confusion matrix, accuracy, precision, recall and F1-score.

DOI: https://doi.org/10.2478/aei-2024-0013 | Journal eISSN: 1338-3957 | Journal ISSN: 1335-8243
Language: English
Page range: 11 - 18
Submitted on: Jun 30, 2024
Accepted on: Aug 13, 2024
Published on: Nov 17, 2024
Published by: Technical University of Košice
In partnership with: Paradigm Publishing Services
Publication frequency: 4 issues per year

© 2024 Lenka Kališková, Peter Butka, published by Technical University of Košice
This work is licensed under the Creative Commons Attribution 4.0 License.