Share your thoughts, 1 month free Claude Pro on usSee more
WorkDL logo mark

KVComm: Enabling Efficient LLM Communication through Selective KV Sharing

About

Large Language Models (LLMs) are increasingly deployed in multi-agent systems, where effective inter-model communication is crucial. Existing communication protocols either rely on natural language, incurring high inference costs and information loss, or on hidden states, which suffer from information concentration bias and inefficiency. To address these limitations, we propose KVComm, a novel communication framework that enables efficient communication between LLMs through selective sharing of KV pairs. KVComm leverages the rich information encoded in the KV pairs while avoiding the pitfalls of hidden states. We introduce a KV layer-wise selection strategy based on attention importance scores with a Gaussian prior to identify the most informative KV pairs for communication. Extensive experiments across diverse tasks and model pairs demonstrate that KVComm achieves comparable performance to the upper-bound method, which directly merges inputs to one model without any communication, while transmitting as few as 30\% of layers' KV pairs. Our study highlights the potential of KV pairs as an effective medium for inter-LLM communication, paving the way for scalable and efficient multi-agent systems.

Xiangyu Shi, Marco Chiesa, Gerald Q. Maguire Jr., Dejan Kostic• 2025

Related benchmarks

TaskDatasetResultRank
Question AnsweringMuSiQuest-E 2022 (Full)
F1 Score46
30
Question AnsweringHotpotQA-E 2018 (Full)
F1 Score71
30
Question AnsweringQASPER-E 2021 (Full)
F1 Score38
30
SummarizationSAMSum Full 2019
F1 Score35
30
Multi-agent Sequential CommunicationAgentLeak
Privacy Score29
20
ReasoningCountries
F1 Score57
19
Multi-agent privacy and utility evaluationAgentLeak Sequential
Privacy24
10
Multi-agent privacy and utility evaluationAgentLeak Hierarchical
Privacy22
10
Multi-agent Sequential CommunicationMAGPIE
Privacy Score46
10
Multi-agent latent communication privacy and utilityMAGPIE Hierarchical
Privacy39
5
Showing 10 of 18 rows

Other info

Follow for update