ParaRNN让循环神经网络更可解释且能并行计算
ParaRNN: An Interpretable and Parallelizable Recurrent Neural Network for Time-Dependent Data

- 用多个小循环单元构建模型,分解动态为可解释成分
- 在非参数回归中表现媲美传统RNN,预测误差有理论保证
- 适合需要可解释性与高效训练的时序建模任务
大规模复杂数据的兴起推动了机器学习在统计建模中的应用。循环神经网络(RNN)作为时序数据的基础模型,可视为经典自回归滑动平均模型的非线性扩展。尽管在机器学习中表现灵活且成功,传统RNN常因可解释性差和训练慢而限制其在统计学中的应用。本文提出并行化RNN(ParaRNN),由多个小型循环单元组成,具备可加表示,将循环动态解耦为可解释组件,其行为可通过递归特征刻画。该可解释性支持其在时序数据的非参数回归中的应用,同时设计支持高效并行化。本文建立了ParaRNN在非参数回归设置下的逼近能力及非渐近预测误差界。三个序列建模任务的实证结果表明,ParaRNN性能可媲美标准RNN,同时提升可解释性与训练效率。
原文摘要 · Abstract (English)
The proliferation of large-scale and structurally complex data has spurred the integration of machine learning methods into statistical modeling. Recurrent neural networks (RNNs), a foundational class of models for time-dependent data, can be viewed as nonlinear extensions of classical autoregressive moving average models. Despite their flexibility and empirical success in machine learning, RNNs often suffer from limited interpretability and slow training, which hinders their use in statistics. This paper proposes the Parallelized RNN (ParaRNN), a novel model composed of multiple small recurrent units. ParaRNN admits an additive representation that decouples recurrent dynamics into interpretable components, whose behavior can be characterized through recurrence features. This interpretability enables its applications in nonparametric regression for time-dependent data, while the design also allows efficient parallelization. The approximation capacity and non-asymptotic prediction error bounds in a nonparametric regression setting are established for ParaRNN. Empirical results on three sequential modeling tasks further demonstrate that ParaRNN achieves performance comparable to vanilla RNNs while offering improved interpretability and efficiency.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。