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

HiPPO:具有最优多项式投影的递归存储器

HiPPO: Recurrent Memory with Optimal Polynomial Projections

Stanford University(斯坦福大学);University at Buffalo, SUNY(纽约州立大学布法罗分校)

Albert Gu, Tri Dao, Stefano Ermon, Atri Rudra, Christopher Re

摘要:

从顺序数据中学习的一个核心问题是随着更多数据的处理,以增量的方式表示累积的历史。我们介绍了一个通用的框架(HiPPO)的连续信号和离散时间序列的投影到多项式基地的在线压缩。给定指定过去每个时间步的重要性的度量,HiPPO为自然在线函数近似问题产生最优解。作为特殊情况下,我们的框架产生了一个简短的推导最近的勒让德记忆单元(LMU)的第一性原则,并概括了无处不在的门控机制的递归神经网络,如GRU。这个正式的框架产生了一个新的内存更新机制(HiPPO-LegS),通过时间来记住所有的历史,避免时间尺度上的先验。HiPPO-LegS具有时间尺度鲁棒性、快速更新和有界梯度的理论优势。通过将记忆动态纳入递归神经网络,HiPPO RNN可以凭经验捕获复杂的时间依赖性。在基准排列的MNIST数据集上,HiPPO-LegS将最新的准确率设定为98.3%。最后,在一个新的轨迹分类任务中,测试了对分布外时间尺度和缺失数据的鲁棒性,HiPPO-LegS的准确率超过了RNN和神经ODE基线的25-40%。

英文摘要:

A central problem in learning from sequential data is representing cumulative history in an incremental fashion as more data is processed. We introduce a general framework (HiPPO) for the online compression of continuous signals and discrete time series by projection onto polynomial bases. Given a measure that specifies the importance of each time step in the past, HiPPO produces an optimal solution to a natural online function approximation problem. As special cases, our framework yields a short derivation of the recent Legendre Memory Unit (LMU) from first principles, and generalizes the ubiquitous gating mechanism of recurrent neural networks such as GRUs. This formal framework yields a new memory update mechanism (HiPPO-LegS) that scales through time to remember all history, avoiding priors on the timescale. HiPPO-LegS enjoys the theoretical benefits of timescale robustness, fast updates, and bounded gradients. By incorporating the memory dynamics into recurrent neural networks, HiPPO RNNs can empirically capture complex temporal dependencies. On the benchmark permuted MNIST dataset, HiPPO-LegS sets a new state-of-the-art accuracy of 98.3%. Finally, on a novel trajectory classification task testing robustness to out-of-distribution timescales and missing data, HiPPO-LegS outperforms RNN and neural ODE baselines by 25-40% accuracy.

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

↑