No official torrent yet — requesting it tells the flock what to host next. You cannot seed a catalog-only page.
Cataloged from perplexity-ai/PII-Tracer-MLX
All models by perplexity-ai →This repository provides PII-Tracer, the detector introduced in PII-TRACE.
MLX weights for PII-Tracer,
a PII masking model for conversational data. The model uses a ~600M-parameter
bidirectional Qwen3 encoder with a
perplexity-ai/pplx-embed-v1-0.6b
backbone and two heads:
private_person, account_number, private_url, private_date,
private_address, private_email, private_phone, other_pii, secret),
decoded with a constrained Viterbi decoder.The backbone's linear projections use 8-bit affine quantization with group size 64. Embeddings and normalization weights remain bf16; both heads remain fp32. The weights file is approximately 779 MB.
Quantized projections retain the backbone.* key names and store packed
uint32 weight tensors with bf16 scales and biases. The unquantized heads
use token_cls_head.* and sensitivity_head.*.
This repository supplies converted weights and configuration. It does not
include an MLX model implementation or loading script. Using these weights
requires an MLX implementation of the bidirectional Qwen3 encoder, affine
quantized projections, and the two heads. It must honor
config.json's backbone.is_causal = false and mask padding during attention
and mean pooling.
Tokenize without adding BOS/EOS tokens. The model's input limit is 4096
tokens; chunk longer documents before inference. For encoder hidden states
hidden_states, compute:
Here mean_pool excludes padding tokens. Decode token_logits with a
constrained BIOES Viterbi decoder using the biases in config.json, then map
token spans to character offsets. Label index 0 is O; each category has
four consecutive labels in B, I, E, S order, with categories ordered as:
The base model provides the reference model behavior and constrained BIOES decoding logic.
If you use or reference this work, please cite:
token_logits = hidden_states @ token_cls_head.weight.T + token_cls_head.biassensitivity_logit = mean_pool(hidden_states) @ sensitivity_head.weight.T + sensitivity_head.biassensitivity = sigmoid(sensitivity_logit)private_person, private_email, private_phone, private_address, private_url,private_date, account_number, secret, other_pii@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}}