提出树形结构训练方法,让深度网络并行计算,速度提升至对数级。
Breaking Chains with Trees: Model-Parallel Deep Learning with $\mathcal{O}(\log N)$ Time Complexity

- 将网络层构建成树形结构,实现前向与反向传播的分层并行计算。
- 在N层网络下,计算时间复杂度降至O(log N),显著提速。
- 支持视觉分类和语言建模,且可应用于循环神经网络。
现代深度神经网络采用误差反向传播训练,需逐层顺序执行前向与反向计算。随着网络加深,层间更新严格依赖,无法并行,导致训练仅能采用数据并行,限制了模型并行训练。本文提出TreeProp,一种架构无关的变分学习框架,将网络层组织为树形层次结构。训练中,用层次化计算替代传统的逐层前向与梯度反传,使中间表示和学习信号的构建时间复杂度降至$/mathcal{O}( ext{log } N)$(N为层数)。据我们所知,TreeProp是首个在训练过程中前向与反向传播均实现对数级并行时间复杂度的深度神经网络学习算法。此外,层次结构中存在多条有效路径,使TreeProp隐式学习具有不同有效深度的子网络,无需额外训练成本。我们在视觉分类与自回归语言建模任务上验证了TreeProp,性能与传统端到端训练相当,优于先前对比学习方法。同时证明其适用于依赖时间反向传播的循环神经网络。
原文摘要 · Abstract (English)
Modern deep neural networks are trained using error backpropagation, which requires sequential forward and backward computations across network layers. As these networks become deeper, this introduces limitations, since layer-wise updates are strictly interdependent and cannot proceed in parallel. These constraints restrict training procedures to data-parallel schemes, thereby prohibiting model-parallel training. We propose TreeProp, an architecture-agnostic variational learning framework that organizes network layers into a tree-structured hierarchy. During training, TreeProp replaces sequential forward computations and backward gradient propagation with hierarchical computations. This allows intermediate representations and learning signals to be constructed in time complexity of $\mathcal{O}(\log N)$ for a network of $N$ layers. To the best of our knowledge, TreeProp is the first learning algorithm for deep neural networks with logarithmic parallel time complexity for both forward computation and backward gradient propagation during training. Furthermore, we show that multiple valid paths through the hierarchy exist, such that TreeProp implicitly learns subnetworks with different effective depths, but without additional training effort. We evaluate TreeProp on vision classification and autoregressive language modeling, matching the performance of conventional end-to-end training for a variety of tasks and outperforming previous contrastive training approaches. We further demonstrate the applicability of TreeProp to recurrent neural networks that otherwise rely on backpropagation through time.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。