An Efficient and Privacy-Preserving Architecture for Cross-Institutional Collaborative RAG
About
Retrieval-Augmented Generation (RAG) empowers LLMs with external knowledge, making cross-institutional domain-specific knowledge base integration a highly promising deployment paradigm. Despite this potential, strict privacy regulations create severe "data silos" that obstruct such collaboration. Building federated RAG systems requires distributed inference, but the Transformer's self-attention mechanism fundamentally conflicts with this by mandating cross-node access to distributed Key-Value caches. To address this challenge, we present FedRAG, a high-throughput, privacy-preserving federated RAG framework. At its core is a novel Scrambled Distributed Attention protocol that utilizes numerically stable feature scrambling and token permutation. By dynamically delegating scrambled computations to collaborating nodes, our system successfully decouples attention execution from data localization without exposing plaintext. Crucially, our approach requires no specialized hardware or model retraining, circumventing the prohibitive latency and communication overheads of cryptographic solutions while robustly defending against intermediate state inversion attacks. Extensive evaluations demonstrate our framework preserves negligible (<0.1\%) model utility degradation and achieves up to a 62$\times$ latency reduction over existing secure baselines, sustaining practical, human-reading throughput for cross-institutional knowledge synergy.
Related benchmarks
| Task | Dataset | Result | Rank | |
|---|---|---|---|---|
| LLM Inference Efficiency | Short sequence prompts | TTFT (s)0.2 | 24 | |
| LLM Inference Efficiency | Sequence prompts Medium | TTFT (s)0.37 | 24 | |
| LLM Inference Efficiency | Long sequence prompts | TTFT (s)1.05 | 24 | |
| Question Answering | SQuAD | Accuracy93.8 | 12 | |
| Question Answering | MARCO Unfiltered | Accuracy18.77 | 12 | |
| Question Answering | MARCO | Accuracy24.23 | 12 | |
| Question Answering | HOTPOTQA Unfiltered | Accuracy65.64 | 12 | |
| Question Answering | MUSIQUE Unfiltered | Accuracy47.37 | 12 | |
| Retrieval-Augmented Generation | meta-llama Llama-3.1-8B-Instruct Short prompt | TTFT (s)3.23 | 3 | |
| Retrieval-Augmented Generation | Llama 3.1 8B Instruct Medium prompt | TTFT (s)4.83 | 3 |