PyTorch とかで複数GPUを使って学習する場合に DP と DDP って何が違うのか毎回調べてたのでまとめました。

本記事は下記記事やドキュメントをさらに噛み砕いたものとなっています。
特に一つ目の記事は少し前の記事ですがわかりやすいのでおすすめです。

• Distributed data parallel training using Pytorch on AWS – Telesens
• DataParallel — PyTorch 1.9.0
• Distributed Data Parallel — PyTorch 1.9.0

Q. DP と DDP どっち使えばよいの？

A. DDP 使え（DDPの方が速いし PyTorch のドキュメントでも DDP が推奨されてるから）

結論: 学習時の DP と DDP の違い

• 複数GPUがある想定で、GPU:0 とそれ以外の GPU を 他GPU とします
• GPU間通信（データの送信）が起きる場所を太字で表します
• 雑に言うと DP の場合1回のパラメータ更新に対し GPU:0 を起点とした GPU 間通信（データ送信）回数が DDP より多いです
  • そのため DP の方が遅くなりやすい = DDP の方が速くなりやすい です

1. Single GPU での学習時の挙動について

ここでは DP と DDP の違いを説明するために、通常の Single GPU での学習の要点を記載します。

• Dataloader を一つ用意する
• 下記のように1batchあたりの学習を行う

for batch in Dataloader
    inputs, targets = batch
    outputs = Model(inputs)  # Run forward pass = forward pass 計算
    loss = Loss(outputs, targets)  # Loss 計算
    loss.backward()  # backward pass 計算と勾配計算
    Model.parameter.update()  # Model のパラメータ更新

2. DP と DDP の学習時の挙動

図は Distributed data parallel training using Pytorch on AWS – Telesens より引用。

2.1 DP (Data Parallel)

ここではGPU0 と GPU0 以外のGPUを 他GPU として説明します。
太字の行はGPU間通信が発生している場所を表します。

DP の場合1回のモデルパラメータ更新に対して複数回 GPU 間通信が発生していることがわかります。

• 1. GPU0 にディスクからデータを読み込み
• 2. GPU0 から 他GPU に対し batch を分割して 送信する
• 3. 最新のモデル(のパラメータ) を GPU0 から 他GPU に送信する
• 4. 各GPUで forward pass 計算(outputs = Model(inputs))
• 5. 他GPU から GPU0 に outputs を集約して loss を計算する
  • loss の計算は分割前の batch で計算される
• 6. loss を GPU0 から 他GPU に送信して各 GPU でモデルの各パラメータの勾配を計算する
  • 他GPU でも勾配を計算するのは、各 GPU で inputs が異なるため forward pass も変わりうる可能性があるため
• 7. 他GPU から GPU0 に各パラメータを送信して集計する
• 8. GPU0 でモデルのパラメータを更新する

2.2 DDP (Distributed Data Parallel)

図は Distributed data parallel training using Pytorch on AWS – Telesens より引用。

太字の行はGPU間通信が発生している場所を表します。

DDP の場合1回のモデルパラメータ更新に対して1度のみGPU間通信を行えば良いです(勾配値の共有)
(正確にはあるGPUでは 全GPU数 - 1 回だけ通信する)

DDP の方が DP よりGPU間の通信（データ送信）回数や量が少ないため、基本的に DDP の方が速いです。

• 1. 各GPUに対し予め Dataloader を用意する(各Dataloader でデータが重複しないようにする)
• 2. 各GPU 担当の Dataloader がディスクからデータを読み込む
• 3. 各GPUで forward pass 計算(outputs = Model(inputs))
• 4. 各GPUでLossを計算し、モデルの各パラメータの勾配を計算してGPU間で勾配を集計する(この時点で各GPUで同じ勾配値を持つ)
  • このとき GPU0 を介さないデータ送信も起きる(ex. GPU1 と GPU2 間)
  • 例えば全 GPU が 3 つの場合、あるGPUで計算した勾配値の 1/3 を他 GPU に向けて送れば良い
• 5. 各GPU でパラメータ更新
  • 初めに一度だけ同じパラメータのモデルを共有し、後に共有される勾配値が同じ値であればパラメータ更新後も同じパラメータのモデルとなる

Reference

• Distributed data parallel training using Pytorch on AWS – Telesens