Skip to content

Tensor network machine learning. Reproduce the experiment from the paper "Supervised Learning with Quantum Inspired Tensor Networks" http://arxiv.org/abs/1605.05775

Notifications You must be signed in to change notification settings

vbertret/PIR-Tensor-Network-MNIST

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

38 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

PIR Tensor Network MNIST

The aim of this study is to reproduce the experience from Miles Stoudenmire and David Schwab described in the publication "Supervised Learning with Tensor Networks" (http://arxiv.org/abs/1605.05775).

The project aims to study the classification of images from the MNIST database which contains handwritten digits using a tensor network. For that, it uses a tensor decomposition training with basic Machine Learning algorithms, such as gradient descent, but also more sophisticated physics-inspired algorithms.

Description of the project

The goal of the project is to optimize a Matrix Product State so that it will be able to classify inputs. For that, there are differents possibilities. Here, the 2 possibilities developped are :

  • a simple algorithm of alternating gradient descent (AGD). At each tensor, the algorithm tries to reduce the loss by computing the gradient of the loss function according to the tensor.
  • the Density Matrix Renormalization Group (DMRG) . The algorithm does some sweeps along the MPS and at each bond, it contracts 2 tensors and tries to reduce the loss by computing the gradient of loss function according to the contraction.

The study is more oriented on the second possibilities DMRG. Indeed, this algorithm has a lot of avantages that you can see in the first report.

If you want to test by yourself, you have to clone the repository. After, you just have to execute the 2 commands to install all the packages needeed for the project and especially the package src :

pip install -r requirements.txt
pip install -e .

Organisation

  • \data-- the data which is processed or external
    • \external MNIST Database without any traitment
    • \processed modified MNIST Database with pictures of 14 pixels by 14 pixels
  • \docs all the documentation used for the project
  • \notebooks different notebooks with example of code to illustrate the projects and to have some example of the syntax in order to use the package
  • \reports the 2 reports made for the projects (the first report is in French)
    • \figures illustrations of reports
  • \src the package developed to train a tensor network. Each method or class is documented
    • \data some method to create random data or to load the MNIST dataset
    • \mapping methods to compute the mapping function
    • \tensor one method to contract a tensor with inputs
    • \tensor_network all the stuff needeed to train a tensor network
      • modelMPS.py the central class of the project. It defines a tensor network with different methods to train it
      • optimizer.py different algorithms to reduce the loss : Adam, Gradient Descent with fixed stepsize and Conjugate Gradient Descent
      • \algo_AGD some methods to compute the gradient and the cost for basic Alternating Gradient Descent
      • \algos_DMRG some methods to compute the gradient and the cost with Density Matrix Renormalization Group
  • requirements.txt all the package you need to install so that you can use the project on your own

About

Tensor network machine learning. Reproduce the experiment from the paper "Supervised Learning with Quantum Inspired Tensor Networks" http://arxiv.org/abs/1605.05775

Topics

Resources

Stars

Watchers

Forks

Packages

No packages published