Benchmarking a tensor network-based machine learning algorithm for supervised image classification

  • Mykhal Gideoni Lim Mangada National Institute of Physics, University of the Philippines Diliman
  • Francis N. C. Paraan National Institute of Physics, University of the Philippines Diliman

Abstract

Real-world data is often scarce and the careful setting of hyperparameters to efficiently train machine learning algorithms is necessary. In this study the role of bond dimension on the performance of an existing tensor network-based machine learning program was benchmarked on the MNIST and fashion-MNIST data sets. In particular, the effects of bond dimension on the accuracy and execution times of the algorithm were investigated. Results show that a bond dimension of less than 100 is sufficient to reach an acceptable classification accuracy, and increasing the bond dimension further extends the execution time and consumes prohibitive amounts of memory without providing a reasonable improvement in accuracy.

Published
2018-05-24
How to Cite
[1]
M. G. Mangada and F. N. Paraan. Benchmarking a tensor network-based machine learning algorithm for supervised image classification, Proceedings of the Samahang Pisika ng Pilipinas 36, SPP-2018-PB-23 (2018). URL: https://paperview.spp-online.org/proceedings/article/view/SPP-2018-PB-23.
Section
Poster Session B (Complex Systems, Simulations, and Theoretical Physics)