Stop Tracking Me! Proactive Defense Against Attribute Inference Attack in LLMs

Dong Yan, Jian Liang, Ran He, Tieniu Tan
2/12/2026
cs.CRcs.AIcs.CL

Official code: Jasper-Yan/TRACE-RPS (19 stars, MIT, last commit 2026-04-06).

Abstract

Recent studies have shown that large language models (LLMs) can infer private user attributes (e.g., age, location, gender) from user-generated text shared online, enabling rapid and large-scale privacy breaches. Existing anonymization-based defenses are coarse-grained, lacking word-level precision in anonymizing privacy-leaking elements. Moreover, they are inherently limited as altering user text to hide sensitive cues still allows attribute inference to occur through models' reasoning capabilities. To address these limitations, we propose a unified defense framework that combines fine-grained anonymization (TRACE) with inference-preventing optimization (RPS). TRACE leverages attention mechanisms and inference chain generation to identify and anonymize privacy-leaking textual elements, while RPS employs a lightweight two-stage optimization strategy to induce model rejection behaviors, thereby preventing attribute inference. Evaluations across diverse LLMs show that TRACE-RPS reduces attribute inference accuracy from around 50\% to below 5\% on open-source models. In addition, our approach offers strong cross-model generalization, prompt-variation robustness, and utility-privacy tradeoffs. Our code is available at https://github.com/Jasper-Yan/TRACE-RPS.

View on arXivView PDF

Code Implementations(1)

MIT

[ICLR'26] Official Repository for The Paper: Stop Tracking Me! Proactive Defense Against Attribute Inference Attack in LLMs

190Shell, PythonFeb 11, 20266 months agoMIT

Similar Papers

Cite this paper

@article{yan2026stop,
  title  = {Stop Tracking Me! Proactive Defense Against Attribute Inference Attack in LLMs},
  author = {Dong Yan and Jian Liang and Ran He and Tieniu Tan},
  year   = {2026},
  eprint = {2602.11528},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2602.11528v1}
}

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/2602.11528.svg)](https://codeofpaper.com/paper/2602.11528)

Discussion