让递归神经网络具备时间流动对称性,提升序列建模性能。
Flow Equivariant Recurrent Neural Networks
- 引入流等变性理论,使RNN隐藏状态随时间平滑变换。
- 在预测和分类任务中训练更快,且能泛化到更长序列和不同速度。
- 适合研究时空对称性建模或需要高效学习的序列任务。
感官接收的数据是连续流,随时间平滑演变,这种变化可视为环境中的连续对称性,定义了时间上刺激间的等价关系。机器学习中,尊重数据对称性的神经网络称为等变网络,具有更好的泛化能力和样本效率。然而,现有研究仅限于静态变换与前馈网络,难以应用于循环神经网络(RNN)这类序列模型及其对应的时间参数化变换。本文将等变网络理论拓展至“流”领域——即刻画自然时间变换的一参数李子群,如视觉运动。我们首先证明标准RNN一般不具备流等变性:对于运动刺激,其隐藏状态未能以几何结构化方式变换。随后提出实现流等变性的方法,并验证该模型在下一时刻预测与序列分类任务中显著优于非等变基线,在训练速度、长度外推和速度外推方面均表现更佳。本工作为构建尊重世界时间对称性的序列模型迈出第一步。
原文摘要 · Abstract (English)
Data arrives at our senses as a continuous stream, smoothly transforming from one instant to the next. These smooth transformations can be viewed as continuous symmetries of the environment that we inhabit, defining equivalence relations between stimuli over time. In machine learning, neural network architectures that respect symmetries of their data are called equivariant and have provable benefits in terms of generalization ability and sample efficiency. To date, however, equivariance has been considered only for static transformations and feed-forward networks, limiting its applicability to sequence models, such as recurrent neural networks (RNNs), and corresponding time-parameterized sequence transformations. In this work, we extend equivariant network theory to this regime of 'flows' -- one-parameter Lie subgroups capturing natural transformations over time, such as visual motion. We begin by showing that standard RNNs are generally not flow equivariant: their hidden states fail to transform in a geometrically structured manner for moving stimuli. We then show how flow equivariance can be introduced, and demonstrate that these models significantly outperform their non-equivariant counterparts in terms of training speed, length generalization, and velocity generalization, on both next step prediction and sequence classification. We present this work as a first step towards building sequence models that respect the time-parameterized symmetries which govern the world around us.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。