Hallam, A;
Grant, E;
Stojevic, V;
Severini, S;
Green, AG;
(2018)
Compact neural networks based on the multiscale entanglement renormalization Ansatz.
In:
British Machine Vision Conference 2018, BMVC 2018.
BMVC: Newcastle-upon-Tyne, UK.
|
Text
0784.pdf - Published version Download (230kB) | Preview |
Abstract
This paper demonstrates a method for tensorizing neural networks based upon an efficient way of approximating scale invariant quantum states, the Multi-scale Entanglement Renormalization Ansatz (MERA). We employ MERA as a replacement for the fully connected layers in a convolutional neural network and test this implementation on the CIFAR-10 and CIFAR-100 datasets. The proposed method outperforms factorization using tensor trains, providing greater compression for the same level of accuracy and greater accuracy for the same level of compression. We demonstrate MERA layers with 14000 times fewer parameters and a reduction in accuracy of less than 1% compared to the equivalent fully connected layers, scaling like O(N).
Archive Staff Only
![]() |
View Item |