用最优控制理论统一神经网络在线学习,实现无需未来信息的持续学习。
A Unified Framework for Neural Computation and Learning Over Time
- 基于微分方程构建统一学习框架,无需外部求解器。
- 可恢复传统梯度学习,支持分布式与无激活存储计算。
- 适合研究持续学习、边缘计算及高效神经网络架构的设计者。
本文提出哈密顿学习(Hamiltonian Learning),一种全新的统一框架,用于在时间上进行神经网络学习,即从可能无限的数据流中以在线方式学习,且无法获取未来信息。现有工作通常局限于已知有限长度或分段数据流的简化场景,并依赖统计机器学习中的成熟学习策略。本文从头重构时间学习问题,借助最优控制理论,为神经计算与学习的时序动态提供统一视角。该框架基于可直接积分的微分方程:(i) 无需外部求解器即可集成;(ii) 一般化前馈与循环网络中的梯度学习概念;(iii) 开启全新研究视角。实验验证其可恢复梯度学习,对比现成优化器表现,并展示其灵活性:可在完全局部、部分/非局部计算方案间切换,支持多设备分布计算,且实现无需存储激活值的反向传播。该框架易于实现,有助于研究者以系统化、创新性方式应对持续学习挑战。
原文摘要 · Abstract (English)
This paper proposes Hamiltonian Learning, a novel unified framework for learning with neural networks "over time", i.e., from a possibly infinite stream of data, in an online manner, without having access to future information. Existing works focus on the simplified setting in which the stream has a known finite length or is segmented into smaller sequences, leveraging well-established learning strategies from statistical machine learning. In this paper, the problem of learning over time is rethought from scratch, leveraging tools from optimal control theory, which yield a unifying view of the temporal dynamics of neural computations and learning. Hamiltonian Learning is based on differential equations that: (i) can be integrated without the need of external software solvers; (ii) generalize the well-established notion of gradient-based learning in feed-forward and recurrent networks; (iii) open to novel perspectives. The proposed framework is showcased by experimentally proving how it can recover gradient-based learning, comparing it to out-of-the box optimizers, and describing how it is flexible enough to switch from fully-local to partially/non-local computational schemes, possibly distributed over multiple devices, and BackPropagation without storing activations. Hamiltonian Learning is easy to implement and can help researches approach in a principled and innovative manner the problem of learning over time.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。