米国国立研究プラットフォームにおけるGrace Hopper AIスーパーチップを用いた冒険

Kernekoncepter

本稿では、米国国立研究プラットフォーム（NRP）において、NVIDIAの最新AIチップであるGrace Hopper（GH200）の性能を、画像分類、物体検出、セマンティックセグメンテーションといったコンピュータビジョンタスクを用いて、従来のA100 GPUと比較評価しています。

Resumé

本稿は、米国国立科学財団（NSF）が資金提供する国立研究プラットフォーム（NRP）において、NVIDIAの最新AIチップであるGrace Hopper（GH200）の性能を評価した研究論文である。
研究の背景

グレートプレーンズ地域は人口がまばらで、サイバーインフラストラクチャの専門家が不足しているため、地域全体で協力し、知識を共有するコミュニティとして活動している。
グレートプレーンズネットワーク（GPN）は、この課題を克服するために設立され、高スループットコンピューティング（HTC）のための複数州に分散した計算ノードの展開において豊富な経験を積んできた。
GP-ENGINEプロジェクトの一環として、NRPにGH200を搭載した最初のサーバーを構築した。
研究の目的
本研究の目的は、GH200の性能を、画像分類、物体検出、セマンティックセグメンテーションといったコンピュータビジョンタスクを用いて、従来のA100 GPUと比較評価することである。
実験方法

3つのコンピュータビジョンタスク（画像分類、物体検出、セマンティックセグメンテーション）を用いて、GH200の性能を評価した。
各タスクにおいて、畳み込みニューラルネットワーク（CNN）とTransformerベースの深層学習モデルを用いた。
GH200の性能を、1基のNVIDIA A100（1xA100）と、分散データ並列処理を利用した4基のNVIDIA A100（4xA100 DDP）の2つのハードウェア構成と比較した。
結果
画像分類

GH200は、ResNet50とResNet152において、1xA100よりもそれぞれ32.3%、39.3%高速であったが、4xA100よりもそれぞれ20.4%、30.2%低速であった。
Transformerベースのアーキテクチャでも同様の傾向が見られ、4xA100は、最も計算量の多いモデルであるViT-Hにおいて、GH200よりも42.7%高速であったが、GH200は、1xA100よりも54.3%高速であった。
物体検出

SWIN-Lバックボーンを持つDINOモデルでは、GH200は1xA100よりも72.5%高速であり、4xA100よりも9.9%高速であった。
ResNet-50バックボーンを持つFaster R-CNNでは、GH200は1xA100よりも58%、4xA100よりも90.9%低速であった。
セマンティックセグメンテーション

GH200は、SegFormerとDeepLabV3+において、1xA100よりもそれぞれ57.9%、33.7%高速であった。
GH200は、4xA100よりも、これらのネットワークにおいてそれぞれ39.3%、57.6%低速であった。
結論

GH200は、1xA100と比較して、コンピュータビジョンタスクにおいて優れた性能を発揮する。
4xA100構成と比較すると、GH200の性能はタスクとモデルのアーキテクチャによって異なる。
GH200は、最新のアーキテクチャ（Transformerベースのモデルなど）を使用した場合に優れた性能を発揮するが、従来のアーキテクチャ（CNNベースのモデルなど）を使用した場合には、従来のGPUよりも性能が劣る場合がある。

Statistik

GP-ENGINEは、2023年10月1日から現在までに、米国で101の研究グループを支援し、115,000 GPU時間と160万 CPU時間を提供している。
GH200は、密結合されたARM CPU（72コア）とH100 GPU（14,592 CUDAコア）、および600 GBの合計RAMを搭載している。
比較対象として使用したA100 GPUは、80GBのGPUメモリと27,648個のCUDAコアを搭載している。

Vigtigste indsigter udtrukket fra

Adventures with Grace Hopper AI Super Chip and the National Research Platform

by J. Alex Hurt... kl. arxiv.org 10-23-2024

https://arxiv.org/pdf/2410.16487.pdf

Dybere Forespørgsler

GH200は、自然言語処理や音声認識などの他のAIタスクにおいても、従来のGPUと比較して優れた性能を発揮するのか？

GH200は、その大規模なメモリ容量と帯域幅、そしてARM CPUとの緊密な統合により、自然言語処理や音声認識といったメモリ集約型のAIタスクにおいても優れた性能を発揮する可能性があります。

自然言語処理(NLP)：Transformerモデルのような大規模言語モデルは、膨大な数のパラメータと学習データを必要とします。GH200の大容量メモリは、これらの巨大なモデルやデータセットを扱う上で有利に働き、学習の高速化やより大規模なモデルの学習を可能にする可能性があります。
音声認識：音声認識タスクも、大量の音声データと複雑なモデルを扱う必要があるため、GH200のメモリ容量と帯域幅が性能向上に寄与すると考えられます。特に、リアルタイム音声認識のような低遅延が求められるタスクでは、CPUとGPU間のデータ転送の高速化が重要となるため、GH200のアーキテクチャは有利に働く可能性があります。
ただし、GH200の実際の性能は、タスクの特性、モデルの設計、ソフトウェアの最適化など、様々な要因に依存します。具体的なタスクにおけるGH200の有効性を評価するためには、更なるベンチマークや実証実験が必要となります。

GH200の電力効率は、その性能向上に見合っているのか？

GH200の電力効率に関する情報は、現時点では公開されていません。電力効率は、性能と消費電力のバランスで決まるため、GH200の電力効率を評価するには、具体的な消費電力に関するデータが必要です。
一般的に、新しい世代のハードウェアは、電力効率が改善される傾向があります。GH200も、TSMCの4nmプロセスで製造されていることから、A100と比較して電力効率が向上している可能性があります。
しかし、電力効率の評価は、性能向上と消費電力のバランスを考慮する必要があります。GH200の電力効率が、その性能向上に見合っているかどうかは、具体的なアプリケーションやユースケースにおける消費電力と性能のトレードオフを分析した上で判断する必要があります。

AIチップの進化は、今後、高性能コンピューティングの分野をどのように変えていくのか？

AIチップの進化は、高性能コンピューティング(HPC)分野に大きな変化をもたらすと予想されます。

HPCとAIの融合: AIチップの進化は、従来のHPCアプリケーションにAI技術を統合する流れを加速させます。創薬、材料科学、金融モデリングなどの分野では、AIとHPCを組み合わせることで、より複雑な問題に取り組めるようになり、精度の向上や開発期間の短縮などが期待されます。
ドメイン特化型アーキテクチャの台頭: GH200に見られるように、特定のタスクに最適化されたドメイン特化型アーキテクチャの開発が進むと考えられます。これにより、汎用的なCPUやGPUと比較して、電力効率や性能が大幅に向上する可能性があります。
エッジコンピューティングの進化: AIチップの低消費電力化が進めば、エッジデバイスでのAI処理が現実的になります。これにより、自動運転、IoT、医療機器など、様々な分野でリアルタイム処理やデータプライバシーの向上が期待されます。
AIチップの進化は、HPC分野に新たな可能性をもたらすと同時に、ソフトウェア開発やシステム設計にも影響を与えます。今後、AIチップの進化がHPC分野にどのような変化をもたらすのか、注目していく必要があります。