BT

最新技術を追い求めるデベロッパのための情報コミュニティ

寄稿

Topics

地域を選ぶ

InfoQ ホームページ ニュース 3世代のtensorプロセッシングユニットから得られた10の教訓

3世代のtensorプロセッシングユニットから得られた10の教訓

ブックマーク

原文(投稿日:2022/04/19)へのリンク

GoogleのTPUグループが発行した最近のレポートでは、TPUv1(2016)から最新の推論ユニットTPUv4i(2020)までの3世代のtensorプロセッシングユニットの開発から得られた10のポイントが挙げられている。著者はまた、これまでの経験が今後のTPUバージョンの開発にどのように影響するかについても説明している。このレポートでは、TPUに関するこれまでの作業だけでなく、ディープラーニング(DL)固有のアーキテクチャ開発の将来についてさらに考察している。

tensorプロセッシングユニットは、ディープニューラルネットワーク推論(TPUv1とTPUv4i)およびトレーニング(TPUv2からv4)用に設計されたGoogleのカスタムメイドのASICである。TPUの設計上の決定と最適化(関連するトレードオフ)は、一般的な(特定の)ディープネットワーク機能とアクティベーションを並列化することを目的としている。したがって、TPUは、ディープネットワークソリューションの開発が比較的安定している実稼働環境の処理に対して費用効果の高い選択肢となる。

ディープネットワークの進化が速くなっていることを考慮すると、TPUの開発にも、過去10年間に数回のイテレーションが必要であった。進捗レポートに記載されている重要な教訓は、次のように要約できる。

  1. 処理装置を構成するハードウェアコンポーネントは、それぞれ異なる速度で進化する(ロジック、ワイヤー、SRAM、DRAMなど)。
  2. アーキテクチャの成功と人気は、関連するコンパイラの最適化(TPUの場合はXLA)に大きく依存するが、設計者は、コンパイラがアーキテクチャ自体に従って反復して最適化されることに注意する必要がある。
  3. ドメイン固有のアーキテクチャを開発する場合は、パフォーマンスを向上させるために総所有コスト(TCO)を最適化することをお勧めする。
  4. 将来のDLシステムでは、深層学習処理での下位互換性のサポートが重要な要素であり、考慮に入れる必要がある。
  5. 推論ユニットのデプロイでは、リクエストの待ち時間を短縮するためにグローバルに分散させるため、空気冷却が必要になる場合がある。必要となる液体冷却は、それぞれの場所で用意できない場合がある。
  6. 一部のアプリケーションでは整数のみ(量子化)の推論が最適でない場合があるため、DLアーキテクチャは浮動小数点演算もサポートすべきである。
  7. 推論処理はマルチテナントで実行できるべきである。推論では勾配累積(Gradient Accumulation)が必要なく、比較的小さいリソースフットプリントで済むため、共有によってコストと全体的な遅延を削減できる可能性がある。
  8. 将来のDLアーキテクチャでは、ディープラーニングモデルのサイズを考慮に入れる必要がある(つまり、ディープネットワークはメモリとコンピューティングに対する要件によって年間約1.5倍に成長する)。
  9. 将来のDLアーキテクチャでは、ディープネットワーク機能とアクティベーションの多様性が増すことを考慮に入れる必要がある(つまり、ディープネットワークアーキテクチャは急速に進化する)。
  10. サービスレベル目標(SLO)の制限は、推論バッチサイズではなく、API p99の遅延によるべきであるため、将来のアーキテクチャでは、より大きなバッチを利用してアドバンテージを得る必要がある。

TPUv2とTPUv3の開発の詳細は、こちらのレポートで入手できる。昨年の夏、Googleはトレーニング用のv4ポッドを発表した。TPUv4の詳細については、GCPの公式ブログ投稿をご覧ください。TensorFlowJAXベースのFlaxに加えて、PyTorch APIでは、TPUでモデルを実行するためのXLAコンパイラのサポートも開始した。詳細については、関連ドキュメントを参照してください。このページでは、TPUによってネイティブにサポートされている事前トレーニング済みのディープネットワークをいくつか参照できる。

作者について

この記事に星をつける

おすすめ度
スタイル

特集コンテンツ一覧

こんにちは

コメントするには InfoQアカウントの登録 または が必要です。InfoQ に登録するとさまざまなことができます。

アカウント登録をしてInfoQをお楽しみください。

HTML: a,b,br,blockquote,i,li,pre,u,ul,p

コミュニティコメント

HTML: a,b,br,blockquote,i,li,pre,u,ul,p

HTML: a,b,br,blockquote,i,li,pre,u,ul,p

BT