arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2501.08040cs.LGmath.PRstat.ML

一类循环神经网络的实时循环学习(RTRL)收敛性分析

Convergence Analysis of Real-time Recurrent Learning (RTRL) for a class of Recurrent Neural Networks

Mathematical Institute, University of Oxford(牛津大学数学研究所);Department of Mathematics & Statistics, Boston University(波士顿大学数学与统计系)

Samuel Chun-Hei Lam, Justin Sirignano, Konstantinos Spiliopoulos

AI总结:

该研究针对截断反向传播时间算法在长序列RNN训练中梯度近似误差大的问题,证明了实时循环学习(RTRL)算法对一类RNN的收敛性,通过数值研究验证理论,并给出其在金融限价订单簿数据中的应用示例。

摘要:

循环神经网络(RNN)通常采用截断反向传播时间(TBPTT)算法进行训练。为保证计算可行性,TBPTT算法会对链式法则进行截断,仅在完整数据序列的有限块上计算梯度。由于截断反向传播的块长度通常远小于整体序列长度,这种近似可能导致显著的误差。与之不同,实时循环学习(RTRL)是一种在线优化算法,当序列时间步数量$t \rightarrow \infty$时,它能渐近逼近数据序列上损失函数的真实梯度。RTRL前向传播RNN隐藏/记忆单元关于参数的导数,并利用这些前向导数,在数据序列的每个时间步对参数进行在线更新。尽管对于参数量较大的模型而言计算成本较高,但RTRL的在线前向传播机制能够在极长数据序列上实现精确优化。我们证明了RTRL算法在一类RNN上的收敛性。该收敛分析表明,当来自序列的数据样本数量与训练步数趋于无穷时,数据序列、RNN隐藏层以及RNN隐藏层前向导数的联合分布存在一个不动点。我们证明RTRL算法会收敛到损失函数的一个驻点。数值研究验证了我们的理论结果。RTRL的一个潜在应用领域是金融数据分析,这类数据通常包含长时间序列,且对应模型的参数量为中小规模,这使得RTRL在计算上具备可行性,是一种有吸引力的模型训练优化方法。因此,我们还提供了一个将RTRL应用于限价订单簿数据的示例。

英文摘要:

Recurrent neural networks (RNNs) are commonly trained with the truncated backpropagation-through-time (TBPTT) algorithm. For the purposes of computational tractability, the TBPTT algorithm truncates the chain rule and calculates the gradient on a finite block of the overall data sequence. Such approximation could lead to significant inaccuracies, as the block length for the truncated backpropagation is typically limited to be much smaller than the overall sequence length. In contrast, Real-time recurrent learning (RTRL) is an online optimization algorithm which asymptotically follows the true gradient of the loss on the data sequence as the number of sequence time steps $t \rightarrow \infty$. RTRL forward propagates the derivatives of the RNN hidden/memory units with respect to the parameters and, using the forward derivatives, performs online updates of the parameters at each time step in the data sequence. RTRL's online forward propagation allows for exact optimization over extremely long data sequences, although it can be computationally costly for models with large numbers of parameters. We prove convergence of the RTRL algorithm for a class of RNNs. The convergence analysis establishes a fixed point for the joint distribution of the data sequence, RNN hidden layer, and the RNN hidden layer forward derivatives as the number of data samples from the sequence and the number of training steps tend to infinity. We prove convergence of the RTRL algorithm to a stationary point of the loss. Numerical studies illustrate our theoretical results. One potential application area for RTRL is the analysis of financial data, which typically involve long time series and models with small to medium numbers of parameters. This makes RTRL computationally tractable and a potentially appealing optimization method for training models. Thus, we include an example of RTRL applied to limit order book data.

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

↑