arXiv:2410.14894cs.AIcs.CR2024-10NeurIPS被引 7

用众包软标签+分布鲁棒优化,提升毒性分类的泛化能力

Soft-Label Integration for Robust Toxicity Classification

论文配图:Soft-Label Integration for Robust Toxicity Classification
图 1 · 摘自论文原文
  • 通过双层优化融合众包标注的软标签
  • 在平均和最差组准确率上均优于基线方法
  • 适合需要抗分布偏移的文本安全检测场景

文本毒性分类仍面临挑战。单一标注者数据难以反映人类观点多样性,亟需引入众包标注以训练高效分类器。传统基于经验风险最小化的训练方法可能因利用伪相关性,在训练集与测试集分布不一致时失效。本文提出一种新型双层优化框架,结合众包标注与软标签技术,并通过分组分布鲁棒优化(GroupDRO)优化软标签权重,增强对分布外(OOD)风险的鲁棒性。理论证明了算法收敛性。实验表明,该方法在平均准确率和最差组准确率上均优于现有基线,验证了其利用众包标注实现更有效、更鲁棒的毒性分类的能力。

原文摘要 · Abstract (English)

Toxicity classification in textual content remains a significant problem. Data with labels from a single annotator fall short of capturing the diversity of human perspectives. Therefore, there is a growing need to incorporate crowdsourced annotations for training an effective toxicity classifier. Additionally, the standard approach to training a classifier using empirical risk minimization (ERM) may fail to address the potential shifts between the training set and testing set due to exploiting spurious correlations. This work introduces a novel bi-level optimization framework that integrates crowdsourced annotations with the soft-labeling technique and optimizes the soft-label weights by Group Distributionally Robust Optimization (GroupDRO) to enhance the robustness against out-of-distribution (OOD) risk. We theoretically prove the convergence of our bi-level optimization algorithm. Experimental results demonstrate that our approach outperforms existing baseline methods in terms of both average and worst-group accuracy, confirming its effectiveness in leveraging crowdsourced annotations to achieve more effective and robust toxicity classification.

毒性分类众包标注分布鲁棒软标签

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。