用代理模型稳定训练,减少数据增强带来的性能波动。
PEER pressure: Model-to-Model Regularization for Single Source Domain Generalization
- 主模型与代理模型协同更新,逐步积累增强数据知识。
- 在PACS、Office-Home等数据集上显著降低域外性能波动。
- 仅用简单随机增强即超越复杂增强策略的先进方法。
数据增强是单源领域泛化常用手段,通过生成模拟域扩展源域,提升对未见目标域的泛化能力。本文发现,此类方法在目标域上的性能在训练过程中普遍波动,导致真实场景下模型选择困难。我们认为波动源于模型无法有效积累多样化增强带来的知识,加剧了训练中的特征失真。为此提出参数空间集成与熵正则化方法(PEER),由代理模型代为主模型学习增强数据,主模型通过与代理模型参数平均实现知识渐进累积。最大化两模型输出表征间的互信息,引导代理模型学习,缓解训练过程中的特征失真。实验表明,PEER能有效降低域外性能波动,在PACS、Digits、Office-Home和VLCS等多个数据集上提升泛化能力。值得注意的是,仅使用简单随机增强即达到当前最优性能,优于依赖复杂增强策略的先前方法。
原文摘要 · Abstract (English)
Data augmentation is a popular tool for single source domain generalization, which expands the source domain by generating simulated ones, improving generalization on unseen target domains. In this work, we show that the performance of such augmentation-based methods in the target domains universally fluctuates during training, posing challenges in model selection under realistic scenarios. We argue that the fluctuation stems from the inability of the model to accumulate the knowledge learned from diverse augmentations, exacerbating feature distortion during training. Based on this observation, we propose a novel generalization method, coined Parameter-Space Ensemble with Entropy Regularization (PEER), that uses a proxy model to learn the augmented data on behalf of the main model. The main model is updated by averaging its parameters with the proxy model, progressively accumulating knowledge over the training steps. Maximizing the mutual information between the output representations of the two models guides the learning process of the proxy model, mitigating feature distortion during training. Experimental results demonstrate the effectiveness of PEER in reducing the OOD performance fluctuation and enhancing generalization across various datasets, including PACS, Digits, Office-Home, and VLCS. Notably, our method with simple random augmentation achieves state-of-the-art performance, surpassing prior approaches on sDG that utilize complex data augmentation strategies.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。