跳到正文
OpenReview · State space models· Zhihao Jia; Matei Zaharia; Alex Aiken·· 2018-01-01精选AI 评分84

超越深度神经网络的数据并行与模型并行

Beyond Data and Model Parallelism for Deep Neural Networks

AI 导读

基于摘要分析。本文研究深度神经网络训练的并行化效率,提出比单独采用数据并行或模型并行更全面的策略搜索空间 SOAP,并通过 FlexFlow 框架为特定并行机器搜索高效的训练策略;主要贡献属于训练系统优化,而非神经信号解码方法。 SOAP 涵盖 Sample、Operation、Attribute 和 Parameter 四个维度的并行化。FlexFlow 使用引导式随机搜索探索这一空间,并以执行模拟器预测候选策略的性能,减少逐一实际执行策略的搜索开销。摘要未提供各维度的具体划分规则、模拟器建模方式或搜索细节,因此不能据此判断其支持的算子范围及性能预测误差。 作者报告,在两个 GPU 集群上的六个真实 DNN 基准中,FlexFlow 相较当时的先进方法,计入搜索时间后仍可实现最高 3.8 倍训练吞吐量,并改善可扩展性;作者还报告,执行模拟器比需要实际执行每个策略的既有方法快三个数量级。具体基准名称、集群配置、对照实现、吞吐量计量方式、实验协议、消融及统计信息尚未验证,这些结果不能直接外推到其他模型或硬件。 平台推测(待验证):其可能对应大规模 EEG 基础模型训练中的多 GPU 计算与通信瓶颈,而非直接解决 EEG 低信噪比或跨被试泛化问题。可复用部分是并行策略搜索与性能模拟思路,但需针对 EEG 模型的算子、序列长度、批大小和硬件通信成本重新适配。对于小数据或较小模型,搜索与通信开销可能抵消收益;并行策略如改变有效批大小,也需核对低信噪比数据下的训练稳定性。一个可检验的假设是:在同一 EEG 模型、固定数据划分、有效批大小及训练预算下,对比 FlexFlow 搜索策略与数据并行,测量包含搜索时间的端到端训练耗时、吞吐量及解码指标,验证系统加速是否伴随任务性能变化。相关 EEG 应用工作尚未检索确认。

阅读价值

值得阅读的是 SOAP 如何扩展训练并行化策略空间,以及 FlexFlow 如何用执行模拟降低策略搜索成本;其对 EEG/BCI 训练的收益尚未验证。

来源:OpenReview · State space models · openreview.net