HyperAIHyperAI

Command Palette

Search for a command to run...

批量归一化嵌入用于深度领域泛化

Mattia Segu Alessio Tonioni Federico Tombari

摘要

领域泛化旨在训练机器学习模型,在不同且未见过的领域上均能保持稳健的性能。近年来的多种方法通过使用多个数据集来训练模型,以提取领域无关的特征,从而期望实现对未知领域的泛化。与此不同,我们首先通过引入特定设计的批归一化层,显式地学习领域相关的表征,以分别收集各领域独立的统计信息。随后,我们提出利用这些统计信息将各个领域映射到一个共享的潜在空间中,在该空间中,领域归属可通过距离函数进行度量。在测试阶段,我们将来自未知领域的样本投影至同一潜在空间,并将其领域属性建模为已知领域属性的线性组合。我们在训练和测试阶段均采用相同的映射策略,从而同时学习一个潜在表征以及一个强大但轻量的集成模型。在主流领域泛化基准数据集PACS、Office-31和Office-Caltech上,我们的方法显著提升了分类准确率,优于当前最先进的技术。


用 AI 构建 AI

从创意到上线——通过免费 AI 协同编码、开箱即用的环境和最优惠的 GPU 价格,加速您的 AI 开发。

AI 协同编码
开箱即用的 GPU
最优定价

HyperAI Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供