用双低秩适配分离选样本与训练,提升噪声标签下的模型性能
Weed Out, Then Harvest: Dual Low-Rank Adaptation is an Effective Noisy Label Detector for Noise-Robust Learning
- 通过清洁与噪声LoRA分离样本选择和模型训练过程
- 在合成与真实数据集上准确识别噪声样本,分类准确率提升显著
- 适合需要抗噪声训练的LLM微调场景
参数高效微调(PEFT)大语言模型在多种下游任务中表现优异。但在现实场景中,训练数据常含噪声标签。现有方法多基于小损失样本进行筛选,但该过程会因初始误选导致恶性循环,影响性能。为此,本文提出Delora框架,将样本选择与模型训练解耦。通过引入清洁与噪声LoRA,利用其记忆效应:清洁LoRA记忆干净数据,噪声LoRA约束记忆错误标签数据,形成可学习的阈值用于区分干净与噪声样本。模型训练则基于筛选出的可靠样本进行微调。在合成及真实噪声数据集上的实验表明,Delora在噪声标签检测与文本分类任务中均表现出色。
原文摘要 · Abstract (English)
Parameter-efficient fine-tuning (PEFT) large language models (LLMs) have shown impressive performance in various downstream tasks. However, in many real-world scenarios, the collected training data inevitably contains noisy labels. To learn from noisy labels, most solutions select samples with small losses for model training. However, the selected samples, in turn, impact the loss computation in the next iteration. An inaccurate initial selection can create a vicious cycle, leading to suboptimal performance. To break this cycle, we propose Delora, a novel framework that decouples the sample selection from model training. For sample selection, Delora establishes a noisy label detector by introducing clean and noisy LoRA. Benefiting from the memory effect, the clean LoRA is encouraged to memorize clean data, while the noisy LoRA is constrained to memorize mislabeled data, which serves as a learnable threshold for selecting clean and noisy samples. For model training, Delora can use carefully selected samples to fine-tune language models seamlessly. Experimental results on synthetic and real-world noisy datasets demonstrate the effectiveness of Delora in noisy label detection and text classification.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。