arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2404.10261stat.MLcs.LG

更轻量、更好、更快的多源域自适应:基于高斯混合模型与最优传输

Lighter, Better, Faster Multi-Source Domain Adaptation with Gaussian Mixture Models and Optimal Transport

CEA, List(CEA LIST);Université Paris-Saclay(巴黎-萨克雷大学)

Eduardo Fernandes Montesuma, Fred Ngolè Mboula, Antoine Souloumiac

AI总结:

本文提出基于高斯混合模型与最优传输的多源域自适应框架,通过高效求解GMM间最优传输及重心计算,提出WBT和DaDiL两种策略,在四个基准上实现更快、参数更少且性能更优的结果。

摘要:

本文研究多源域自适应(MSDA),这是迁移学习中的一项任务,其中需要将多个异质且有标签的源概率测度适应到一个不同的、无标签的目标测度。我们提出了一种基于最优传输(OT)和高斯混合模型(GMMs)的新型MSDA框架。该框架具有两个关键优势:第一,GMMs之间的最优传输可以通过线性规划高效求解;第二,它为监督学习(尤其是分类)提供了一个便捷的模型,因为GMM中的分量可以与现有类别相关联。基于GMM-OT问题,我们提出了一种计算GMM重心(barycenters)的新技术。基于这一新算法,我们提出了两种新的MSDA策略:GMM- Wasserstein重心传输(WBT)和GMM-数据集字典学习(DaDiL)。我们在图像分类和故障诊断的四个基准上对我们的方法进行了实证评估,结果表明我们在优于现有技术的同时,速度更快且参数更少。我们的代码公开在https://github.com/eddardd/gmm_msda。

英文摘要:

In this paper, we tackle Multi-Source Domain Adaptation (MSDA), a task in transfer learning where one adapts multiple heterogeneous, labeled source probability measures towards a different, unlabeled target measure. We propose a novel framework for MSDA, based on Optimal Transport (OT) and Gaussian Mixture Models (GMMs). Our framework has two key advantages. First, OT between GMMs can be solved efficiently via linear programming. Second, it provides a convenient model for supervised learning, especially classification, as components in the GMM can be associated with existing classes. Based on the GMM-OT problem, we propose a novel technique for calculating barycenters of GMMs. Based on this novel algorithm, we propose two new strategies for MSDA: GMM-Wasserstein Barycenter Transport (WBT) and GMM-Dataset Dictionary Learning (DaDiL). We empirically evaluate our proposed methods on four benchmarks in image classification and fault diagnosis, showing that we improve over the prior art while being faster and involving fewer parameters. Our code is publicly available at https://github.com/eddardd/gmm_msda

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

↑