arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2001.08286stat.MLcs.LGquant-ph
收藏

A Multi-Scale Tensor Network Architecture for Classification and Regression

University of Central Florida(中佛罗里达大学);Flatiron Institute(弗莱德森研究所)

Justin Reyes, Miles Stoudenmire

英文摘要:

We present an algorithm for supervised learning using tensor networks, employing a step of preprocessing the data by coarse-graining through a sequence of wavelet transformations. We represent these transformations as a set of tensor network layers identical to those in a multi-scale entanglement renormalization ansatz (MERA) tensor network, and perform supervised learning and regression tasks through a model based on a matrix product state (MPS) tensor network acting on the coarse-grained data. Because the entire model consists of tensor contractions (apart from the initial non-linear feature map), we can adaptively fine-grain the optimized MPS model backwards through the layers with essentially no loss in performance. The MPS itself is trained using an adaptive algorithm based on the density matrix renormalization group (DMRG) algorithm. We test our methods by performing a classification task on audio data and a regression task on temperature time-series data, studying the dependence of training accuracy on the number of coarse-graining layers and showing how fine-graining through the network may be used to initialize models with access to finer-scale features.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑