← 返回信息流
技术博客arXiv cs.CL·14 小时前

小数据大噪声:对抗训练提升参数高效微调鲁棒性

原标题:Small Data, Big Noise: Adversarial Training for Robust Parameter-Efficient Fine-Tuning

速览

针对参数高效微调(PEFT)在噪声和有限数据下鲁棒性不足的问题,研究提出SDBN框架,将对抗训练引入PEFT以增强模型鲁棒性和泛化能力。该框架包含SDBN-h和SDBN-p两种变体,分别利用字符级编辑枚举和LLM生成变体进行鲁棒优化。实验表明,该方法在低资源设置及词级、字符级扰动下均表现优异,且未引入额外参数或仅增加少量计算开销,提升了PEFT在现实场景中的可靠性。

AI 深度解读

Small Data, Big Noise: Adversarial Training for Robust Parameter-Efficient Fine-Tuning

背景

随着基础模型(Foundation Models)在自然语言处理(NLP)领域的广泛应用,如何高效地将这些庞大的模型适配到特定的下游任务中,成为了业界的核心痛点。参数高效微调(Parameter-Efficient Fine-Tuning, PEFT)技术因此应运而生,并迅速成为主流解决方案。PEFT 的核心优势在于,它通过仅训练极少数的额外参数(如 LoRA、Adapter 等),即可在保持基础模型大部分权重冻结的情况下,实现模型性能的显著提升,从而大幅降低了计算成本和存储需求。

然而,现有的 PEFT 方法在实际部署中面临着两个严峻的挑战:鲁棒性不足和数据稀缺。首先,PEFT 模型往往对输入数据中的噪声(Noise)非常敏感,无论是词级别的拼写错误、同义词替换,还是字符级别的扰动,都可能导致模型性能的大幅下降。其次,在许多低资源场景(Low-resource Settings)中,可用的标注训练数据非常有限。在数据量少且存在噪声的情况下,模型极易过拟合,导致泛化能力减弱。

尽管对抗训练(Adversarial Training)作为一种经典的提升模型鲁棒性的技术,在完整模型微调中已被广泛研究,但将其引入 PEFT 框架的研究却相对匮乏。尽管对抗训练和 PEFT 在理论上具有互补性——前者增强鲁棒性,后者降低计算门槛——但两者的结合点尚未被充分探索。

核心内容

本文提出了一种名为 SDBN (Small Data Big Noise) 的统一框架,旨在将对抗训练引入 PEFT 领域,以解决上述鲁棒性和泛化性问题。SDBN 的核心思想是在 PEFT 的训练过程中引入对抗扰动,迫使模型在训练阶段就适应最坏情况下的输入变化,从而提升其在真实世界复杂环境中的表现。

1. SDBN 框架概述

SDBN 框架并没有引入额外的模型参数,或者仅引入极少量的参数,因此不会显著增加计算开销。它通过修改损失函数的优化目标,在每一步梯度更新时,寻找能使损失最大化的“对抗样本”方向,并沿着该方向的相反方向更新参数。这种方法使得模型在有限的数据上能够学习到更具鲁棒性的特征表示。

2. 两种变体方法

为了适应不同类型的任务和噪声场景,作者提出了 SDBN 的两种具体变体,均使用离散不确定性集合(Discrete Uncertainty Sets)来生成对抗样本:

  • SDBN-h (Character-level / Hard Edits): 该方法主要针对字符级别的扰动。它通过枚举字符级别的编辑操作(如插入、删除、替换字符)来生成潜在的对抗样本。然后,利用梯度信息来评估这些变体对损失函数的影响,并选择那些能导致损失最大化的“最坏情况”变体(Worst-case variants)进行对抗训练。这种方法特别适用于处理拼写错误、OCR 识别错误等细粒度的噪声。

  • SDBN-p (LLM-generated / Prompt-based): 该方法主要针对生成式任务(Generative Tasks)。它利用大型语言模型(LLM)生成语义等价但形式不同的变体,作为对抗样本。通过这种方式,SDBN-p 能够在保持语义一致性的前提下,增加输入数据的多样性,从而提升模型在生成任务中的鲁棒性和泛化能力。

3. 实验验证

作者在多个基准测试(Benchmarks)上对 SDBN 进行了评估,实验结果涵盖了低资源设置以及词级别和字符级别的噪声干扰。

  • 性能提升:实验显示,SDBN 在多种场景下均显著优于现有的替代方法。特别是在低资源设置下,SDBN 能够更有效地利用有限的数据,避免过拟合。
  • 鲁棒性增强:在面对人为注入的噪声(如随机字符替换、单词删除)时,SDBN 微调后的模型表现出更强的稳定性,性能下降幅度远小于未使用对抗训练的基线模型。
  • 计算效率:与完整模型的对抗训练相比,SDBN 在 PEFT 框架下运行,保持了极低的计算成本,证明了其在资源受限环境下的可行性。

关键要点

  • 填补研究空白:首次系统性地探讨了将对抗训练与参数高效微调(PEFT)相结合的可能性,解决了两者结合点研究不足的问题。
  • 统一框架 SDBN:提出了 SDBN 框架,通过引入对抗扰动增强 PEFT 模型的鲁棒性和泛化能力,且未引入显著额外参数。
  • 双变体策略
    • SDBN-h:基于字符级编辑和梯度选择,适用于处理细粒度噪声和分类任务。
    • SDBN-p:基于 LLM 生成的变体,适用于生成式任务和语义层面的鲁棒性优化。
  • 低资源优势:在训练数据稀缺的场景下,SDBN 表现出显著的性能优势,有效缓解了数据不足导致的过拟合问题。
  • 实战价值:该方法计算开销小,易于集成到现有的 PEFT 流程中,特别适合数据稀缺且语言变异性高的真实世界部署场景。

意义与影响

SDBN 框架的提出具有重要的理论意义和实用价值。

理论层面来看,它证明了在参数高效微调的约束下,对抗训练依然能够有效发挥作用,并且能够与 PEFT 产生协同效应。这为后续研究如何在更广泛的模型压缩和适配技术中引入鲁棒性优化提供了新的思路。

应用层面来看,SDBN 解决了 PEFT 落地过程中的关键痛点。在医疗、法律、金融等对数据质量和模型可靠性要求极高的领域,标注数据往往稀缺且敏感。SDBN 使得开发者能够在不增加巨大算力负担的前提下,构建出对噪声不敏感、泛化能力强的专用模型。这对于推动 AI 技术在资源受限边缘设备或数据隐私敏感场景中的部署具有重要意义。

总之,SDBN 不仅是一个新的算法框架,更是一种在“小数据”与“大噪声”并存的现实世界中,实现可靠、高效 AI 适配的实践指南。

查看原文 →arxiv.org