4 个月前

通过共性-特性低秩分解实现有效的领域泛化

通过共性-特性低秩分解实现有效的领域泛化

摘要

领域泛化是指训练一个模型,使其能够推广到在训练过程中未见过的新领域。本文介绍了CSD(Common Specific Decomposition,共性特异性分解),该方法在此设置下同时学习一个共性组件(能够推广到新领域)和一个领域特异性组件(在训练领域上过拟合)。训练完成后,领域特异性组件将被丢弃,仅保留共性组件。该算法极其简单,只需修改任意给定神经网络架构的最终线性分类层即可实现。我们进行了原则性的分析以理解现有方法,提供了CSD的可识别性结果,并研究了低秩对领域泛化的影响。实验结果表明,CSD在基于领域擦除、领域扰动数据增强和元学习的领域泛化方法中,要么与最先进方法持平,要么优于这些方法。进一步对旋转MNIST数据集进行诊断分析,其中领域具有可解释性,证实了CSD成功分离了共性和领域特异性组件,从而实现了更好的领域泛化性能。

代码仓库

vihari/csd
官方
pytorch
GitHub 中提及

基准测试

基准方法指标
domain-generalization-on-lipitkCSD (Ours)
Accuracy: 87.3
domain-generalization-on-pacs-2CSD (Resnet-18)
Average Accuracy: 80.69
domain-generalization-on-rotated-fashionCSD
Accuracy: 78.9

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
通过共性-特性低秩分解实现有效的领域泛化 | 论文 | HyperAI超神经