Performance analysis of the TensorFlow library with different optimisation algorithms

Main Article Content

DOI

Maciej Wadas

maciej.wadas@pollub.edu.pl

Jakub Smołka

jakub.smolka@pollub.pl

Abstract

This paper presents the results of performance analysis of the Tensorflow library used in machine learning and deep neural networks. The analysis focuses on comparing the parameters obtained when training the neural network model for optimization algorithms: Adam, Nadam, AdaMax, AdaDelta, AdaGrad. Special attention has been paid to the differences between the training efficiency on tasks using microprocessor and graphics card. For the study, neural network models were created in order to recognise Polish handwritten characters. The results obtained showed that the most efficient algorithm is AdaMax, while the computer component used during the research only affects the training time of the neural network model used.

Keywords:

machine learning; neural networks

References

Article Details

Wadas, M., & Smołka, J. (2021). Performance analysis of the TensorFlow library with different optimisation algorithms. Journal of Computer Sciences Institute, 21, 330–335. https://doi.org/10.35784/jcsi.2738