arXiv論文メモ
新着一覧
cs.LG · 査読状況未確認

事前学習した言語モデルを連続表現の文章オートエンコーダーへ

Repurposing Pre-trained LLMs as High Fidelity Continuous Text Autoencoders

Arkanath Pathak, Unnat Jain, Alexander C. Berg

この論文をやさしく読む

ひとことで言うと

事前学習済み言語モデルの内部に固定長の連続表現を作り、文章を高精度で復元する方法。

何に役立つ?

文章を連続潜在空間で扱う生成モデルや、画像説明文の生成などへの応用を考える材料になる。

この研究の面白いところ

デコーダー専用モデルを再利用し、最大1,024トークンの文章でほぼ完全な復元を報告した。

どこまで分かった?

実装は2億7千万パラメータのGemma 3で、下流の例は詳しい画像説明文の生成である。

v1のアブストラクトに基づくAI解説。日本語訳とは別に、用途の解釈を含みます。

アブストラクトの日本語訳

次のトークンを予測する学習によって、流暢な自己回帰言語モデルが可能になったが、文章全体の構造は、順序に沿った分解を通じて間接的にしか表現されない。対照的に、高精度なオートエンコーダーは画像生成の標準的な部品となり、連続的な潜在空間で生成モデルを動かせるが、文章には同程度に忠実な連続表現がない。本研究は、事前学習済みのデコーダー専用言語モデルの内部活性に、長さを固定した中間の潜在表現の細い通路を設け、連続表現の文章オートエンコーダーとして再利用する方法LLMAEを提案する。パラメータ効率のよい2億7千万パラメータのGemma 3モデルで実装し、構造化した注意マスク、LoRAによる適応、KL正則化を用いて、元の言語モデルの生成に関する事前知識を生かす符号化・復号の仕組みを学ぶ。最大1,024トークンの文章系列を復元するよう学習し、この課題で大きく改善して、ほぼ完全な復元を達成した。さらに、学んだLLMAEを使い、詳しい画像説明文を作る潜在空間の文章拡散モデルを学習させて、この表現の後続課題での有用性を示した。文章を固定長の連続潜在空間へ写すことで、元の言語モデルの流暢さを生かしながら、後続課題への適応に有効な基盤を提供する。

v1の要旨から自動生成。本文の精読・人による確認は未実施。

初稿
2026-09-23(UTC)
最新改訂
2026-09-23 · v1
査読・掲載
査読状況未確認
arXivで読むPDF

更新履歴

取得できた版を表示。版の更新は査読済みを意味しません。過去版の本文差分は未解析です。

原文の要旨

Next-token prediction has enabled highly fluent autoregressive language models, but it represents global structure only indirectly through sequential factorization. In contrast, high-fidelity autoencoders have become a standard primitive in image generation, enabling generative models to operate over continuous latent spaces; text lacks a comparably faithful continuous representation. We propose LLMAE, a method for repurposing a pretrained decoder-only language model as a continuous text autoencoder by exposing an intermediate fixed-length latent bottleneck within its internal activations. Instantiated with a parameter-efficient 270M Gemma 3 model, LLMAE uses structured attention masks, LoRA adaptation, and KL regularization to learn an autoencoding interface that leverages the generative prior of the original LLM. We train LLMAE to reconstruct text sequences up to 1024 tokens, significantly improving on this task to achieve near-perfect reconstruction. Furthermore, we demonstrate the downstream utility of this representation by training a latent text diffusion model for detailed image captioning using the learned LLMAE autoencoder. By mapping text into a fixed-length continuous latent space, our approach provides an effective substrate for downstream adaptation while benefiting from the fluency of the original LLM.

arXiv ID: 2609.27248 / 要約の誤りについて