arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2007.07645cs.CV
收藏

基于变分信息瓶颈的领域泛化学习方法

Learning to Learn with Variational Information Bottleneck for Domain Generalization

University of Amsterdam(阿姆斯特丹大学);Nankai University(南开大学);Mohamed bin Zayed University of Artificial Intelligence(穆罕默德·本·扎耶德人工智能大学);Duke University(杜克大学);Inception Institute of Artificial Intelligence(起源人工智能研究院)

Yingjun Du, Jun Xu, Huan Xiong, Qiang Qiu, Xiantong Zhen, Cees G. M. Snoek, Ling Shao

摘要:

领域泛化模型学习泛化到以前看不见的领域,但遭受预测不确定性和领域转移。在本文中,我们解决这两个问题。我们引入了一个概率元学习模型的领域泛化,在跨域共享的分类器参数建模为分布。这使得能够更好地处理未知域上的预测不确定性。为了处理域移位,我们通过提出Meta变分信息瓶颈原理来学习域不变表示,我们称之为MetaVIB。MetaVIB是来自新的变分边界的互信息,利用元学习设置的领域泛化。通过情景训练,MetaVIB学会逐渐缩小领域差距,建立领域不变的表示,同时最大限度地提高预测准确性。我们进行实验的三个基准跨域视觉识别。全面的消融研究验证了MetaVIB在领域泛化方面的优势。比较结果表明,我们的方法优于以前的方法一致。

英文摘要:

Domain generalization models learn to generalize to previously unseen domains, but suffer from prediction uncertainty and domain shift. In this paper, we address both problems. We introduce a probabilistic meta-learning model for domain generalization, in which classifier parameters shared across domains are modeled as distributions. This enables better handling of prediction uncertainty on unseen domains. To deal with domain shift, we learn domain-invariant representations by the proposed principle of meta variational information bottleneck, we call MetaVIB. MetaVIB is derived from novel variational bounds of mutual information, by leveraging the meta-learning setting of domain generalization. Through episodic training, MetaVIB learns to gradually narrow domain gaps to establish domain-invariant representations, while simultaneously maximizing prediction accuracy. We conduct experiments on three benchmarks for cross-domain visual recognition. Comprehensive ablation studies validate the benefits of MetaVIB for domain generalization. The comparison results demonstrate our method outperforms previous approaches consistently.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑