Generation-Augmented Generation: A Plug-and-Play Framework for Private Knowledge Injection in Large Language Models

Rongji Li, Jian Xu, Xueqing Chen, Yisheng Yang, Jiayi Wang +4 more
1/13/2026
cs.CL

Official code: 360CVGroup/GAG (7 stars, last commit 2026-04-21).

Abstract

In domains such as biomedicine, materials, and finance, high-stakes deployment of large language models (LLMs) requires injecting private, domain-specific knowledge that is proprietary, fast-evolving, and under-represented in public pretraining. However, the two dominant paradigms for private knowledge injection each have pronounced drawbacks: fine-tuning is expensive to iterate, and continual updates risk catastrophic forgetting and general-capability regression; retrieval-augmented generation (RAG) keeps the base model intact but is brittle in specialized private corpora due to chunk-induced evidence fragmentation, retrieval drift, and long-context pressure that yields query-dependent prompt inflation. Inspired by how multimodal LLMs align heterogeneous modalities into a shared semantic space, we propose Generation-Augmented Generation (GAG), which treats private expertise as an additional expert modality and injects it via a compact, representation-level interface aligned to the frozen base model, avoiding prompt-time evidence serialization while enabling plug-and-play specialization and scalable multi-domain composition with reliable selective activation. Across two private scientific QA benchmarks (immunology adjuvant and catalytic materials) and mixed-domain evaluations, GAG improves specialist performance over strong RAG baselines by 15.34% and 14.86% on the two benchmarks, respectively, while maintaining performance on six open general benchmarks and enabling near-oracle selective activation for scalable multi-domain deployment. Code is publicly available at https://github.com/360CVGroup/GAG.

View on arXivView PDF

Code Implementations(1)

360CVGroup/GAGOfficial100%

Generation Augmented Generation, resolving fragmented knowledge defect of RAG

70Shell, PythonJan 7, 20266 months ago

Similar Papers

Cite this paper

@article{li2026generationaugmented,
  title  = {Generation-Augmented Generation: A Plug-and-Play Framework for Private Knowledge Injection in Large Language Models},
  author = {Rongji Li and Jian Xu and Xueqing Chen and Yisheng Yang and Jiayi Wang and Xingyu Chen and Chunyu Xie and Dawei Leng and Xu-Yao Zhang},
  year   = {2026},
  eprint = {2601.08209},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2601.08209v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2601.08209.svg)](https://codeofpaper.com/paper/2601.08209)

Discussion