表示
超・人工知能研究所 Super AI Research Lab トップへ超・人工知能研究所AI Research Lab / ぱそ太郎工事中

since
平成38年9月29日

← 検索結果に戻る

LLMが話題を切り替える前、内部表現に何が現れるか

Semantic Search Signatures

保存した記事を見るこのブラウザに保存します。無料会員で端末間の同期・フォルダ整理
原論文の図

Semantic Search Signaturesの原論文から掲載した図です。細部はクリックして拡大できます。

原論文の図の説明を読む

Figure 1: Semantic fluency and LLM J-lens activations. A. Left: Example Semantic Fluency Task (SFT) output. Cluster labels are given by colours and switches between clusters are indicated by vertical black lines in-between distinct clusters. Right: Uniform Manifold Approximation (UMAP) of Word2Vec embeddings of animal names. Colours indicate extended Troyer category designations. B. Illustration of LLM J-lens activations during SFT. Left neural network represents an LLM. Beige-highlighted layer represents residual stream representations of a layer under study by the J-lens. Beige circle containing concepts represents a set of concepts included within the J-space for that layer. Note that in our experiments we considered the J-space as the top-25 strongest J-lens activations. J-lens activations are indicated by numbers to the right of concepts. Items that have already been verbalised are indicated by strikethroughs. Higher-level abstract category-related labels (“water” and “pet”) are indicated in blue and red font, respectively. The in-category J-space supply is computed as the number of items (not including category-related labels) within the J-space that belong to the category from which the model is presently naming items but that have not been verbalised yet. In this example, as the LLM names water animals and depletes the in-category J-space supply for that category, it switches to naming animals within the pet category. Note that the J-lens activation for the abstract category-related “pet” label increases in anticipation of switching to that category (See Figure 3 for analysis).

出典:Semantic Search Signatures / arXiv ↗

概要

  • 動物名列挙で、カテゴリ切り替えの前に次語確率や中間層の活性が変化した。
  • 内部状態への介入で、特定カテゴリへの移行や切り替え頻度を変えられると著者らは報告した。
  • 結果は主に動物名の列挙課題に基づき、業務上の発想支援への効果は未検証。
補足:編集した模式図で仕組みを確認する

VISUAL EXPLAINER

図でつかむ、Semantic Search Signaturesの仕組み

左から右へ読む
  1. 01動物名の列挙指示
  2. 02中間層の活性を読む
  3. 03切り替えを分析
  4. 04列挙結果を比較

原論文の説明をもとにした模式図です。処理の細部は省略しています。

研究の背景

LLMに動物名を続けて挙げさせると、似た動物をまとめて出し、その後に別の種類へ移ることがある。意味流暢性課題とは、指定された分類に属する語を次々に挙げる課題で、このまとまりと切り替えを観察できる。

出力だけでは、切り替えが内部の情報探索と結び付くのか分からない。内部状態に予兆があれば、生成中に話題が移る仕組みを調べる手掛かりになる。ただし、その知見を一般の業務生成に使えるかは別途検証が要る。

手法

著者らはまず、LLMに動物名をコンマ区切りで列挙させ、隣り合う項目が共通のカテゴリを持たない場合を切り替えとした。J-lensは中間層の内部状態を語の活性へ対応付ける分析方法で、まだ出していない動物名が強く活性化しているかを調べる。上位25件に含まれる現在のカテゴリの未出項目が減ると、切り替えが増えるかを測った。

次に「水」などカテゴリに関係する語の活性が、そこへ移る前に上がるかを確認した。別の分析では、内部状態から切り替えを予測する方向を求めた。最後に、その方向を生成中の内部状態へ加える介入を行い、次に出る動物名のカテゴリや切り替え頻度が変わるかを調べた。

新規性

論文は、概念を列挙するLLMの出力を、関連項目を続ける局面と別のまとまりへ移る局面に分け、内部表現との対応を調べた。次の語の確率だけでなく、中間層に現れる項目や上位概念の活性から切り替えを予測する。

さらに、特定のカテゴリへ向かう方向と、カテゴリに依存しない切り替え方向を内部状態に加える介入を試した。観測上の相関に加えて、切り替え率を動かせるかまで検証した点が主張である。

従来手法との違い

基準となる見方は、生成された動物名の並びだけから「同じカテゴリが続いたか」を数える方法である。論文はその出力に加え、次語確率、中間層の活性、まだ挙げていない候補の数を調べる。

カテゴリ内の候補が減るほど切り替えやすいという関係は、カテゴリ全体に残る動物名の数だけを見る場合とも比較した。また、介入前の生成を基準に、内部状態へ方向ベクトルを加えた後の切り替えを観測する。これは一般的な検索エージェントの性能比較ではない。

実験結果

著者らは20億〜90億パラメータの五つの公開重みモデルで動物名列挙を実施し、モデルと層の組み合わせごとに100個の乱数種を用いた。全モデルで切り替え直前の次語確率が低く、線形回帰でχ²が32.39以上、p値が0.0001未満だったと報告する。

中間層の上位活性候補に残る同カテゴリの動物が減るほど切り替え確率が上がった。内部状態から切り替えを予測する分析では、Gemma-2-9Bの最大AUCが事前位置で0.72、切り替え位置で0.87だった。特定カテゴリへの介入では、同モデルの目標カテゴリへの切り替え確率が最大0.68と報告された。

応用の可能性

編集上の応用案 · 論文が実証した用途とは区別しています。

編集上の応用案:社内の企画担当者が「保守費用を減らす案」を列挙させる場面で、同じ種類の案が続く区間と新しい種類へ移る区間を記録する。入力は列挙の指示、出力は案の一覧とカテゴリ切り替え位置である。

得られた切り替え傾向を、企画案が一つの話題に偏っていないか点検する補助情報として使う。この論文が示した介入効果は動物名の列挙課題での結果であり、企画の質や多様性の改善を実証したものではない。

応用先の候補からNauhに相談できます。

導入時の検証

導入時の検証案 · 対象データでの再評価が必要です。

導入時の検証案:対象モデルに、あらかじめ分類できる項目を列挙させ、まず出力だけから重複率とカテゴリ切り替え率を測る。内部状態を読める環境なら、切り替え前後の活性を保存し、次語確率だけを使う基準と予測AUCを比較する。

方向ベクトルを加える試験は少数の固定課題で行い、切り替え率と目標カテゴリへの到達率を測る。同時に、分類外の項目、重複、意味の通らない出力が増えないかを確認してから用途を広げる。

限界と課題

確認できた本文の範囲では、主な実験対象は動物名を挙げる意味流暢性課題である。著者らは五つのモデルを使ったが、J-lensが事前計算されていないLlama-3.2-3Bでは、その手法を使う実験を行っていない。切り替えの定義も、動物カテゴリの付け方に依存する。

編集上の確認事項は、動物以外の課題でも同じ内部方向が使えるか、生成の多様性を増やしても無関係な語や重複が増えないかである。人間の情報探索と同じ仕組みで動く、という結論まで広げて読むべきではない。

応用先の候補
研究内容と応用案に基づく分類です。業界での導入実績を示すものではありません。

出典

元のタイトル・要旨を確認する(英語)

Signatures of semantic search in the activations of large language models

When recalling lists of concepts (e.g., animals) during the semantic fluency task (SFT), both humans and large language models (LLMs) organise their output into clusters of related items (e.g., sea animals) that are punctuated by strategic switches between clusters. In humans, this pattern can be explained by a semantic foraging process, whereby distinct neural and behavioural signatures accompany within-cluster production ("exploit") and between-cluster switching ("explore"). Whether LLMs likewise represent these two search regimes within their internal states is unknown. Here, we apply a range of mechanistic interpretability techniques to provide evidence for this. In Study 1, we use the Jacobian lens (J-lens), which maps intermediate-layer residual-stream representations to token-level activations, to show that concept-level activations predict switching. First, we find that switching coincides with low next-token activations. Moreover, the probability of switching rises as the set of strongest J-lens activations (the J-space) becomes depleted of items from the category currently being produced, analogous to explore-exploit decision-making during patch foraging. We then show that middle-layer J-lens activations of abstract category-related labels (e.g., "water") increase in anticipation of switching into that category. We confirm these representations to causally influence switching by deriving steering vectors that target category switching. In Study 2, we identify generic residual stream directions that are activated during and in anticipation of switching. By steering activations along these directions, we bias increased or decreased rates of switching. Our study extends the semantic foraging framework to artificial intelligences and provides evidence that LLMs maintain distinct representational signatures for exploration and exploitation as they verbalise conceptual information.

論文本体の取得範囲に基づく解説。AIが作成した未校閲の記事です。性能の数値は著者の評価条件に依存します。応用例と検証計画は編集上の提案です。解説更新:2026-09-29