paper-with-me

Papers

Memory Inception: Latent-Space KV Cache Manipulation for Steering LLMs

2026-05-07 · Andy Zeyi Liu, Michael Zhang, Ilana Greenberg, Adam Alnasser, Lucas Baker, John Sous arxiv

Steering large language models (LLMs) is usually done by either instruction prompting or activation steering. Prompting often gives strong control, but caches guidance tokens at every layer and can clutter long interactions; activation steering is compact but typically weaker and does not support large structured reminders. We introduce memory inception (MI), a training-free method that steers in latent attention space by inserting text-derived key-value (KV) banks only at selected layers. Rather than materializing reminder content throughout the prompt cache, MI treats steering as selective KV allocation, injecting latent slots only where the model routes to them. On matched personality-steering tasks, MI gives the best overall control--drift trade-off, remaining competitive with prompting while consistently outperforming CAA. On updateable guidance, MI supports mid-conversation behavior shifts without rewriting the visible transcript, achieving the highest post-shift alignment on Qwen3. On structured reasoning, MI outperforms visible prompting on HARDMath and PHYSICS (10/12 subject$\times$mode cells), serving as proxies for structured reasoning in verifiable domains, while cutting content-matched KV storage by up to 118$\times$. These results position MI as a powerful steering method when guidance is persistent, structured, or expensive to keep in the visible transcript.

📄 PDF Abstract BibTeX arXiv:2605.06225

Code (0)

등록된 구현이 없습니다.

Similar Papers 제목 키워드 기반

Make a Face: Towards Arbitrary High Fidelity Face Manipulation

2019-08-20 · ICCV 2019 10 · Shengju Qian, Kwan-Yee Lin, Wayne Wu, Yangxiaokang Liu 외

Recent studies have shown remarkable success in face manipulation task with the advance of GANs and VAEs paradigms, but the outputs are sometimes limited to low-resolution and lack of diversity. In this work, we propose …

ClusteringDisentanglementDiversityVocal Bursts Intensity Prediction

Multi-head Temporal Latent Attention

2025-05-19 · Keqi Deng, Philip C. Woodland

While Transformer self-attention offers strong parallelism, the Key-Value (KV) cache grows linearly with sequence length and becomes a bottleneck for inference efficiency. Multi-head latent attention was recently develop…

GPUspeech-recognitionSpeech RecognitionTranslation

EG-MLA: Embedding-Gated Multi-head Latent Attention for Scalable and Efficient LLMs

2025-09-20 · Zhengge Cai, Haowen Hou arxiv

Reducing the key-value (KV) cache size is a crucial step toward enabling efficient inference in large language models (LLMs), especially under latency and memory constraints. While Multi-Head Attention (MHA) offers stron…

GTA: Grouped-head latenT Attention

2025-06-15 · Luoyang Sun, Jiwen Jiang, Cheng Deng, Xinjian Wu 외

Attention mechanisms underpin the success of large language models (LLMs), yet their substantial computational and memory overhead poses challenges for optimizing efficiency and performance. A critical bottleneck arises …

Latent Spatial Memory for Video World Models

2026-06-08 · Weijie Wang, Haoyu Zhao, Yifan Yang, Feng Chen 외 arxiv

Video world models that maintain 3D spatial consistency across generated frames typically rely on explicit point cloud memory constructed in RGB space. This design is both computationally expensive, requiring repeated re…

Video Generation