Need to Know: Contextual-Integrity-Grounded Query Rewriting for Privacy-Conscious LLM Delegation
Not on the map. Why?
Abstract (arXiv)
As LLMs become increasingly woven into everyday workflows, user queries sent to cloud hosted LLMs routinely mix task-essential content with task non-essential sensitive disclosures, yet type based PII redaction is context agnostic and may raise two issues: over disclosing untyped sensitive context and over removing answer bearing spans. We recast privacy preserving query rewriting under Contextual Integrity: a span should be forwarded only if it is necessary for the task. We introduce DelegateCI-Bench, the first task based Contextual Integrity benchmark for privacy-conscious delegation, comprising 3,167 samples that combine high quality synthetic data spanning 11 tasks and 20 task types, WildChat based real user queries, and a medical challenge set with dense sensitive information. Building on this benchmark, we propose a CI-guided reinforcement learning framework that converts essential and non-essential sensitive spans into verifiable optimization signals, and train a query rewriter to preserve task critical information while suppressing unnecessary sensitive disclosure. Experiments show that our learned rewriter achieves the best privacy-utility tradeoff, achieving up to +10.1 average utility over on-device baselines.
Key facts. Cited by 3 papers in the map (data through 7 Oct 2026). Builds on 18 of its 31 references that are in the map. First posted on arXiv 2 Jun 2026.
Builds on
18 of 31 references.
References · 18
- 2017Proximal Policy Optimization Algorithmscited by 14,441
- 2019RoBERTa: A Robustly Optimized BERT Pretraining Approachcited by 13,266
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
- 2023Efficient Memory Management for Large Language Model Serving with PagedAttentioncited by 5,767
- 2019BERTScore: Evaluating Text Generation with BERTcited by 5,462
- 2023FlashAttention-2: Faster Attention with Better Parallelism and Work Partitioningcited by 2,551
- 2024HybridFlow: A Flexible and Efficient RLHF Frameworkcited by 1,980
- 2017Ray: A Distributed Framework for Emerging AI Applicationscited by 516
- 2023Analyzing Leakage of Personally Identifiable Information in Language Modelscited by 278
- 2024HuatuoGPT-o1, Towards Medical Complex Reasoning with LLMscited by 257
- 2023Can LLMs Keep a Secret? Testing Privacy Implications of Language Models via Contextual Integrity Theorycited by 218
- 2024PrivacyLens: Evaluating Privacy Norm Awareness of Language Models in Actioncited by 161
6 more
- 2022The Text Anonymization Benchmark (TAB): A Dedicated Corpus and Evaluation Framework for Text Anonymizationcited by 94
- 2023Privacy-Preserving In-Context Learning with Differentially Private Few-Shot Generationcited by 94
- 2023DP-OPT: Make Large Language Model Your Privacy-Preserving Prompt Engineercited by 56
- 2023Hide and Seek (HaS): A Lightweight Framework for Prompt Privacy Protectioncited by 37
- 2024CI-Bench: Benchmarking Contextual Integrity of AI Assistants on Synthetic Datacited by 34
- 2023Protecting User Privacy in Remote Conversational Systems: A Privacy-Preserving framework based on text sanitizationcited by 23
Built on by
Citing papers per year: 2026: 3.
3 papers.