arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2109.01902cs.LGstat.ML
收藏

用于领域泛化的重心对齐与重构损失最小化

Barycentric-alignment and reconstruction loss minimization for domain generalization

Tufts University(塔夫茨大学);Boston University(波士顿大学)

Boyang Lyu, Thuan Nguyen, Prakash Ishwar, Matthias Scheutz, Shuchin Aeron

AI总结:

本文针对领域泛化中因理论风险上界存在双重依赖项而难以优化的问题,利用最优传输不等式提出了完全可优化的新上界,并基于此提出WBAE算法,通过同时最小化分类、重心对齐与重构损失,在多个数据集上优于现有方法。

摘要:

本文推进了机器学习中领域泛化(DG)的理论与实践。我们考虑典型的DG设定,其中假设由表示映射和标记函数组成。在此设定下,大多数流行的DG方法旨在通过最小化未见域中分类风险的一个已知上界来联合学习表示和标记函数。然而在实践中,基于这一理论上界的方法忽略了一项,由于该项双重依赖于表示映射和未见域中未知的最优标记函数,因此无法直接优化。为了弥合理论与实践之间的差距,我们引入了一个新的上界,该上界不包含具有此类双重依赖的项,从而为未见域产生了一个完全可优化的风险上界。我们的推导利用了经典和最新的传输不等式,将最优传输度量与信息论度量联系起来。与先前的上界相比,我们的上界引入了两个新项:(i) 在域之间对齐分布的Wasserstein-2重心项,以及(ii) 评估表示在重构原始数据时质量的重构损失项。基于这一新上界,我们提出了一种名为Wasserstein重心自编码器(WBAE)的新型DG算法,该算法同时最小化分类损失、重心损失和重构损失。数值结果表明,所提方法在多个数据集上优于当前最先进的DG算法。

英文摘要:

This paper advances the theory and practice of Domain Generalization (DG) in machine learning. We consider the typical DG setting where the hypothesis is composed of a representation mapping followed by a labeling function. Within this setting, the majority of popular DG methods aim to jointly learn the representation and the labeling functions by minimizing a well-known upper bound for the classification risk in the unseen domain. In practice, however, methods based on this theoretical upper bound ignore a term that cannot be directly optimized due to its dual dependence on both the representation mapping and the unknown optimal labeling function in the unseen domain. To bridge this gap between theory and practice, we introduce a new upper bound that is free of terms having such dual dependence, resulting in a fully optimizable risk upper bound for the unseen domain. Our derivation leverages classical and recent transport inequalities that link optimal transport metrics with information-theoretic measures. Compared to previous bounds, our bound introduces two new terms: (i) the Wasserstein-2 barycenter term that aligns distributions between domains, and (ii) the reconstruction loss term that assesses the quality of representation in reconstructing the original data. Based on this new upper bound, we propose a novel DG algorithm named Wasserstein Barycenter Auto-Encoder (WBAE) that simultaneously minimizes the classification loss, the barycenter loss, and the reconstruction loss. Numerical results demonstrate that the proposed method outperforms current state-of-the-art DG algorithms on several datasets.

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

↑