arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2307.04033cs.LGcs.AI

基于变分邻居标注的概率性测试时泛化

Probabilistic Test-Time Generalization by Variational Neighbor-Labeling

University of Amsterdam(阿姆斯特丹大学);TU Munich(慕尼黑工业大学);United Imaging Healthcare, Co., Ltd.(联影医疗科技股份有限公司);Core42

Sameer Ambekar, Zehao Xiao, Jiayi Shen, Xiantong Zhen, Cees G. M. Snoek

AI总结:

针对严格源训练与目标测试下的域泛化问题,提出将测试时伪标签建模为概率分布并融合邻居信息的变分邻居标签方法,且通过元泛化训练模拟目标适配,实验验证其有效性。

摘要:

本文致力于研究域泛化问题,即模型仅在源域上进行训练,随后部署到未见过的目标域上。我们遵循源训练与目标测试严格分离的原则,但在推理过程中利用未标注目标数据本身的价值。我们作出了三点贡献。第一,我们提出对目标样本进行概率性伪标注,以便在测试时将源域训练的模型泛化到目标域。我们将测试时泛化表述为一个变分推断问题,通过将伪标签建模为分布,来考虑泛化过程中的不确定性,并缓解不准确伪标签带来的误导性信号。第二,我们学习融合相邻目标样本信息的变分邻居标签,以生成更鲁棒的伪标签。第三,为了学习融合更具代表性的目标信息并生成更精确、更鲁棒的变分邻居标签的能力,我们在训练期间引入了一个元泛化阶段,以模拟泛化过程。在七个广泛使用的数据集上进行的实验证明了我们所提方案的优势、能力和有效性。

英文摘要:

This paper strives for domain generalization, where models are trained exclusively on source domains before being deployed on unseen target domains. We follow the strict separation of source training and target testing, but exploit the value of the unlabeled target data itself during inference. We make three contributions. First, we propose probabilistic pseudo-labeling of target samples to generalize the source-trained model to the target domain at test time. We formulate the generalization at test time as a variational inference problem, by modeling pseudo labels as distributions, to consider the uncertainty during generalization and alleviate the misleading signal of inaccurate pseudo labels. Second, we learn variational neighbor labels that incorporate the information of neighboring target samples to generate more robust pseudo labels. Third, to learn the ability to incorporate more representative target information and generate more precise and robust variational neighbor labels, we introduce a meta-generalization stage during training to simulate the generalization procedure. Experiments on seven widely-used datasets demonstrate the benefits, abilities, and effectiveness of our proposal.

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

↑