複素数Transformerの学習を改善する注意スコア
Complex-valued Phase-Coherent Transformers
この論文をやさしく読む
ひとことで言うと
複素数を使うTransformerで、注意の計算前にベクトルの長さをそろえ、スコアの大きさも適切に保つと学習が改善するという研究です。
何に役立つ?
複素信号だけでなく、長距離記憶や階層的推論にも複素数モデルを使う設計の参考になります。正規化とスコアの尺度を切り分けて確認できます。
この研究の面白いところ
複素数を使うこと自体よりも、照合スコアが何を測り、どの尺度になるかが重要だと示しています。Path-Xの結果では、学習可能パラメータ中の複素数の割合も示しています。
どこまで分かった?
報告は挙げられた課題・モデル規模・深さでの実験です。深さ20までの結果は、それ以上の深さを保証しません。Path-Xを解く初の本格的複素数ネットワークという位置付けは著者の主張です。
v1のアブストラクトに基づくAI解説。日本語訳とは別に、用途の解釈を含みます。
アブストラクトの日本語訳
複素数値Transformerは、生の複素内積にsoftmax注意を適用する方式を引き継いできた。もともと複素数で表される領域の外では、この標準形の性能は偶然水準付近にとどまり、それを改善する複素数注意方式はこれまで示されていなかった。本研究では、照合に使う値をスケーリングしたコサインスコアにする必要があることを示す。クエリとキーをL2正規化して、スコアが大きさを無視してコサイン類似度を読み取るようにし、そのスコアを1程度の尺度に保つ。 これにより、同じモデルが2種類の異なるゲートの下で4つの診断用課題を学習する。正規化がなければ、どちらのゲートでもListOpsとNeedleでは偶然水準にとどまり、他の2課題でも大幅に劣る。また、正規化したスコアでも尺度を小さくしすぎると失敗する。 こうして得られる位相整合型Transformer(PCT)群は、長距離記憶、位置に基づく検索、階層的推論、周波数領域の分類、物理的な複素信号の課題において、最も強い実数値の比較手法に匹敵するか、それを上回る。深さ20まで性能の劣化は見られず、パラメータ数を61倍の範囲で変化させると、損失は対数線形に減少する。この系列の一つである、複素数スクリーニングと位相整合的な再帰を組み合わせたモデルは、Path-Xを解く初の本格的な複素数値ニューラルネットワークである。学習可能なパラメータの91.6%が複素数値であり、S4では38.2%である。これらを、複素数値ニューラルネットワークでは従来見られなかった汎化の兆候として報告する。
v1の要旨から自動生成。本文の精読・人による確認は未実施。
- 初稿
- 2026-09-18(UTC)
- 最新改訂
- 2026-09-18 · v1
- 査読・掲載
- 査読状況未確認
更新履歴
- v1 2026-09-18 この版を読む
取得できた版を表示。版の更新は査読済みを意味しません。過去版の本文差分は未解析です。
原文の要旨
Complex-valued Transformers have inherited softmax attention over the raw complex inner product. Outside natively complex domains this standard form stays near chance, and no complex attention had been shown to correct it. We show that the match must be a scaled cosine score: L2-normalise queries and keys, so the score reads their cosine similarity and ignores their magnitudes, and hold that score at order-one scale. With this the same models train on four diagnostic tasks under two different gates; without the normalisation they stay at chance on ListOps and Needle under both gates and fall far below on the other two, and a normalised score placed at too small a scale fails as well. The resulting family of phase-coherent Transformers (\PCT) matches or exceeds the strongest real-valued baseline across long-range memory, positional retrieval, hierarchical reasoning, frequency-domain classification and physical complex signals; it shows no degradation up to depth 20; and its loss decreases log-linearly over a 61-fold range of parameters. A member of the family, complex screening combined with a phase-coherent recurrence, is the first genuinely complex-valued neural network to solve Path-X, with 91.6% of its trainable parameters complex-valued against 38.2% for S4. We record these as signs of generalisation not previously seen in complex-valued neural networks.
arXiv ID: 2609.22415 / 要約の誤りについて