本文にスキップ
AI News HubLIVE
サイト内リライト2 分で読了

層をスキップするか、ループさせるか:LLMにおける層プログラムの学習

記事の要約

本論文はprogram-of-layers(PoLar)という概念を提唱する。事前学習済みLLMの層をモジュールとして扱い、入力ごとに動的にスキップやループを行うことで、多くの場合より少ない層数で同等以上の精度を達成し、元モデルの誤予測を訂正できることを示した。軽量なPoLar予測ネットワークが実行プログラムを生成し、数学推論ベンチマークで標準推論や既存の動的深度手法を一貫して上回り、分布外評価でも利得が持続する。

ソースarXiv Machine Learning著者: Ziyue Li, Yang Li, Tianyi Zhou
層をスキップするか、ループさせるか:LLMにおける層プログラムの学習
誤りを報告

訂正窓口はまだ利用できません。記事情報をコピーして保存できます。

訂正案内
本文へ

大規模言語モデル(LLM)は通常、固定された深さと順序で全層を一度だけ非再帰的に実行する前向き計算によって推論を行う。しかし、ICML 2026 に採択された論文「Skip a Layer or Loop It? Learning Program-of-Layers in LLMs」は、訓練を必要としない柔軟かつ動的な「program-of-layers(PoLar)」が広く存在することを明らかにした。本論文は Ziyue Li ら3名の著者によるもので、2026年6月4日に arXiv(番号 2606.06574、cs.LG カテゴリ)へ投稿され、先行研究 arXiv:2507.07996 を大幅に拡張した内容となっている。コードへのリンクも論文内で示されている。

研究では、事前学習済みの層をモジュールとしてまとめ、各入力に対して層をスキップしたりループ(繰り返し)したりすることで、カスタマイズされた実行プログラムを構成できることが示された。つまり、推論は必ずしも全層を順に通る必要はない。実験的な観察によれば、多くの入力に対しては、大幅に短いプログラム実行でも元のモデルと同等以上の精度が得られる。さらに、元の LLM が誤った予測を出した場合でも、より少ない層からなる代替プログラムによってその誤りを訂正できることが多い。これらの結果は、推論において標準的な前向き伝播以外にも複数の有効な潜在計算経路が存在することを示唆している。

PoLar を実用的に効率よく実現するため、著者らは軽量な PoLar 予測ネットワークを提案した。このネットワークは、各入力に対して実行プログラムを生成することを学習し、どの事前学習層をスキップし、あるいは繰り返すかを動的に決定する。人手による設計や固定ルールに依存する従来の動的深さ手法とは異なり、データ駆動型でサンプルごとの層実行計画を出力する点が特徴である。

数学的推論のベンチマークにおける実験では、PoLar は標準的な推論や従来の動的深さ手法と比較して一貫して精度を向上させ、しかも実行する層数が少ない場合が多いことが示された。これらの利点は分布外(out-of-distribution)評価においても維持された。

論文の結論は、固定深度の実行では LLM の潜在的な推論能力のごく一部しか捉えられていないというものである。層の実行プログラムを再編成することで、モデルの重みを再訓練することなく、より多くの推論能力を引き出しつつ推論コストを抑えられる可能性がある。なお、投稿履歴として、v1 は 2026年6月4日 17:59:58 UTC に提出され、全文は約 574 KB で、PDF、実験的な HTML、TeX ソースなどの形式で提供されている。

要点と分析を開く

記事インテリジェンス

投資家上級

要点

  • 学習不要で層を柔軟にスキップ・ループでき、多くの入力では短いプログラムが同等以上の精度を達成する。
  • 軽量なPoLar予測ネットワークが、事前学習済み層をスキップまたは繰り返す入力ごとの実行プログラムを学習する。
  • 数学推論ベンチマークで精度が一貫して向上し、しばしばより少ない層数で実行でき、分布外評価でも利得が続く。
  • 固定深度の実行は、LLMの潜在的な推論能力のごく一部しか捉えていないことを示唆する。

要点と分析は自動生成され、誤りを含む場合があります。原典をご確認ください。