Skip to main content
QUICK REVIEW

[論文レビュー] Gamora: Graph Learning based Symbolic Reasoning for Large-Scale Boolean Networks

Nan Wu, Yingjie Li|arXiv (Cornell University)|Mar 14, 2023
Machine Learning in Materials ScienceMaterials Science被引用数 3
ひとこと要約

Gamoraは、ゲートレベルのネットリストにおいて構造的ハッシュ化と機能的伝搬を同時に実行する、マルチタスク学習を活用したGNNベースの記号的推論フレームワークを提案する。ABCなどの最先端ツールに比べて最大6桁の速度向上を達成し、大規模なブールネットワーク(最大6桁のスケーリング)でほぼ100%の精度を実現する。GPUアクセラレーションにより、10億ノード規模のスケーラビリティと、複雑なテクノロジーマッピングにおいても強力な一般化性能を維持する。

ABSTRACT

Reasoning high-level abstractions from bit-blasted Boolean networks (BNs) such as gate-level netlists can significantly benefit functional verification, logic minimization, datapath synthesis, malicious logic identification, etc. Mostly, conventional reasoning approaches leverage structural hashing and functional propagation, suffering from limited scalability and inefficient usage of modern computing power. In response, we propose a novel symbolic reasoning framework exploiting graph neural networks (GNNs) and GPU acceleration to reason high-level functional blocks from gate-level netlists, namely Gamora, which offers high reasoning performance w.r.t exact reasoning algorithms, strong scalability to BNs with over 33 million nodes, and generalization capability from simple to complex designs. To further demonstrate the capability of Gamora, we also evaluate its reasoning performance after various technology mapping options, since technology-dependent optimizations are known to make functional reasoning much more challenging. Experimental results show that (1) Gamora reaches almost 100% and over 97% reasoning accuracy for carry-save-array (CSA) and Booth-encoded multipliers, respectively, with up to six orders of magnitude speedups compared to the state-of-the-art implementation in the ABC framework; (2) Gamora maintains high reasoning accuracy (>92%) in finding functional modules after complex technology mapping, upon which we comprehensively analyze the impacts on Gamora reasoning from technology mapping.

研究の動機と目的

  • 大規模なブールネットワーク(BNs)における従来の記号的推論手法のスケーラビリティと非効率性を解決すること。これらの手法は、並列性が限定された構造的ハッシュ化と機能的伝搬に依存している。
  • ABCのような既存ツールの限界を克服すること。これらは10億ノード規模の設計において高い実行時間とメモリ消費量を示す。
  • 単純な設計から複雑な設計(不規則構造を有するテクノロジーマッピング済みネットリストを含む)への記号的推論の一般化を可能にすること。
  • 現代のGPUアクセラレーションとグラフ学習を活用して、電子設計自動化(EDA)分野における高パフォーマンスでスケーラブルかつ並列処理可能な記号的推論を実現すること。
  • 機能的・構造的パターンを曇らせる複雑なテクノロジーマッピングの下でも、堅牢な推論精度を維持できることを示すこと。

提案手法

  • Gamoraは、ブール関数とトポロジーをエンコードするノード特徴量を統合することで、構造的類似性と機能的同等性の両方を同時に最適化するマルチタスクGNNアーキテクチャを採用する。
  • メッセージパッシング機構を用いて、記号的伝搬(機能的集約)と構造的ハッシュ化を同時にエミュレートし、従来の推論を模倣しつつも、スケーラビリティを向上させる。
  • 計算コストとメモリフットプリントを削減するために、ドメイン固有の技術を用いてノード特徴量を圧縮し、10億ノード規模のBNを効率的に処理可能にする。
  • 単一のGPU上でバッチ推論をサポートし、並列処理とGPUアクセラレーションを活用して、高スルーレートで大規模設計にスケーリングする。
  • テクノロジーライブラリの属性をノードおよびエッジ特徴量に統合することで、複雑なテクノロジーマッピング下での推論の堅牢性を向上させる。
  • モデルは、例えばCSAやブース符号化された乗算器などの多様な乗算器設計で学習され、再訓練やファインチューニングなしに、より大きなビット幅やマッピング済みネットリストへと一般化可能である。

実験結果

リサーチクエスチョン

  • RQ1GNNベースのフレームワークは、10億ノード規模の大規模なブールネットワークにおいて、高精度な記号的推論を達成できるか?
  • RQ2複雑なテクノロジーマッピングにより構造的不規則性と機能的複雑性が増加する状況下で、提案されたマルチタスクGNNの機能的ブロック推論性能はいかがなものか?
  • RQ3モデルは、単純な設計(例:小規模ビット幅乗算器)から、複雑で大規模な設計(例:2048ビット乗算器)へとどの程度一般化できるか?
  • RQ4大規模BNにおいて、GamoraはABCのような従来の記号的推論ツールに比べて、実行時間とスケーラビリティでどの程度の利点を示すか?
  • RQ5ノード特徴量にテクノロジーライブラリ属性を統合することで、高度なテクノロジーマッピング下での推論性能にどのような影響を与えるか?

主な発見

  • Gamoraは、2048ビットのキャリーベースアレイ(CSA)乗算器においてほぼ100%の推論精度を達成し、ブース符号化乗算器では97%以上の精度を示す。従来手法に比べ、精度と速度の両面で顕著に優れている。
  • 最大3400万ノードおよび6700万エッジを有する大規模BNにおいて、GamoraはABCの最先端フレームワークに比べ最大6桁の速度向上を達成する。
  • ASAP 7nmテクノロジーマッピングを経た後でも、機能モジュールの同定において92%以上の予測精度を維持し、テクノロジーディペンデントな変換に対して強い耐性を示す。
  • 再訓練なしに小規模なビット幅乗算器から大規模なビット幅乗算器へと一般化が効果的に可能であり、複雑なテクノロジーマッピング後の精度低下はわずかであり、より大きなトレーニングデータセットを用いることで緩和される。
  • 単一GPU上でのバッチ推論は、スルーレートの向上とレイテンシの低減を示し、将来的なマルチGPUデプロイメントの強力な可能性を示している。
  • ノードおよびエッジ特徴量にテクノロジーライブラリ属性を統合することで、特にマルチ出力セルや不規則構造を有する複雑なマッピング下での推論性能が向上する。

より良い研究を、今すぐ始めましょう

論文の読解から最終レビューまで、研究時間を劇的に削減しましょう。

クレジットカード登録不要

このレビューはAIが作成し、人間の編集者が確認しました。