arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2108.08659cs.LG
收藏

Residual Tensor Train:一种学习多重多线性相关的量子启发方法

Residual Tensor Train: A Quantum-inspired Approach for Learning Multiple Multilinear Correlations

Zhejiang University(浙江大学);University of New South Wales(新南威尔士大学)

Yiwei Chen, Yu Pan, Daoyi Dong

摘要:

量子多体系统的态被定义在高维希尔伯特空间中,子系统之间丰富而复杂的相互作用可以被建模。在机器学习中,复杂的多重多线性相关性也可能存在于输入特征内。在本文中,我们提出了一个量子启发的多线性模型,名为剩余张量训练(ResTT),捕捉多个多线性相关的功能,从低到高的订单,在一个单一的模型。ResTT能够在高维空间中构建鲁棒的决策边界,用于解决拟合和分类任务。特别地,我们证明了全连通层和Volterra级数可以作为ResTT的特例。此外,我们推导出了基于平均场分析的稳定ResTT训练的权重初始化规则。我们证明了这样的规则比TT更宽松,这意味着ResTT可以很容易地解决现有TT模型中存在的消失和爆炸梯度问题。数值实验表明,ResTT在MNIST和Fashion-MNIST数据集上的性能优于最先进的张量网络和基准深度学习模型。此外,ResTT实现了更好的性能比其他统计方法在两个实际的例子有限的数据,已知有复杂的功能相互作用。

英文摘要:

States of quantum many-body systems are defined in a high-dimensional Hilbert space, where rich and complex interactions among subsystems can be modelled. In machine learning, complex multiple multilinear correlations may also exist within input features. In this paper, we present a quantum-inspired multilinear model, named Residual Tensor Train (ResTT), to capture the multiple multilinear correlations of features, from low to high orders, within a single model. ResTT is able to build a robust decision boundary in a high-dimensional space for solving fitting and classification tasks. In particular, we prove that the fully-connected layer and the Volterra series can be taken as special cases of ResTT. Furthermore, we derive the rule for weight initialization that stabilizes the training of ResTT based on a mean-field analysis. We prove that such a rule is much more relaxed than that of TT, which means ResTT can easily address the vanishing and exploding gradient problem that exists in the existing TT models. Numerical experiments demonstrate that ResTT outperforms the state-of-the-art tensor network and benchmark deep learning models on MNIST and Fashion-MNIST datasets. Moreover, ResTT achieves better performance than other statistical methods on two practical examples with limited data which are known to have complex feature interactions.

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

↑