論文解説: Your Transformer Can Hold Two Thoughts at Once — 2本の文章を混ぜてもLLMは両方の続きを覚えている
無関係な2本の文章の埋め込みを平均して1回だけ通すと、LLMの出力には両方の次トークン候補が残る。この『重ね合わせ線形性』が学習でむしろ弱まること、軽い微調整で戻ること、2本の続きを分けて生成する試みまでを論文本文に沿って解説する。
要点
- 2026年9月24日公開の論文は、別々の文章からの入力を線形に混ぜると出力が各次トークン分布の重ね合わせになると主張する。
- 著者らはこれを「重ね合わせ線形性仮説」と呼び、学習で得る性質ではなくTransformerの構造そのものに備わると述べる。
- この性質は事前学習が進むと弱まるが、軽い微調整でかなり取り戻せ、混合入力の出力と2分布の平均とのずれが縮む。
- 実験では長さの等しい2文章の埋め込みを同じ位置のトークンごとに平均し、LLMに1回通して次トークン分布を調べる。
- 目標分布は2つの予測の平均で、実際の出力とのずれを無関係な2文章の予測どうしのずれで割った比率で測る。
Your Transformer Can Hold Two Thoughts at Once: Evidence of Linear Superposition in LLMs
一次資料 — この記事の根拠
論文の発表 2026-09-24→この解説の公開 2026-09-26同月
Your Transformer Can Hold Two Thoughts at Once: Evidence of Linear Superposition in LLMsPavel Tikhonov, Anton Korznikov, Matvey Mikhalchuk ほか · 2026-09-24 · v1arXiv:2609.29845論文ページ·PDF原文の要旨(Abstract)を読む
While Large Language Models (LLMs) rely on highly non-linear components, in this work we demonstrate that they exhibit fundamental linearity: when inputs from distinct text streams are linearly combined, the model outputs a superposition of the individual next-token distributions. We term this the \textit{Superposition Linearity Hypothesis}. We provide evidence that superposition is an intrinsic property of the Transformer architecture rather than an emergent consequence of training; in fact, we observe that it tends to diminish as pretraining progresses. However, we demonstrate that linearity can be substantially restored through lightweight fine-tuning, significantly reducing the divergence between the predicted next-token distribution and the average of the individual next-token distributions. Finally, we introduce a guided decoding procedure that disentangles superposed outputs, enabling the simultaneous generation of two coherent continuations from a single forward pass.
この論文は何を言っているのか
今回読むのは、2026年9月24日にarXivに公開された "Your Transformer Can Hold Two Thoughts at Once: Evidence of Linear Superposition in LLMs"(Tikhonovら、arXiv:2609.29845)です。
要旨を日本語でまとめるとこうなります。LLMは非線形な部品の塊なのに、根っこのところで線形にふるまう。別々の文章から来た入力を線形に混ぜると、出力はそれぞれの次トークン分布を重ね合わせたものになる。著者らはこれを「重ね合わせ線形性仮説(Superposition Linearity Hypothesis)」と呼ぶ。この性質は学習で身につくものではなくTransformerという構造そのものに備わっていて、事前学習が進むとむしろ弱まる。ただし軽い微調整でかなり取り戻せて、混ぜた入力の出力分布と「2つの分布の平均」とのずれは大きく縮む。最後に、重なった出力をほどいて、1回の順伝播から2本のまともな続きを同時に生成するデコード法を提案する。
比喩: 2つの放送局が同じ周波数に乗ったラジオ
ラジオで2つの局が混信している場面を思い浮かべてください。ニュースと野球中継が同時に聞こえてきます。耳障りではあっても、慣れた人なら「今の単語はニュース側」「今の歓声は中継側」とかなり聞き分けられます。音は空気中で足し算されるだけなので、元の2つの信号は壊れずに重なっているからです。
問いは、LLMの中でも同じことが起きるかです。Transformerには注意のsoftmaxや活性化関数など、足し算を素直に通さない部品が積み重なっています。2局の放送が溶け合い、どちらでもない雑音になってもおかしくありません。
直感: 「平均した単語」を読ませる
実験の操作は驚くほど単純です。長さをそろえた2本の文章AとBを用意し、同じ位置のトークン同士で埋め込みベクトルを平均します。1トークン目はAとBの1トークン目の平均、という具合です。これを普通のLLMに1回だけ通し、次トークン分布を調べます。
埋め込みは単語を向きを持った数値の並びに変えたものです(埋め込み(Embedding)を1から理解する)。下の図で2本のベクトルの角度を変えると、向きが離れるほど内積と余弦が小さくなります。平均ベクトルは両方から少しずつ離れた「中間の向き」なので、モデルから見れば「どちらの単語とも言い切れない入力」が続くことになります。
仕組み: 3つの式で問いを定義する
著者らはまず混ぜ方を式にします(§2.1)。
はトークンを埋め込みベクトルに変える関数、 と は2本の文章の 番目のトークン、 はモデルに実際に渡す混合入力です。言い換えれば「同じ位置の2単語のベクトルを足して2で割る」だけです。
理想的に線形なら、出力は次の分布に近づくはずです(§2.3)。
は文章Aだけを入れたときの次トークン分布です。つまり「Aだけ入れた場合とBだけ入れた場合の予測を、半々で混ぜたもの」が目標になります。
ずれを測るため、著者らは比率を定義しています。
はKLダイバージェンスなど分布どうしの距離、 は混ぜた入力に対する実際の出力です。分子は「目標と実際のずれ」、分母は「無関係な2文章の予測どうしのずれ」です。つまり1より十分小さければ、混ぜた出力は無関係な分布よりずっと目標に近いと言えます。KLダイバージェンス自体はAIのための情報理論で扱っています。
コメント
コメントにはログインが必要です