用于机器学习的 TensorNetwork
机器学习
2019-06-17 v1 强关联电子
计算机视觉与模式识别
计算物理
机器学习
摘要
我们展示了利用 TensorNetwork 开源库进行图像分类的张量网络应用。我们详细解释了将图像数据编码为矩阵乘积态形式的方法，并描述了以可并行且适于自动梯度优化方式收缩网络的过程。将该技术应用于 MNIST 与 Fashion-MNIST 数据集，在相同张量网络架构下分别获得 98% 与 88% 的现成准确率。TensorNetwork 库使我们能从 CPU 无缝迁移至 GPU 硬件，且使用 GPU 时计算速度提升超过 10 倍。
引用
@article{arxiv.1906.06329,
title = {TensorNetwork for Machine Learning},
author = {Stavros Efthymiou and Jack Hidary and Stefan Leichenauer},
journal= {arXiv preprint arXiv:1906.06329},
year = {2019}
}
备注
9 pages, 8 figures. All code can be found at https://github.com/google/tensornetwork