BudgetLeak: Membership Inference Attacks on RAG Systems via the Generation Budget Side Channel

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Li, Hao, He, Jiajun, Wang, Guangshuo, Feng, Dengguo, Li, Zheng, Zhang, Min
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866917081452118016
author Li, Hao
He, Jiajun
Wang, Guangshuo
Feng, Dengguo
Li, Zheng
Zhang, Min
author_facet Li, Hao
He, Jiajun
Wang, Guangshuo
Feng, Dengguo
Li, Zheng
Zhang, Min
contents Retrieval-Augmented Generation (RAG) enhances large language models by integrating external knowledge, but reliance on proprietary or sensitive corpora poses various data risks, including privacy leakage and unauthorized data usage. Membership inference attacks (MIAs) are a common technique to assess such risks, yet existing approaches underperform in RAG due to black-box constraints and the absence of strong membership signals. In this paper, we identify a previously unexplored side channel in RAG systems: the generation budget, which controls the maximum number of tokens allowed in a generated response. Varying this budget reveals observable behavioral patterns between member and non-member queries, as members gain quality more rapidly with larger budgets. Building on this insight, we propose BudgetLeak, a novel membership inference attack that probes responses under different budgets and analyzes metric evolution via sequence modeling or clustering. Extensive experiments across four datasets, three LLM generators, and two retrievers demonstrate that BudgetLeak consistently outperforms existing baselines, while maintaining high efficiency and practical viability. Our findings reveal a previously overlooked data risk in RAG systems and highlight the need for new defenses.
format Preprint
id arxiv_https___arxiv_org_abs_2511_12043
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle BudgetLeak: Membership Inference Attacks on RAG Systems via the Generation Budget Side Channel
Li, Hao
He, Jiajun
Wang, Guangshuo
Feng, Dengguo
Li, Zheng
Zhang, Min
Cryptography and Security
Retrieval-Augmented Generation (RAG) enhances large language models by integrating external knowledge, but reliance on proprietary or sensitive corpora poses various data risks, including privacy leakage and unauthorized data usage. Membership inference attacks (MIAs) are a common technique to assess such risks, yet existing approaches underperform in RAG due to black-box constraints and the absence of strong membership signals. In this paper, we identify a previously unexplored side channel in RAG systems: the generation budget, which controls the maximum number of tokens allowed in a generated response. Varying this budget reveals observable behavioral patterns between member and non-member queries, as members gain quality more rapidly with larger budgets. Building on this insight, we propose BudgetLeak, a novel membership inference attack that probes responses under different budgets and analyzes metric evolution via sequence modeling or clustering. Extensive experiments across four datasets, three LLM generators, and two retrievers demonstrate that BudgetLeak consistently outperforms existing baselines, while maintaining high efficiency and practical viability. Our findings reveal a previously overlooked data risk in RAG systems and highlight the need for new defenses.
title BudgetLeak: Membership Inference Attacks on RAG Systems via the Generation Budget Side Channel
topic Cryptography and Security
url https://arxiv.org/abs/2511.12043