アーカイブ / Computer Vision
PIMの電圧低下を重みのビット構成と実行時制御から抑えるAIM
AIM
AIMの原論文から掲載した図です。細部はクリックして拡大できます。
原論文の図の説明を読む
Figure 10. (a) Power supply and V-f adjustment at Macro Group granularity (b) IR-Booster regulated by IRFailure
概要
- AIMは量子化済みの重みからHRを求め、入力で変動するビット切り替わり率Rtogの理論上限として使う。
- LHRとWDSによる重みの調整、HRに基づくタスク配置、IR-Boosterによる電圧と周波数の選択を組み合わせる。
- 著者らは7nm、256-TOPSのPIMチップ設計のレイアウト後シミュレーションで、IRドロップを最大69.2%低減したと報告している。
補足:編集した模式図で仕組みを確認する
VISUAL EXPLAINER
図でつかむ、AIMの仕組み
- 01重みと入力画像
- 02重みのHR低減
- 03HRに基づく配置
- 04電圧と周波数の調整
- 05推論結果
原論文の説明をもとにした模式図です。処理の細部は省略しています。
研究の背景
AIMは、SRAMというメモリーの中で計算するPIMの電源電圧低下を扱う。PIMでは多くの計算回路が同時に動き、電流が増えると回路に届く電圧が下がる。この現象をIRドロップと呼ぶ。
電圧低下は処理の遅れや動作不良につながり得る。著者らは、対象の7nm、256-TOPSのSRAM PIMチップで140mVのIRドロップを観測したと述べ、実行するモデルの性質を対策に利用している。
手法
AIMはまず、メモリーに置く量子化済みの重みについて、ビットのうち1が占める割合をHRとして計算する。一方、Rtogはメモリーから加算回路へ送られるビット列がサイクル間で切り替わる割合で、入力によって変わる。例えば画像推論で入力画像が変わっても、保存した重みのHRは変わらず、そのRtogの理論上限になる。
次にLHRを量子化を伴う学習へ加えてHRの高い重みを抑え、WDSで量子化後の重みの分布を調整する。コンパイラーはHRを参照して計算タスクをPIMの計算単位へ配置する。実行時にはIR-BoosterがHRとチップ内の電圧低下監視を使い、各計算単位の供給電圧と動作周波数の組を選ぶ。
新規性
AIMは、メモリー内に置く量子化済みの値だけから求まるHRを、入力に応じて変動するRtogの理論上限として使う。これにより、入力を逐一調べなくても、重みに基づいて電圧低下への対策を組み立てられる。
著者らは、重みの調整とタスク配置を、チップ内の電圧低下監視を使うIR-Boosterにつないだ。事前のソフトウェア処理と実行時のハードウェア制御を組み合わせる構成が提案の中心である。
従来手法との違い
従来の回路段階の対策として、著者らは電源配線やセル配置の変更、電荷を蓄える素子の追加を挙げる。これらには設計時間や資源がかかり、電力、性能、面積に影響し得ると説明している。
AIMは、重みから計算したHRを使って配置と動作条件を変える。確認できた本文の範囲では、個々の回路対策とAIMを同じ条件で比較した数値は示されておらず、対策間の優劣までは判断できない。
実験結果
著者らは、7nmのデジタルPIMと28nmのアナログPIMの計算単位をHSPICEで調べ、RtogとIRドロップの相関係数をそれぞれ0.977、0.998と報告した。また、ResNet18とViTの演算について5万サイクルのRtog分布を調べ、観測最大値が対応するHRを超えなかったと述べる。
7nm、256-TOPSのPIMチップ設計では、RedHawkとHSPICEによるレイアウト後のシミュレーションを用いた。著者らはIRドロップを最大69.2%低減し、エネルギー効率を最大2.29倍、または性能を1.152倍に改善したと報告する。確認できた本文の範囲では、これらの改善値の比較対象と集計条件の詳細は読み取れない。
応用の可能性
編集上の応用案 · 論文が実証した用途とは区別しています。
編集上の応用案:対象PIMで画像分類モデルを運用する際、画像ごとの入力変化を追わずに電圧低下を管理できるか試す。入力はモデルの重みと推論画像、出力は分類結果と、調整後の重み・タスク配置・動作条件である。
導入には、量子化を伴う学習に使う画像と正解ラベル、品質確認用の画像とラベルが要る。重みをLHRで調整し、WDSとHRに基づく配置を行ったうえで、対象PIMの電圧低下監視をIR-Boosterに接続する。この用途での改善は検証対象である。
応用先の候補からNauhに相談できます。
導入時の検証
導入時の検証案 · 対象データでの再評価が必要です。
導入時の検証案:対象の画像分類モデルの重み、量子化を伴う学習用の画像と正解ラベル、評価用の画像とラベルを固定する。対象PIMの電圧低下監視値を記録できるようにし、元の重みと配置でIRドロップ、処理速度、エネルギー効率、分類精度を測る。
次にLHRによる重み調整、WDS、HRに基づく配置、IR-Boosterを順に適用し、同じ評価画像で各段階を比較する。HRと監視値の対応、配置に要する時間、分類精度の変化も測る。監視機能と各処理を対象PIMに適合させる作業量も確認する。
限界と課題
著者らは、HRに基づくタスク配置では複数回のシミュレーションにより秒単位の遅延が生じるため、利用に注意が必要と述べる。重みの追加学習が難しい場合には、LHRを使わずIR-Boosterを単独で使う構成も示している。
編集上の検証事項は、対象チップでもHRと電圧低下の関係が成り立つか、重み調整後にモデルの精度を保てるかである。報告された改善値はレイアウト後のシミュレーションによるもので、別のチップや運用条件への適用結果は確認できない。
出典
元のタイトル・要旨を確認する(英語)
AIM: Software and Hardware Co-design for Architecture-level IR-drop Mitigation in High-performance PIM
SRAM Processing-in-Memory (PIM) has emerged as the most promising implementation for high-performance PIM, delivering superior computing density, energy efficiency, and computational precision. However, the pursuit of higher performance necessitates more complex circuit designs and increased operating frequencies, which exacerbate IR-drop issues. Severe IR-drop can significantly degrade chip performance and even threaten reliability. Conventional circuit-level IR-drop mitigation methods, such as back-end optimizations, are resource-intensive and often compromise power, performance, and area (PPA). To address these challenges, we propose AIM, comprehensive software and hardware co-design for architecture-level IR-drop mitigation in high-performance PIM. Initially, leveraging the bit-serial and in-situ dataflow processing properties of PIM, we introduce Rtog and HR, which establish a direct correlation between PIM workloads and IR-drop. Building on this foundation, we propose LHR and WDS, enabling extensive exploration of architecture-level IR-drop mitigation while maintaining computational accuracy through software optimization. Subsequently, we develop IR-Booster, a dynamic adjustment mechanism that integrates software-level HR information with hardware-based IR-drop monitoring to adapt the V-f pairs of the PIM macro, achieving enhanced energy efficiency and performance. Finally, we propose the HR-aware task mapping method, bridging software and hardware designs to achieve optimal improvement. Post-layout simulation results on a 7nm 256-TOPS PIM chip demonstrate that AIM achieves up to 69.2% IR-drop mitigation, resulting in 2.29x energy efficiency improvement and 1.152x speedup.
論文本体の取得範囲に基づく解説。AIが作成した未校閲の記事です。性能の数値は著者の評価条件に依存します。応用例と検証計画は編集上の提案です。解説更新:2026-10-01




