原创

1804.03235-Large scale distributed neural network training through online distillation.md

版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://dwsun.blog.csdn.net/article/details/87778587

1804.03235-Large scale distributed neural network training through online distillation.md

现有分布式模型训练的模式

  • 分布式SGD
    • 并行SGD: 大规模训练中,一次的最长时间取决于最慢的机器
    • 异步SGD: 不同步的数据,有可能导致权重更新向着未知方向
  • 并行多模型 :多个集群训练不同的模型,再组合最终模型,但是会消耗inference运行时
  • 蒸馏:流程复杂
    • student训练数据集的选择
      • unlabeled的数据
      • 原始数据
      • 留出来的数据

协同蒸馏

  1. using the same architecture for all the models;
  2. using the same dataset to train all the models; and
  3. using the distillation loss during training before any model has fully converged.

特点
- 就算thacher和student是完全相同的模型设置,只要其内容足够不同,也是能够获得有效的提升的
- 即是模型未收敛,收益也是有的
- 丢掉teacher和student的区分,互相训练,也是有好处的
- 不是同步的模型也是可以的。

1251624-20180705233934478-106316767.png

算法简单易懂,而且步骤看上去不是很复杂。

使用out of state模型权重的解释:

  1. every change in weights leads to a change in gradients, but as training progresses towards convergence, weight updates should substantially change only the predictions on a small subset of the training data;
  2. weights (and gradients) are not statistically identifiable as different copies of the weights might have arbitrary scaling differences, permuted hidden units, or otherwise rotated or transformed hidden layer feature space so that averaging gradients does not make sense unless models are extremely similar;
  3. sufficiently out-of-sync copies of the weights will have completely arbitrary differences that change the meaning of individual directions in feature space that are not distinguishable by measuring the loss on the training set;
  4. in contrast, output units have a clear and consistent meaning enforced by the loss function and the training data.

所以这里似乎是说,随机性的好处?

一种指导性的实用框架设计:

  1. Each worker trains an independent version of the model on a locally available subset of the training data.
  2. Occasionally, workers checkpoint their parameters.
  3. Once this happens, other workers can load the freshest available checkpoints into memory and perform codistillation.
    再加上,可以在小一些的集群上使用分布式SGD。

另外论文中提到,这种方式,比起每次直接发送梯度和权重,只需要偶尔载入checkpoint,而且各个模型集群在运算上是完全相互独立的。这个倒是确实能减少一些问题。
但是,如果某个模型垮掉了,完全没收敛呢?

另外,没看出来这种框架哪里简单了,管理模型和checkpoint不是一个简单的事情。

实验结论

20TB的数据,有钱任性

论文中提到,并不是机器越多,最终模型效果越好,似乎32-128是比较合适的,更多了,模型收敛速度和性能不会更好,有时反而会有下降。
1251624-20180705233947458-96230653.png

论文中的实验结果2a,最好的还是双模型并行,其次是协同蒸馏,最差的是unigram的smooth0.9,label smooth 0.99跟直接训练表现差不多,毕竟只是一个随机噪声。
另外,通过对比相同数据的协同蒸馏2b,和随机数据的协同整理,实验发现,随机数据实际上让模型有更好的表现
3在imagenet上的实验,出现了跟2a差不多的结果。
4中虽然不用非得用最新的模型,但是,协同蒸馏,使用太久远的checkpoint还是会显著降低训练效率的。

欠拟合的模型是有用的,但是过拟合的模型在蒸馏中可能不太有价值。
协同蒸馏比双步蒸馏能更快的收敛,而且更有效率。

3.5中介绍的,也是很多时候面临的问题,因为初始化,训练过程的参数不一样等问题,可能导致两次训练出来的模型的输出有很大区别。例如分类模型,可能上次训练的在某些分类上准确,而这次训练的,在这些分类上就不准确了。模型平均或者蒸馏法能有效避免这个问题。

总结

balabalabala
实验只尝试了两个模型,多个模型的多种拓扑结构也值得尝试。

很值得一读的一个论文。

posted on 2018-07-05 23:40 dwSun 阅读(...) 评论(...) 编辑 收藏

文章最后发布于: 2018-07-05 23:40:00
展开阅读全文
0 个人打赏
评论将由博主筛选后显示,对所有人可见 | 还能输入1000个字符

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 技术黑板 设计师: CSDN官方博客

分享到微信朋友圈

×

扫一扫,手机浏览