概要
- 最先端LLM (GPT-6 Astra)はニューラルネットワークの重みを 人間や従来手法を超える精度で読解・編集 可能
- 生の重みだけで ネットワークの訓練目的や失敗要因を特定・修正 できる
- 環境からのフィードバックなし で新しいネットワークを一から作成し、既存手法より高性能を実現
- この手法を Weight-Space Programming と呼び、最適化や解釈性の新しい可能性を提示
- LLMが重みを直接操作することで、従来のシンボリック最適化手法からの大きな転換点となる
LLMはニューラルネットワークの重みを「読む」「書く」
- GPT-6 Astra は、ニューラルネットワークの重みを 生データとして直接解釈・編集 できる能力を持つ
- 例えば、生の重みだけから「このネットワークは何のゲームに訓練されたか」を 高精度で言い当てる ことが可能
- ゲームの種類や出力動作を 重みの分布や特徴から推測 する解析力を持つ
- Diffusionモデルや言語モデルの重みを解析し、 生成画像の形状や学習言語まで特定 できる
- Diffusionモデルでは 出力層の重み構造から図形のエッジ を推定
- 言語モデルでは 中間層だけから学習言語やAttention Headの役割 を推測
LLMによる重みの「修正」と「新規作成」
- Astraは 失敗しているネットワークの重みを直接修正 し、タスク成功率を劇的に向上させる
- 例:Qwen3.5-0.8Bの数学能力向上で、 重みの0.016%を編集するだけでMATH-500スコアが7.9%→48.1% に上昇
- 同時に命令追従能力(IFEval)も3倍に向上
- 新しいネットワークをゼロから設計・生成 することも可能
- ゲームプレイ用ネットワークを 環境フィードバックなしで作成し、従来手法を凌駕する性能 を発揮
Weight-Space Programmingという新たなパラダイム
- 従来はAdamやMuonのような シンボリックな最適化アルゴリズム で重み調整を行っていた
- Weight-Space Programmingでは、 LLMが自然言語仕様に基づいて直接重みを編集・生成
- 勾配降下法や明示的なトレーニングループ不要 で、自然言語から「この重みの問題を特定・修正」「新しいネットワーク作成」などが可能
- 推論駆動型の最適化パラダイム が、微分可能なニューラルネットワークにも拡張されつつある
実験概要
- 実験1 :4つのゲーム(Breakout, Pong, Pac-Man, Freeway)用ネットワークの重みをAstraに提示し、訓練ゲームを推測→全て正答
- 実験2 :Diffusionモデル(円・三角・四角)の重みだけから生成画像の形状を推定→全て正答
- 実験3 :言語モデル(Python, Romaji, English)の中間層だけから学習言語やAttention Headの役割を特定→全て正答
- 実験4 :物理制御タスクで失敗するネットワークの重みをAstraが直接修正→成功率0%→100%
- 実験5-6 :Astraがゼロからゲームプレイ用ネットワークを設計・生成→従来手法より高性能
今後の展望
- LLMが重み空間をネイティブに最適化・解釈 する時代の到来
- 人間や従来アルゴリズムでは難しい ネットワーク内部の解釈・修正・設計 が自然言語で容易に
- Weight-Space Programming が深層学習の最適化・可読性・汎用性を大きく変革する可能性
今はラボがRSIのためにモデルをトレーニングしてるから、ガードレールが緩ければオープンソースモデルの改善にも同じトレーニングが役立つと思うよ :) でも、その状況は長くは続かないだろうね。