Introduction to Quantum-Train Toolkit
Quantum-Train is an innovative approach that leverages quantum machine learning (QML) to train classical neural network (NN) models, offering significant parameter reduction on a polylogarithmic scale, eliminating data encoding issues, and enabling inference on purely classical computers. We introduce the Quantum-Train Toolkit, which consolidates the source code from various studies on the Quantum-Train framework. This includes image classification on MNIST, FashionMNIST, and CIFAR-10 datasets using convolutional neural networks, reinforcement learning with policy networks in Cartpole-v1 and MiniGrid environments, and Long Short-Term Memory models for flood prediction. The Quantum-Train Toolkit simplifies the execution of PyTorch multilayer perceptron models using Quantum-Train and provides guidance for adapting more complex architectures into the Quantum-Train format.
Authors
- Min-Hsiu Hsieh (ORCID: https://orcid.org/0000-0002-3396-8427)
- Wei‐Jia Huang (ORCID: https://orcid.org/0000-0002-8695-5689)
- Chen-Yu Liu (ORCID: https://orcid.org/0000-0002-5437-5188)
- Chu-Hsuan Abraham Lin
Institutions
- Foxconn (Taiwan) (TW)
- Imperial College London (GB)
Publication Details
- Published
- 2024-09-15
- DOI
- https://doi.org/10.1109/qce60285.2024.10353
- Citations
- 5
- Primary Topic
- Quantum Computing Algorithms and Architecture
- Type
- article
- Field-Weighted Citation Impact
- 1.74