中文

$k$Folden:用于分布外检测的 $k$ 折集成

计算与语言 2021-11-09 v2

摘要

分布外 (OOD) 检测是自然语言处理 (NLP) 中的一个重要问题。在本工作中,我们提出了一个简单而有效的框架 kkFolden,它在训练时模拟 OOD 检测的行为且无需使用任何外部数据。对于一个具有 kk 个训练标签的任务,kkFolden 诱导出 kk 个子模型,每个子模型在含 k1k-1 类且将剩余一类对该子模型掩码为未知的子集上训练。在训练中向子模型暴露未知标签,模型被鼓励对未知标签将概率均等地分配给所见到的 k1k-1 个标签,使该框架能够通过 OOD 模拟以自然方式同时处理分布内与分布外样本。以文本分类为原型,我们利用现有文本分类数据集建立了 OOD 检测基准。通过在所建基准上的综合比较与分析,我们证明了 kkFolden 在提升 OOD 检测性能的同时保持更高的域内分类准确率方面优于现有方法。代码与数据集见:\url{https://github.com/ShannonAI/kfolden-ood-detection}。

关键词

引用

@article{arxiv.2108.12731,
  title  = {$k$Folden: $k$-Fold Ensemble for Out-Of-Distribution Detection},
  author = {Xiaoya Li and Jiwei Li and Xiaofei Sun and Chun Fan and Tianwei Zhang and Fei Wu and Yuxian Meng and Jun Zhang},
  journal= {arXiv preprint arXiv:2108.12731},
  year   = {2021}
}

备注

To appear at EMNLP 2021