arXiv Machine Learning By Vihan Lakshman, Peter Baile Chen, Zhuohan Gu, Michael Cafarella

CacheGraft: Enhancing Retrieval-Augmented Generation via Latent-Space Document Enrichment

Read the original on arXiv Machine Learning →

CacheGraft is a retrieval‑augmented generation system that enriches a small language model’s latent space by injecting fused key‑value cache representations of corpus passages, encoded offline by a larger model. The approach eliminates the need to run the larger model at inference time, and on five QA benchmarks it improves answer quality in 29 of 30 comparisons, with 17 statistically significant gains. These results demonstrate that latent‑space document enrichment can boost RAG performance, especially when only a small model can be deployed.

Machine-generated by The Flow from the publisher's headline and feed description — not written or checked by a human. The full article lives at arXiv Machine Learning.

arXiv Machine Learning
Oct 15

Latent Compaction with Test-Time Training

Latent Compaction is a test‑time training method that learns a compact, document‑specific key–value cache for large‑context language models. It initializes cache entries using reconstruction importance, teacher attention, and key‑space diversity, then fits attention outputs and log‑normalizers to the full cache before jointly optimizing the entire cache via end‑to‑end distillation. Experiments on RULER, LongBench, and Phonebook Lookup show significant accuracy gains over budget‑matched compression baselines, achieving 96.7% accuracy at 32× compression on RULER and transferring well across model scales and families.

By Yingfa Chen, Xingyu Shen, Chaojun Xiao, Xu Han
arXiv AI
Aug 24

Dual-Cache Latent Space Communication between Heterogeneous Language Models

The paper introduces XKV, a latent protocol that enables efficient communication between heterogeneous language models by translating a sharer's key‑value cache into a receiver's context. XKV overcomes limitations of prior methods by jointly pooling both caches, reconciling differing layer depths, and allowing each receiver position to retrieve its own residual in native KV geometry. Across 45 dataset‑model pairings, XKV outperforms previous protocols and text communication while using fewer parameters and achieving faster translation times.

By Jiyao Liu, Qi Zhang, Yaoyi Jia, Ziwen Kan, Song Wang