arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2406.05045cs.LG
收藏

二阶RNN的张量分解视角

A Tensor Decomposition Perspective on Second-order RNNs

Mila(米拉研究所);Université de Montréal(蒙特利尔大学);CIFAR(加拿大高级研究所)

Maude Lizaire, Michael Rizvi-Martel, Marawan Gamal Abdel Hameed, Guillaume Rabusseau

AI总结:

本文从张量分解视角研究二阶RNN,提出基于CP分解的CPRNN模型,分析了秩和隐藏层大小对模型容量的影响,并在Penn Treebank数据集上验证了其在固定参数预算下优于RNN、2RNN和MIRNN。

摘要:

二阶循环神经网络(2RNNs)通过利用二阶交互进行序列建模来扩展RNN。这些模型已被证明比一阶对应模型更具表达能力,并且与形式语言理论中经过深入研究的模型存在联系。然而,其庞大的参数张量使得计算难以处理。为规避这一问题,一种被称为MIRNN的方法在于限制模型所使用的交互类型。另一种方法是利用张量分解来减少参数数量。在这项工作中,我们研究了使用CP分解对2RNN进行参数化所得到的模型,我们称之为CPRNN。直观上,分解的秩应当会降低表达能力。我们分析了秩和隐藏层大小如何影响模型容量,并基于这些参数展示了RNN、2RNN、MIRNN和CPRNN之间的关系。我们通过在Penn Treebank数据集上的实验从经验上支持了这些结果,实验表明,在固定参数预算下,只要正确选择秩和隐藏层大小,CPRNN的性能优于RNN、2RNN和MIRNN。

英文摘要:

Second-order Recurrent Neural Networks (2RNNs) extend RNNs by leveraging second-order interactions for sequence modelling. These models are provably more expressive than their first-order counterparts and have connections to well-studied models from formal language theory. However, their large parameter tensor makes computations intractable. To circumvent this issue, one approach known as MIRNN consists in limiting the type of interactions used by the model. Another is to leverage tensor decomposition to diminish the parameter count. In this work, we study the model resulting from parameterizing 2RNNs using the CP decomposition, which we call CPRNN. Intuitively, the rank of the decomposition should reduce expressivity. We analyze how rank and hidden size affect model capacity and show the relationships between RNNs, 2RNNs, MIRNNs, and CPRNNs based on these parameters. We support these results empirically with experiments on the Penn Treebank dataset which demonstrate that, with a fixed parameter budget, CPRNNs outperforms RNNs, 2RNNs, and MIRNNs with the right choice of rank and hidden size.

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

↑