アーカイブ / Computer Vision
格子の変換規則を反復して推定するLoop-ViT：ARC-AGIでの精度と停止判定
Loop-ViT
Loop-ViTの原論文から掲載した図です。細部はクリックして拡大できます。
原論文の図の説明を読む
Figure 1 : Acc-Params comparisons with recurrent and vision methods. The vertical axis is accuracy (ARC-AGI-1); the horizontal axis is Parameters (memory cost). Our Loop-ViT outperforms previous methods while requiring significantly cheaper Params.
概要
- Loop-ViTは共有した画像Transformerの中核ブロックを反復し、格子の変換予測を更新する。
- 著者らは18MパラメータのLargeについて、ARC-AGI-1のPass@2を65.8%と報告した。
- 予測の不確実性による早期停止を備えるが、実務の配置表への適用は別途検証が必要である。
補足：編集した模式図で仕組みを確認する
VISUAL EXPLAINER
図でつかむ、Loop-ViTの仕組み
- 01作例と対象格子
- 02格子を内部表現へ
- 03共有ブロックで反復
- 04不確実性で停止判定
- 05予測格子
原論文の説明をもとにした模式図です。処理の細部は省略しています。
研究の背景
ARC-AGIは、少数の「入力格子・正しい出力格子」の作例から変換規則を読み取り、別の格子に適用する課題である。線を延ばす局所操作から物体の移動まで課題が異なり、必要な処理の回数も一様ではない。
通常の画像Transformerは、入力を決まった層に一度通す。著者らは、この固定された計算経路では規則を段階的に適用する処理を表しにくいと考え、同じ中核部分を繰り返し使う画像モデルを提案した。
手法
Loop-ViTは作例と対象格子を内部表現に変え、同じ重みの中核ブロックを繰り返し適用する。重みの共有とは、反復のたびに別の層を用意せず、同じ計算規則を再利用することを指す。例えば「色付きの物体を移す」作例が与えられた場合、対象格子の表現と出力予測を段階的に更新する仕組みである。
ブロック内の畳み込みは近くのマスを調べ、画像部分にだけ適用される。注意機構は離れた位置や課題の情報を参照する。推論時は各マスの色の予測確率からエントロピーという不確実性の値を計算し、その平均が0.05未満なら停止する。達しなければ設定された最大回数まで反復し、最後の状態から格子を予測する。
新規性
Loop-ViTは、画像を扱うTransformerの中核部分に同じ重みを繰り返し使う。反復回数を増やしても、その中核部分のパラメータ数は増えない。局所的な畳み込みと格子全体を参照する注意機構を組み合わせた点も設計上の特徴である。
著者らはさらに、出力の予測確率から不確実性を測り、低くなれば反復を止めるDynamic Exitを導入した。停止判定用の追加パラメータは不要だと説明している。
従来手法との違い
比較対象のVARCは、格子を画像として扱い、異なる層を積んだモデルを一度通して出力する。Loop-ViTは画像としての入力形式を保ちながら、共有した中核部分を繰り返し通して内部表現と予測を更新する。
著者らの表では、ARC-AGI-1のPass@2は18MパラメータのVARCが54.5%、73MのVARCアンサンブルが60.4%、Loop-ViTの3.8M版が60.1%、11.2M版が63.8%だった。モデル構造以外の条件もあるため、この差を反復だけの効果とは判断できない。
実験結果
著者らはARCと生成データRE-ARCで事前学習し、評価時には各課題の作例を回転・反転・色の置換で増やして重みを調整した。指標は2回の試行で正解する割合を表すPass@2で、ARC-AGI-1ではLoop-ViTのSmall、Medium、Largeが順に60.1%、63.8%、65.8%と報告されている。
同じ表でLargeは18Mパラメータ、ARC-AGI-2では14.2%だった。73MパラメータのVARCアンサンブルはARC-AGI-1で60.4%、ARC-AGI-2で11.1%と記載される。著者らはARC-AGI-1で動的停止が固定回数の反復より高精度かつ平均計算量が少ないと述べるが、確認できた本文にはその差の数値が示されていない。
応用の可能性
編集上の応用案 · 論文が実証した用途とは区別しています。
編集上の応用案：製造現場で使う色付きのデジタル配置表について、変更規則に沿った次の配置を提案する。規則の作例として「変更前・正しい変更後」の組を入力し、新しい変更前の格子に対する変更後の候補を出力する。人が候補を確認して採否を決める運用を想定する。
必要なのは、元の配置表、作例、正しい変更後の格子を示すラベルである。デジタル格子を直接使うならセンサーは不要。論文と同じ学習条件を試す場合はARCとRE-ARCによる事前学習、案件ごとの作例による調整が必要で、実務への転用効果は未検証である。
応用先の候補からNauhに相談できます。
導入時の検証
導入時の検証案 · 対象データでの再評価が必要です。
導入時の検証案：過去の配置表から、規則を示す作例と、評価用の変更前・正しい変更後の格子を案件単位で分けて用意する。正しい変更後をラベルにし、デジタル格子を直接使う場合はセンサーを使わない。画像から読み取る場合は、読み取り誤差を別に測る。
論文の条件に沿う試験ではARCとRE-ARCによる事前学習、案件ごとの作例を使う評価時の調整を含める。人が規則を適用する現行手順を基準に、未使用案件で格子全体の正答率、誤った変更の件数、調整時間、反復回数を測る。早期停止の有無は同じ案件で比較する。
限界と課題
確認できた本文の範囲では、著者らは実務への適用範囲や代表的な失敗例を明確な限界として整理していない。示された精度はARC-AGIの格子課題に対するものであり、製造現場の配置表でも同じ精度になることは示していない。
編集上の検証事項は、案件ごとの調整に要する時間、誤答時にも予測の不確実性が低くなるか、早期停止によって誤答が増えないかである。比較表の手法間で学習データや評価時の調整条件がどこまで揃っているかも、数値を実務判断に使う前に確認する必要がある。
出典
元のタイトル・要旨を確認する（英語）
LoopViT: Scaling Visual ARC with Looped Transformers
Recent advances in visual reasoning have leveraged vision transformers to tackle the ARC-AGI benchmark. However, we argue that the feed-forward architecture, where computational depth is strictly bound to parameter size, falls short of capturing the iterative, algorithmic nature of human induction. In this work, we propose a recursive architecture called Loop-ViT, which decouples reasoning depth from model capacity through weight-tied recurrence. Loop-ViT iterates a weight-tied Hybrid Block, combining local convolutions and global attention, to form a latent chain of thought. Crucially, we introduce a parameter-free Dynamic Exit mechanism based on predictive entropy: the model halts inference when its internal state ``crystallizes" into a low-uncertainty attractor. Empirical results on the ARC-AGI-1 benchmark validate this perspective: our 18M model achieves 65.8% accuracy, outperforming massive 73M-parameter ensembles. These findings demonstrate that adaptive iterative computation offers a far more efficient scaling axis for visual reasoning than simply increasing network width. The code is available at https://github.com/WenjieShu/LoopViT.
論文本体の取得範囲に基づく解説。AIが作成した未校閲の記事です。性能の数値は著者の評価条件に依存します。応用例と検証計画は編集上の提案です。解説更新：2026-10-01