数据并行对训练结果的影响

我目前正在尝试使用Tensorflow网络实现多GPU训练。一种解决方案是每个GPU运行一个模型,每个模型都有自己的数据批次,并在每次训练迭代后合并它们的权重。换句话说,这就是“数据并行”。

例如,如果我使用2个GPU,并行训练,然后合并它们的权重,那么与在一个GPU上按顺序训练这两个数据批次相比,结果权重应该有所不同吧?因为两个GPU有相同的输入权重,而单个GPU对第二个批次的权重进行了修改。

这种差异只是微小的,因此在多次迭代后的最终结果中并不重要吗?


回答:

输入训练的批次顺序确实会有一些影响。但如果你有大量批次,这种差异可能会很小。每个批次都会将模型中的变量稍微拉向损失的最小值。不同的顺序可能会使通向最小值的路径略有不同。但只要损失在减少,模型就在训练中,其评估结果会越来越好。

有时,为了避免相同的批次“拉”模型在一起,并且避免模型只对某些输入数据过于优秀,每个模型副本的输入会在输入训练程序之前随机打乱。

Related Posts

使用LSTM在Python中预测未来值

这段代码可以预测指定股票的当前日期之前的值,但不能预测…

如何在gensim的word2vec模型中查找双词组的相似性

我有一个word2vec模型,假设我使用的是googl…

dask_xgboost.predict 可以工作但无法显示 – 数据必须是一维的

我试图使用 XGBoost 创建模型。 看起来我成功地…

ML Tuning – Cross Validation in Spark

我在https://spark.apache.org/…

如何在React JS中使用fetch从REST API获取预测

我正在开发一个应用程序,其中Flask REST AP…

如何分析ML.NET中多类分类预测得分数组?

我在ML.NET中创建了一个多类分类项目。该项目可以对…

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注