TY - JOUR
T1 - Analysis of Deep Learning Libraries: Keras, PyTorch, and MXnet
AU - Kim, Seongsoo
AU - Wimmer, Hayden
AU - Kim, Jongyeop
PY - 2022/6/30
Y1 - 2022/6/30
N2 - - As many artificial neural libraries are developing the deep learning algorithm and implementing it became accessible to anyone. This study points out the disparity of performance in deep learning models such as convolutional neural networks (CNN) when implemented with different artificial neural libraries. Libraries such as Keras, Pytorch, and MXnet was utilized for each three CNN model then binary image classification was done based on the Dogs vs. Cats dataset from Kaggle. With using 75% of the dataset as the training set and the rest of 25% as a testing set, and as a result, each CNN model gave a different F1 score value and accuracy.
AB - - As many artificial neural libraries are developing the deep learning algorithm and implementing it became accessible to anyone. This study points out the disparity of performance in deep learning models such as convolutional neural networks (CNN) when implemented with different artificial neural libraries. Libraries such as Keras, Pytorch, and MXnet was utilized for each three CNN model then binary image classification was done based on the Dogs vs. Cats dataset from Kaggle. With using 75% of the dataset as the training set and the rest of 25% as a testing set, and as a result, each CNN model gave a different F1 score value and accuracy.
KW - Binary image classification
KW - CNN
KW - Keras
KW - MXnet
KW - Pytorch
KW - artificial neural network
UR - https://digitalcommons.georgiasouthern.edu/information-tech-facpubs/175
UR - https://doi.org/10.1109/SERA54885.2022.9806734
U2 - 10.1109/SERA54885.2022.9806734
DO - 10.1109/SERA54885.2022.9806734
M3 - Article
JO - IEEE/ACIS 20th International Conference on Software Engineering Research, Management and Applications (SERA) Proceedings
JF - IEEE/ACIS 20th International Conference on Software Engineering Research, Management and Applications (SERA) Proceedings
ER -