updates

README.md

@@ -11,7 +11,10 @@ tags:
 pipeline_tag: token-classification
 ---
 
-# pplx-pii-masking
+# PII-Tracer
+
+This repository provides PII-Tracer, the detector introduced in
+[PII-TRACE](https://arxiv.org/abs/2609.22200).
 
 PII masking model for conversational data. A ~600M-parameter bidirectional
 Qwen3 encoder
@@ -34,7 +37,7 @@ backbone, run encoder-only via `is_causal=false`) with two heads:
 from transformers import AutoModel
 
 model = AutoModel.from_pretrained(
-    "perplexity-ai/pplx-pii-masking", trust_remote_code=True
+    "perplexity-ai/PII-Tracer", trust_remote_code=True
 )
 
 text = ("Hi, I'm Daniel Whitfield, you can reach me at "
@@ -77,15 +80,29 @@ config, including `"is_causal": false` for bidirectional attention.
 
 ## Alternative packagings
 
-- **vLLM:** [perplexity-ai/pplx-pii-masking-vllm](https://huggingface.co/perplexity-ai/pplx-pii-masking-vllm)
+- **vLLM:** [perplexity-ai/PII-Tracer-vLLM](https://huggingface.co/perplexity-ai/PII-Tracer-vLLM)
   repackages the weights for token classification serving, with a scoring
   adapter and a decoding client.
-- **GGUF:** [perplexity-ai/pplx-pii-masking-GGUF](https://huggingface.co/perplexity-ai/pplx-pii-masking-GGUF)
+- **GGUF:** [perplexity-ai/PII-Tracer-GGUF](https://huggingface.co/perplexity-ai/PII-Tracer-GGUF)
   provides an f16 backbone for llama-server, fp32 heads, and a self-contained
   decoding client.
-- **MLX:** [perplexity-ai/pplx-pii-masking-MLX](https://huggingface.co/perplexity-ai/pplx-pii-masking-MLX)
+- **MLX:** [perplexity-ai/PII-Tracer-MLX](https://huggingface.co/perplexity-ai/PII-Tracer-MLX)
   provides 8-bit converted projection weights for MLX. An MLX implementation
   of the bidirectional encoder and two heads is required; a loader is not
   included in that repository.
 
 See each repository for its inference requirements and limitations.
+
+## Citation
+
+If you use or reference this work, please cite:
+
+```bibtex
+@article{zhang2026piitrace,
+  title = {{PII-TRACE}: A Benchmark for Context-Aware {PII} Detection in Multi-Turn {LLM} Conversations},
+  author = {Zhang, Kaiyuan and Wang, Chuan and Zhong, Joey and Fryzel, Paul and Polley, Kyle and Ma, Jerry and Li, Ninghui},
+  journal = {arXiv preprint arXiv:2609.22200},
+  year = {2026},
+  url = {https://arxiv.org/abs/2609.22200}
+}
+```