arXiv:2501.08361cs.CVcs.LG2025-01

通过增强权重平均的模型多样性,提升分布外泛化与少样本领域自适应性能。

Weight Averaging for Out-of-Distribution Generalization and Few-Shot Domain Adaptation

  • 引入梯度相似性作为正则项,显式增加权重平均中的模型多样性。
  • 在多个数字数据集和领域自适应数据集上,显著提升分布外泛化与少样本域适应准确率。
  • 结合权重平均与尖锐感知优化,适用于小样本场景下的跨域学习任务。

经验风险最小化(ERM)对数据分布变化不鲁棒。当测试数据分布与训练数据不一致时,即为分布外泛化问题。近年来,计算机视觉中提出了两种应对策略:权重平均(WA)和尖锐感知最小化(SAM)。WA通过训练多个不同超参数的模型并平均其权重,显著提升分布外泛化性能;SAM则寻找平坦极小值,已被证明在分布偏移下表现优异。尽管已有进展,仍可进一步优化。本文提出在WA中显式增加模型多样性,引入梯度相似性作为损失正则项,以进一步提升分布外泛化性能。同时,将WA与SAM结合,用于解决少样本领域自适应问题。在数字数据集(MNIST、SVHN、USPS、MNIST-M)及领域自适应数据集(VLCS、PACS)上的大量实验表明,该方法不仅提升分布外泛化性能,还显著提高少样本域适应准确率。

原文摘要 · Abstract (English)

Empirical risk minimization (ERM) is not robust to changes in the distribution of data. When the distribution of test data is different from that of training data, the problem is known as out-of-distribution generalization. Recently, two techniques have been developed for addressing out-of-distribution generalization in computer vision: weight averaging (WA) and sharpness-aware minimization (SAM). WA involves training multiple models with different hyperparameters and then averaging the weights of these models, which can significantly improve out-of-distribution generalization performance. SAM optimizes a neural network to find minima in flat regions, which have been proven to perform well under distribution shifts. While these techniques have made great progress, there is still room for improvement and further exploration. In this thesis, we propose increasing the model diversity in WA explicitly by introducing gradient similarity as a loss regularizer to further improve out-of-distribution generalization performance. We also propose combining WA and SAM to solve the problem of few-shot domain adaptation. Our extensive experiments on digits datasets (MNIST, SVHN, USPS, MNIST-M) and other domain adaptation datasets (VLCS, PACS) show that combining WA and SAM leads to improved out-of-distribution generalization performance and significantly increases few-shot domain adaptation accuracy.

分布外泛化权重平均少样本学习领域自适应

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