STAMP: Selective Task-Aware Mechanism for Text Privacy

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Tian, Fengwei, Bhattacharjee, Payel, Hanson, Heidi, Rubin, Geoffrey D., Lo, Joseph Y., Tandon, Ravi
Format: Preprint
Published: 2026
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866914388102873088
author Tian, Fengwei
Bhattacharjee, Payel
Hanson, Heidi
Rubin, Geoffrey D.
Lo, Joseph Y.
Tandon, Ravi
author_facet Tian, Fengwei
Bhattacharjee, Payel
Hanson, Heidi
Rubin, Geoffrey D.
Lo, Joseph Y.
Tandon, Ravi
contents We present STAMP (Selective Task-Aware Mechanism for Text Privacy), a new framework for task-aware text privatization that achieves an improved privacy-utility trade-off. STAMP selectively allocates privacy budgets across tokens by jointly considering (i) each token's importance to the downstream task (as measured via a task- or query-specific representation), and (ii) its privacy sensitivity (e.g., names, dates, identifiers). This token-level partitioning enables fine-grained, group-wise control over the level of noise applied to different parts of the input, balancing privacy protection with task relevance. To privatize individual token embeddings, we introduce the polar mechanism, which perturbs only the direction of embeddings on the unit sphere while preserving their magnitude. Decoding is performed via cosine nearest-neighbor search, aligning the perturbation geometry with the decoding geometry. Unlike isotropic noise mechanisms, the polar mechanism maintains semantic neighborhoods in the embedding space and better preserves downstream utility. Experimental evaluations on SQuAD, Yelp, and AG News datasets demonstrate that STAMP, when combined with the normalized polar mechanism, consistently achieves superior privacy-utility trade-offs across varying per-token privacy budgets.
format Preprint
id arxiv_https___arxiv_org_abs_2603_12237
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle STAMP: Selective Task-Aware Mechanism for Text Privacy
Tian, Fengwei
Bhattacharjee, Payel
Hanson, Heidi
Rubin, Geoffrey D.
Lo, Joseph Y.
Tandon, Ravi
Machine Learning
Cryptography and Security
Information Theory
We present STAMP (Selective Task-Aware Mechanism for Text Privacy), a new framework for task-aware text privatization that achieves an improved privacy-utility trade-off. STAMP selectively allocates privacy budgets across tokens by jointly considering (i) each token's importance to the downstream task (as measured via a task- or query-specific representation), and (ii) its privacy sensitivity (e.g., names, dates, identifiers). This token-level partitioning enables fine-grained, group-wise control over the level of noise applied to different parts of the input, balancing privacy protection with task relevance. To privatize individual token embeddings, we introduce the polar mechanism, which perturbs only the direction of embeddings on the unit sphere while preserving their magnitude. Decoding is performed via cosine nearest-neighbor search, aligning the perturbation geometry with the decoding geometry. Unlike isotropic noise mechanisms, the polar mechanism maintains semantic neighborhoods in the embedding space and better preserves downstream utility. Experimental evaluations on SQuAD, Yelp, and AG News datasets demonstrate that STAMP, when combined with the normalized polar mechanism, consistently achieves superior privacy-utility trade-offs across varying per-token privacy budgets.
title STAMP: Selective Task-Aware Mechanism for Text Privacy
topic Machine Learning
Cryptography and Security
Information Theory
url https://arxiv.org/abs/2603.12237