本文介绍Protegrity-PII,一种面向结构化和非结构化文本的个人身份信息(PII)识别决策模型。作者同时提出新基准PII-SCOPE,在该基准上模型取得0.915的类型化召回率和0.907的精确率。效率方面,借助批处理推理,单块RTX 6000显卡上处理速度达每百万字符0.49秒,成本为每百万字符0.0005美元。得益于训练所用实体类型体系覆盖面广,该模型在与多种先进模型和数据集的对比中表现出更优的分布外泛化能力。该工作为PII检测提供了兼顾准确率、吞吐量与泛化性的方案,并给出了可复用的评测基准。
| PII | 个人可识别信息,指能够识别特定个人身份的数据。 |
| PII-SCOPE | 本文提出的用于评估PII检测模型性能的新基准数据集。 |
| typed recall | 类型化召回率,衡量模型正确识别出特定类型实体的能力。 |
| out-of-distribution generalization | 分布外泛化,指模型在训练数据分布之外的数据上的表现能力。 |
| batched inference | 批处理推理,一次性处理多个输入以提高计算效率的推理方式。 |
📱 每天一份 AI 前沿日报
关注公众号,每天 09:00 推送 · 不错过任何重磅