Repurposing Pre-trained LLMs as High Fidelity Continuous Text Autoencoders
Not on the map. Why?
Abstract (arXiv)
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 the activations of an intermediate layer as a fixed-length latent bottleneck. LLMAE achieves this interface with structured attention masks and LoRA adaptation, leveraging the generative prior of the original LLM. Across two backbones (270M Gemma 3 and 0.5B Qwen2.5), LLMAE reconstructs sequences up to 1024 tokens with high accuracy, reproducing up to 97% of documents verbatim. We demonstrate downstream utility by training a lightweight diffusion model that generates detailed image captions directly in the frozen LLMAE latent space. 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. Code and models are available at https://github.com/arkanath/LLMAE.
Key facts. No citing papers in the map yet (data through 14 Oct 2026). Builds on 46 of its 54 references that are in the map. First posted on arXiv 23 Sep 2026.
Builds on
46 of 54 references.
References · 46
- 2020Denoising Diffusion Probabilistic Modelscited by 21,076
- 2021High-Resolution Image Synthesis with Latent Diffusion Modelscited by 18,863
- 2019Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformercited by 14,599
- 2019BART: Denoising Sequence-to-Sequence Pre-training for Natural Language Generation, Translation, and Comprehensioncited by 5,893
- 2019BERTScore: Evaluating Text Generation with BERTcited by 5,512
- 2024Qwen2.5 Technical Reportcited by 4,942
- 2023Improved Baselines with Visual Instruction Tuningcited by 4,418
- 2022Flamingo: a Visual Language Model for Few-Shot Learningcited by 4,386
- 2024Qwen2-VL: Enhancing Vision-Language Model's Perception of the World at Any Resolutioncited by 4,234
- 2022LAION-5B: An open large-scale dataset for training next generation image-text modelscited by 4,152
- 2021Prefix-Tuning: Optimizing Continuous Prompts for Generationcited by 3,915
- 2021The Power of Scale for Parameter-Efficient Prompt Tuningcited by 3,795
34 more
- 2024Gemini 1.5: Unlocking multimodal understanding across millions of tokens of contextcited by 3,224
- 2023Tree of Thoughts: Deliberate Problem Solving with Large Language Modelscited by 3,211
- 2023Mistral 7Bcited by 3,164
- 2023Sigmoid Loss for Language Image Pre-Trainingcited by 3,109
- 2019The Curious Case of Neural Text Degenerationcited by 2,916
- 2021CLIPScore: A Reference-free Evaluation Metric for Image Captioningcited by 2,424
- 2023Pythia: A Suite for Analyzing Large Language Models Across Training and Scalingcited by 1,786
- 2021Perceiver: General Perception with Iterative Attentioncited by 1,104
- 2022Diffusion-LM Improves Controllable Text Generationcited by 1,027
- 2023ShareGPT4V: Improving Large Multi-Modal Models with Better Captionscited by 1,005
- 2025Large Language Diffusion Modelscited by 912
- 2023Discrete Diffusion Modeling by Estimating the Ratios of the Data Distributioncited by 745
- 2024Simplified and Generalized Masked Diffusion for Discrete Datacited by 626
- 2022Efficiently Scaling Transformer Inferencecited by 539
- 2025Block Diffusion: Interpolating Between Autoregressive and Diffusion Language Modelscited by 409
- 2023Learning to Compress Prompts with Gist Tokenscited by 390
- 2025Layer by Layer: Uncovering Hidden Representations in Language Modelscited by 311
- 2025Diffusion Transformers with Representation Autoencoderscited by 295
- 2022Continuous diffusion for categorical datacited by 214
- 2023In-context Autoencoder for Context Compression in a Large Language Modelcited by 199
- 2024The Remarkable Robustness of LLMs: Stages of Inference?cited by 146
- 2023Understanding Catastrophic Forgetting in Language Models via Implicit Inferencecited by 144
- 2024DOCCI: Descriptions of Connected and Contrasting Imagescited by 132
- 2022Latent Diffusion for Language Generationcited by 115
- 2025Generalized Interpolating Discrete Diffusioncited by 70
- 2023PLANNER: Generating Diversified Paragraph via Latent Language Diffusion Modelcited by 58
- 2023Diffusion Models for Non-autoregressive Text Generation: A Surveycited by 44
- 2025CapArena: Benchmarking and Analyzing Detailed Image Captioning in the LLM Eracited by 41
- 2025Cosmos: Compressed and Smooth Latent Space for Text Diffusion Modelingcited by 25
- 2024TEncDM: Understanding the Properties of the Diffusion Model in the Space of Language Model Encodingscited by 22
- 2024Diffusion Guided Language Modelingcited by 15
- 2025Creation-MMBench: Assessing Context-Aware Creative Intelligence in MLLMcited by 14
- 2025Long Context In-Context Compression by Getting to the Gist of Gistingcited by 13
- 2025Vision-Language-Vision Auto-Encoder: Scalable Knowledge Distillation from Diffusion Modelscited by 4
Built on by
Not cited yet.
Similar, not cited
Close in wording, no citation either way
- 2026AURORA-LM: Autoencoding Unified Representation for Continuous-Latent Diffusion Language Modeling
- 2024LMFusion: Adapting Pretrained Language Models for Multimodal Generation
- 2025A Comprehensive Study of Decoder-Only LLMs for Text-to-Image Generation
- 2024Exploring the Role of Large Language Models in Prompt Encoding for Diffusion Models
- 2023Language Quantized AutoEncoders: Towards Unsupervised Text-Image Alignment
- 2024Multimodal Latent Language Modeling with Next-Token Diffusion
- 2025Unleashing the Potential of Large Language Models for Text-to-Image Generation through Autoregressive Representation Alignment
- 2026RepFusion: Leveraging Multimodal Priors for Denoising in Representation Space