通过两阶段优化,同时提升语言模型的泛化与鲁棒性。
Unified Enhancement of the Generalization and Robustness of Language Models via Bi-Stage Optimization
- 前向阶段用自适应丢弃增强对抗样本输出分布多样性
- 后向阶段仅更新关键参数,降低不必要的偏差
- 理论证明可平滑损失曲面,适合对鲁棒性有要求的场景
神经网络语言模型在泛化与鲁棒性方面面临重大挑战。当前多数研究仅关注单一目标,缺乏同时提升二者的方法。本文提出双阶段优化框架UEGR,统一增强语言模型的泛化与鲁棒性。前向传播阶段,通过自适应丢弃丰富对抗样本的输出概率分布,生成多样子模型,并引入JS散度与对抗损失强化输出稳定性;后向传播阶段,计算参数显著性得分,仅更新最关键参数,减少不必要偏差,增强模型韧性。理论分析表明,该框架包含梯度正则化以限制对输入扰动的敏感性,以及选择性参数更新以平滑损失景观,从而同时提升泛化与鲁棒性。实验结果表明,相比现有方法,本方法在13个公开语言数据集上均显著提升模型性能,达到当前最优(SOTA)水平。
原文摘要 · Abstract (English)
Neural network language models (LMs) are confronted with significant challenges in generalization and robustness. Currently, many studies focus on improving either generalization or robustness in isolation, without methods addressing both aspects simultaneously, which presents a significant challenge in developing LMs that are both robust and generalized. In this paper, we propose a bi-stage optimization framework to uniformly enhance both the generalization and robustness of LMs, termed UEGR. Specifically, during the forward propagation stage, we enrich the output probability distributions of adversarial samples by adaptive dropout to generate diverse sub models, and incorporate JS divergence and adversarial losses of these output distributions to reinforce output stability. During backward propagation stage, we compute parameter saliency scores and selectively update only the most critical parameters to minimize unnecessary deviations and consolidate the model's resilience. Theoretical analysis shows that our framework includes gradient regularization to limit the model's sensitivity to input perturbations and selective parameter updates to flatten the loss landscape, thus improving both generalization and robustness. The experimental results show that our method significantly improves the generalization and robustness of LMs compared to other existing methods across 13 publicly available language datasets, achieving state-of-the-art (SOTA) performance.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。