あなたは、ロボットにこれまでに作られたあらゆるビデオゲームを次から次へとプレイさせる方法を教えようとしていると想像してください。問題は、ロボットが「スペースインベイド」を学習すると、「パックマン」の遊び方を忘れてしまうことがよくある点です。これは「破滅的忘却(catastrophic forgetting)」と呼ばれ、人工知能にとって大きな悩みの種となっています。これを解決するために、科学者たちは通常、新しいことを学べなくなる代わりにロボットの脳を凍結させて忘却を防ぐか、あるいは自由に学習させる代わりに古いことを忘れてしまうかのどちらかを試みます。しかし、自然界にはもっと優れたトリックがあります。私たちの脳は、すべてを永遠に保存するのではなく、重要なもののためのスペースを作るために、退屈なことを積極的に忘れるのです。また、記憶にはそれぞれ異なる持続時間を持つ異なる種類のメモリが存在します。本論文では、この生物学的なバランス調整を模倣した新しいAIの構築方法を探求していますが、それは低速でエネルギー消費の激しいコンピュータチップを使うのではなく、生きたメモリセルのように機能する特殊な電子部品を使用しています。
イシュ・ジャン氏と浙江大学や清華大学などの機関のチームが率いる研究者たちは、「ハイブリッド・ライフロング・ラーニング(継続学習)」システムを作り上げました。これは、2つの非常に異なる部分が連携して機能するロボットの脳のようなものだと考えてください。まず、彼らは2つの生物学的なアイデアを組み合わせたスマートなソフトウェア・アルゴリズムを設計しました。それは「能動的忘却」(効率を維持するために物事を忘れるショウジョウバエから着想を得たもの)と、「メタ可塑性」(重要な記憶を強化する人間から着想を得たもの)です。次に、彼らはソフトウェアと同じように自然に振る舞う「自己選択型メモリスタ(SSM)」と呼ばれる物理的なハードウェア・デバイスを構築しました。これらの微小なデバイスは、電気抵抗を変化させて情報を保存できますが、通常のコンピュータメモリとは異なり、弱い記憶をゆっくりと消去しながら強い記憶を保持するという、人間の脳のように自然に備わった「漏れ(リーク)」を持っています。
チームはこれをコンピュータ上でシミュレーションしただけでなく、実際に1,024個のこれらのメモリスタ(32x32のグリッド状に配置)を備えた小さなチップを製作し、テストを行いました。彼らは、これらのデバイスが「忘れる」速度を調整することで、チップ内に5つの異なる「メモリ・コア」のチームを作り出せることを見出しました。いくつかのコアは非常に忘れやすく柔軟に設定されており(難易度の高い新しいタスクを素早く学習するのに適している)、他のコアは非常に安定していて頑固に設定されていました(古い重要なスキルを保持するのに適している)。このシステムを、一連の20種類の画像認識タスク(CIFARというデータセットを使用)でテストしたところ、システムは前のタスクを忘れることなく、それらすべてを学習することに成功しました。
結果は目覚ましいものでした。標準的なコンピュータチップ(CMOS)と比較して、この新しいシステムは3.42倍速く、エネルギー消費量は91分の1になると予測されました。著者らは、このアプローチが「古い知識を安全に保持すること」と「新しいことを学ぶこと」の間の長年の葛藤を解決し、私たちと同じように、現実世界で真に継続的に学習できるAIへの道筋を示すものであると示唆しています。彼らは、核となる学習ルールはハードウェア上でテストされたものの、20のタスク・チャレンジにおける完全なパフォーマンス数値は、彼らの実機測定値に基づいて校正されたシミュレーションに基づいていることを強調しており、彼らの新しいデバイスの物理特性が、複雑なライフロング・ラーニングを確かにサポートできることを示しています。
技術要約:自己選択型メンリスタに基づく生物学的着想によるハイブリッド生涯学習システム
問題提起
動的かつオープンエンドな環境で作動する人工知能システムには、以前のスキルを保持しながら知識を継続的に獲得する能力、すなわち「生涯学習(Lifelong Learning)」能力が求められる。現在のAIモデルは、「安定性と可塑性のジレンマ」に直面している。これは、新しいタスクを学習すると既存の表現が上書きされ、壊滅的忘却(Catastrophic Forgetting)を招く現象である。既存の解決策は、一般に二つの相反するカテゴリーに分かれる。一つは適応性を犠牲にして知識を保護する正則化ベースの手法であり、もう一つは柔軟性を促進するものの学習された表現の不安定化を招く恐れのある動的アーキテクチャやメモリ再生手法である。さらに、これらのソフトウェア中心のアプローチは、多大な計算オーバーヘッドとメモリコストを強いる。複雑な生物学的着想に基づくアルゴリズムと、それを実行するためのハードウェアとの間には決定的な乖離が存在する。ほとんどの生物学的モデルは、個別の生物学的メカニズム(例:能動的な忘却、あるいはメタ可塑性)を孤立して採用しており、それらの相乗的な統合を実現していない。また、メモリとプロセッシングを分離したフォン・ノイマン型アーキテクチャに依存しているため、エッジアプリケーションにおいて過大なエネルギー消費とレイテンシの問題を引き起こしている。
手法
著者らは、補完的な神経生物学的原理をハードウェアネイティブな学習システムへと統合することにより、この溝を埋めるハードウェア・アルゴリズムの協調設計(Co-design)パラダイムを提案する。
- アルゴリズム・フレームワーク (Hybrid Lifelong Learning - HLL):
HLLモデルは、二つの異なる生物学的メカニズムを融合させている:
- ショウジョウバエに着想を得た動的能動的忘却 (Drosophila-inspired Dynamic Active Forgetting - DAF): ショウジョウバエのキノコ体(Mushroom Body)を模倣し、リソース配分を最適化し可塑性を高めるために冗長なシナプスを剪定する。これは、「能動的忘却率」(α) を伴い、弱く増強された入力は急速に減衰する一方で、強く増強された入力は保持される。
- ヒトの脳 (Human Brain - HB) に着想を得たメタ可塑性と固定化: ヒトの脳のメカニズムから着想を得たこのコンポーネントは、固定化された知識を保護し、前方転移(Forward Transfer)を促進する。タスク固有の期待値や誤差に基づいてシナプス可塑性を変調するために、グローバルなゲインパラメータ (β) を利用する。これは、ドーパミンによるメタ可塑性に類似している。
- ハードウェア実装 (Self-Selective Memristors - SSM):
これらのダイナミクスを物理的に実現するために、著者らはITO/InOx/GaOx/Pt自己選択型メンリスタを用いた1-kb (32×32) クロスバーアレイを製作した。
- デバイス物理: この二層ヘテロ構造は、整流障壁として機能する低酸素空孔濃度(7.6%)のGaOx層と、抵抗スイッチングを促進する高酸素空孔濃度(65.33%)のInOx層を備えている。この非対称性により、高い自己整流性(SR > 10⁶)と非線形性(NL > 10⁶)が得られ、外部セレクタを必要とせず、クロストークを防止する。
- 固有のダイナミクス: デバイスは、能動的忘却をエミュレートする調整可能なコンダクタンス減衰を示す。減衰率は刺激依存的であり、プログラム可能な保持比率を可能にする。また、デバイスはSTDP、PPH、PPDといった不可欠なシナプス機能をサポートしている。
- システムアーキテクチャ: システムは5つの並列なシナプス・プロセッシング・コア(SPC)を統合している。各SPCは、パルス振幅と周波数を介して、異なる調整可能な重み更新ダイナミクス(可塑性と安定性の勾配)を持つように構成されている。SPC1は高い可塑性(急速な適応、忘却率 ~92%)に設定され、SPC5は高い安定性(知識保持、忘却率 ~37%)に設定されている。
- 協調設計戦略:
システムは、すべての5つのコアが並列にインカミング・タスクを処理する分散型プロセッシング・アーキテクチャを通じて動作する。可塑性と安定性のプロファイルが現在のタスクに最も適合するコアが、支配的なアトラクターとなる。ハードウェア互換のゲインパラメータ (β) が、アルゴリズムの規則とデバイス物理を橋渡しし、パルスパラメータを調整することで、ニューロモジュレーションとスパースゲーティングを実現する。
主な貢献
- 相乗的な生物学的統合: 本研究は、ショウジョウバエ由来の能動的忘却とヒト由来のメタ可塑性を単一の統一された学習モデルへと統合することで、孤立した生物学的メカニズムを超越している。
- ハードウェアネイティブなエミュレーション: SSMデバイスは、安定性と可塑性の連続体を物理的に具現化する。デバイス固有のコンダクタンス減衰と自己整流特性が、ソフトウェアベースの正則化や外部セレクタに頼ることなく、学習規則(忘れることと固定化)を直接実行する。
- スケーラブルなアーキテクチャ: 高い自己選択性を持つSSMがスニークパス電流を抑制することで、1/3Vスキームの下で最大17.1 Tbに達するテラビット規模のストレージ容量を実現している。
- 勾配的な可塑性・安定性連続体: 5つのコア・アーキテクチャは、急速な再構成可能性から長期保持に至るハードウェアネイティブな連続体を作成し、明示的なコア選択ロジックなしでの創発的な協調的タスク分割を可能にする。
結果
システムは、ハードウェアで校正されたシミュレーション(タイルレベルのハードウェアテストによって検証済み)を用い、20個の連続的なCIFAR派生タスク(S-CIFAR-100, R-CIFAR-100, R-CIFAR-10/100)を用いて評価された。
- 性能: システムは平均テスト精度68.5%を達成し、これはCPU(70%)およびGPU(70.3%)の実装に匹敵する数値であり、DAF単独のモデルを大幅に上回った。
- 効率性: 従来のCMOSアクセラレータと比較して、SSMベースのシステムは3.42倍の高速化と91倍のエネルギー効率向上を示した。
- 安定性と可塑性のバランス: システムは前方転移(FT)と後方転移(BWT)のバランスを成功裏に保った。最適な能動的忘却率 (α) の範囲は約120%であることが判明した。これより低い値では転移が不十分となり、高い値では壊滅的な性能低下を招いた。
- 堅牢性: デバイス間(D2D)およびサイクル間(C2C)のばらつきにもかかわらず、性能を維持した。これは、堅牢な製造(LRSにおけるD2Dのばらつきは2.91%)によって最小化されている。
- ハードウェア検証: 1-kbアレイにおける読み出しエラーは±0.06%以内に抑えられ、タイルレベルの相対誤差は0.08%であり、高い計算忠実度が確認された。
意義
本論文は、生涯学習を効率的に展開するための、ハードウェアとアルゴリズムが統合されたスケーラブルな基盤を確立したと主張している。メンリスタの物理特性に直接埋め込まれたマルチメカニズムの相乗効果を通じて、安定性と可塑性のトレードオフを解決することで、本研究は複雑な生物学的着想アルゴリズムと実用的なハードウェア実装との間の長年の隔たりを埋めている。これは、ニューロモーフィックAIが、単なる受動的なメモリ基質を超えて、動的な環境において継続的かつリソース効率の高い適応が可能な、能動的で適応的なコンピューティング・プラットフォームへと進化できることを示している。著者らは、この協調設計パラダイムを、従来のフォン・ノイマン型アーキテクチャのようなエネルギーやレイテンシのペナルティを負うことなく、オープンエンドな現実世界のシナリオで効果的に動作できる自律システムへの重要なステップとして位置づけている。
毎週最高の other 論文をお届け。
スタンフォード、ケンブリッジ、フランス科学アカデミーの研究者に信頼されています。
受信トレイを確認して登録を完了してください。
問題が発生しました。もう一度お試しください。
スパムなし、いつでも解除可能。
週刊ダイジェスト — 最新の研究をわかりやすく。登録