2024年8月24日土曜日

メモリ内で「データ保存」と「計算処理」を実行する「CRAM」 実証実験に成功 AIタスクの消費電力を大幅削減へ



ニュース


Innovative Tech(AI+)

Innovative Tech(AI+):

このコーナーでは、2014年から先端テクノロジーの研究を論文単位で記事にしているWebメディア「Seamless」(シームレス)を主宰する山下裕毅氏が執筆。新規性の高いAI分野の科学論文を山下氏がピックアップし、解説する。

X: @shiropen2

 米ミネソタ大学と米アリゾナ大学に所属する研究者らが発表した論文「Experimental demonstration of magnetic tunnel junction-based computational random-access memory」は、データを転送せずにメモリ内で直接演算処理を行う手法を実験で実証した研究報告である。


(a)演算ユニットとメモリユニットの間でデータ転送を行う従来のコンピュータアーキテクチャ (b)メモリ内でだけでデータ保存と計算処理を行うCRAM (c)CRAMは、ニューラルネットワークや画像処理、エッジコンピュータなどへの応用が期待される

 機械学習などのデータ集約型アプリケーションは、膨大な計算能力を必要とするが、従来のコンピューティングアーキテクチャでは、演算ユニットとメモリユニットの間で絶え間なくデータを転送する必要があり、この転送プロセスに多大な電力とエネルギーを消費している。

 この課題に対処するため、研究者たちは「Computational Random-Access Memory」(CRAM)と呼ばれる技術に着目。CRAMの特徴は、メモリセル自体を使って論理演算を直接実行できる点で、データをメモリから外部に転送することなく、メモリ内で直接計算処理が可能。この方式により、従来のアーキテクチャと比較して、大幅なエネルギー効率の向上と処理性能の改善が期待できる。

advertisement

 CRAMのコンセプトは以前から先行研究で提案されており、数値シミュレーションによってその潜在的な利点が示されていた。例えば、機械学習の処理において、CRAMを使うと従来のシステムと比べて約1000倍効率が良くなると推定されている。別の例として、手書き数字を認識するタスクでは、CRAMを使用すると消費エネルギーが2500分の1に、処理時間が1700分の1になるという結果が得られている。

 しかし、これまでCRAMの実験的な実証や、実際の計算精度の評価は行われていなかった。そこでこの研究では、磁気トンネル接合(Magnetic Tunnel Junction、MTJ)デバイスを用いた1×7のCRAMアレイを実際に作製し、その動作を実験的に実証することに成功。MTJとは、2つの強磁性層の間に非常に薄い絶縁層が挟まれた構造を指し、読み取り機能と書き込み機能の両方を備えた電気的に動作するメモリデバイスとして使用できる。


CRAM実験セットアップ

 研究チームはまず、基本的なメモリ操作(CRAMアレイにデータを書き込み、そのデータを読み出す)と2入力(2つの入力に基づいて1つの出力を生成する演算)、3入力、5入力の論理演算を調査した。その結果、2入力のNAND演算では99.4%という高い精度を達成できることが分かった。一方、入力数が増えるにつれて精度は低下し、3入力のMAJ3演算では86.5%、5入力のMAJ5演算では75%の精度となった。

 次に、より複雑な演算として1ビット全加算器を2つの異なる設計で実装した。1つはNANDゲートのみを使用する設計で、もう1つは多数決(MAJ)ゲートと反転(NOT)ゲートを組み合わせた設計である。実験の結果、NANDゲートのみを使用した設計の方が、78.5%とより高い精度を示した。

 CRAMは特に、バイオインフォマティクス、画像処理、信号処理、ニューラルネットワーク、エッジコンピューティングなどのアプリケーションに適している可能性があり、今後の研究開発が期待される。

0 コメント:

コメントを投稿