🔁 「じっくり考えるほど賢くなる」AIを、勾配爆発を起こさずに学習させる方法が見つかりました。
タイトル: Thinking with Looped Flows
URL:
再帰的に隠れ状態を更新して「考える」ループモデルは、これまでBPTT(時間方向の逆伝播)の不安定さに悩まされてきました。Looped Flowsは拡散モデルの学習原理を借りることでこれを解決した手法です。注目ポイントを3つ紹介します。
🧠 BPTTなしで再帰計算を学習
ノイズレベルを徐々に減らしながら各ステップをローカルな損失で学習することで、勾配消失・爆発を起こさずに「考え続ける」状態を安定して学習できます。
⏱ 推論時に計算量を後から増やせる
学習時より細かい時間刻みで推論するだけで性能が伸び、Sudokuでは8ステップの74.5%から128ステップで97.9%まで向上します。
🏆 ARC-AGIで既存手法を上回る
ARC-AGI-1で58.8%、ARC-AGI-2で12.2%を達成し、いずれも従来のループモデル手法を上回りました。
推論時にじっくり計算するほど賢くなる、というテスト時スケーリングの新しい実現方法として興味深い研究です。
#
LLM推論# #
機械学習#