新しいものづくりがわかるメディア

RSS


GoogleがAIを使って次世代AIチップを設計——数カ月の工程を6時間で

Googleは、機械学習のひとつ、深層強化学習を利用して、コンピューターチップを短期間で設計する手法を開発した。その手法を使うと、通常数カ月かかる処理を6時間以下に短縮し、性能や消費電力は同等かそれ以上のチップデザインを導き出せる。既に、次世代のAIプロセッサ「TPU(Tensor Processing Unit)」の設計にも利用された。研究結果は、2021年6月9日付けの『Nature』に掲載されている。

ムーアの法則とデナード則の速度が鈍化する一方、人々がコンピューターに求める性能は指数関数的に増加している。コンポーネントや配線など、コンピューターチップの物理的なレイアウトを決める「フロアプランニング」は、チップの性能に関わる重要な作業だ。しかし、過去50年にも及ぶ研究にも関わらず、この工程は自動化できていないため、専門のエンジニアが多くの時間と労力をかけて製造可能なレイアウトを作成している。

今回、Google ResearchとGoogle Hardware and Architectureのチームは、フロアプランニングを「強化学習問題」として提示し、グラフ畳み込みニューラルネットワーク(GCN)を構築。1万個のラベル付き配置パターンを使ってGCNを訓練し、最適なチップ配置を割り出した。これは、AIがフロアプランニングという名のボードゲームをしているようなもので、具体的にはチップのコンポーネントが駒、基板がゲーム盤で、制約条件を満たしつつ消費電力などを最小化するようにチップを動かしている。

最新のアクセラレータに対するフロアプランニングを人間とAIで比較すると、人の手では許容可能なフロアプランを作成するのに数カ月かかったのに対し、AIは6時間以内に完了しただけでなく、消費電力、性能、チップ面積などあらゆる指標で、人の手による設計解と同等もしくは上回っていた。

さらに、このAIは過去の経験から学び、時とともに改善していくことができる。特に、チップブロックの訓練数が増えるにつれて、見たことのないチップブロックに遭遇した場合でも、より良い配置を即座に生成できるようになった。このようにAIを活用すれば、世代が変わるごとに費やしていた何千時間もの労力が節約できる可能性がある。

今回はTPUの最適化を提示したが、ASICなど他のチップにも応用できるという。AI自身がチップ設計サイクルを短縮できるようになれば、AI設計の強力なハードウェアがAIの発展を促進するというように、AIとハードウェアの間に共生関係を構築できるだろうと、研究チームは考えている。

fabcross for エンジニアより転載)

関連情報

おすすめ記事

 

コメント

今人気の記事はこちら

  1. Raspberry PiでPCを遠隔管理——KVM over IP用拡張ボード「PiKVM v3 HAT」
  2. 低価格でも精密な造形——DLP方式3Dプリンター「Anycubic Photon Ultra」
  3. Raspberry Piで制御するロボットアーム——音声コントロール対応6軸ロボットアーム「mechArm」
  4. ステンレスへの刻印も——7.5Wダイオードレーザー加工機「OKU Desk」
  5. ArduinoがSeggerと提携——Arduio IDE 2.0で正式サポートへ
  6. モジュール式の竹製家具が製作できる3Dプリントのコネクタ——宇宙での利用にも期待
  7. 音波を減衰させて騒音を半減させるねじ——床や天井の建築材を薄くして空間を広く取ることも可能に
  8. 豊橋技術科学大学ICD-LAB「弱いロボット」に学ぶものづくりのアイデアのヒント
  9. 超音速微粒子の衝突に耐える超軽量「ナノアーキテクチャ材料」を開発
  10. バイオ燃料の限界を超える人工酵母を開発

ニュース

編集部のおすすめ

連載・シリーズ

注目のキーワード

もっと見る