无需干净数据,训练中实时自动修正错误标签
Learning from Noisy Labels via Self-Taught On-the-Fly Meta Loss Rescaling
- 用模型自身特征动态调整样本权重,不依赖真实标签
- 在多个NLP任务上提升性能,对话建模效果显著
- 适合标签噪声多的场景,尤其适合资源有限的研究者
高质量标签对训练有效机器学习模型至关重要,但标注成本高,且专业标注仍含错误与模糊。传统数据清洗需额外处理且损失信息。替代方案是在训练中实时重加权样本以降低错误标签影响,但通常需要干净种子数据。本文提出无监督的在线元损失重标定方法,仅利用待训练模型提供的特征,在无需真实标签分布的前提下实时学习重标函数。通过新颖的元学习框架,从噪声训练集中采样验证数据用于元更新。该方法在多种NLP任务中持续提升性能,计算开销极小。首次在挑战性的对话建模任务中实现在线数据重加权,对噪声和干净数据均鲁棒,缓解类别不平衡,防止过拟合噪声标签。随着训练推进,正确标签影响力增强,错误标签影响被抑制,模型可不断从自身信号中学习优化。
原文摘要 · Abstract (English)
Correct labels are indispensable for training effective machine learning models. However, creating high-quality labels is expensive, and even professionally labeled data contains errors and ambiguities. Filtering and denoising can be applied to curate labeled data prior to training, at the cost of additional processing and loss of information. An alternative is on-the-fly sample reweighting during the training process to decrease the negative impact of incorrect or ambiguous labels, but this typically requires clean seed data. In this work we propose unsupervised on-the-fly meta loss rescaling to reweight training samples. Crucially, we rely only on features provided by the model being trained, to learn a rescaling function in real time without knowledge of the true clean data distribution. We achieve this via a novel meta learning setup that samples validation data for the meta update directly from the noisy training corpus by employing the rescaling function being trained. Our proposed method consistently improves performance across various NLP tasks with minimal computational overhead. Further, we are among the first to attempt on-the-fly training data reweighting on the challenging task of dialogue modeling, where noisy and ambiguous labels are common. Our strategy is robust in the face of noisy and clean data, handles class imbalance, and prevents overfitting to noisy labels. Our self-taught loss rescaling improves as the model trains, showing the ability to keep learning from the model's own signals. As training progresses, the impact of correctly labeled data is scaled up, while the impact of wrongly labeled data is suppressed.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。