arXiv reaDer
MultiMatch: 半教師ありドメイン汎化のためのマルチタスク学習
MultiMatch: Multi-task Learning for Semi-supervised Domain Generalization
ドメイン一般化 (DG) は、ソース ドメインでモデルを学習して、目に見えないターゲット ドメインで適切に一般化することを目的としています。大きな成功を収めていますが、既存の方法のほとんどは、ソース ドメイン内のすべてのトレーニング サンプルのラベル情報を必要とします。これは、実際のアプリケーションでは時間と費用がかかります。この論文では、各ソースドメインにいくつかのラベル情報がある半教師付きドメイン汎化 (SSDG) タスクを解決することに頼っています。このタスクに対処するために、最初にマルチドメイン学習の理論を分析します。これは、1) ドメインギャップの影響を軽減し、2) すべてのサンプルを活用してモデルをトレーニングすることで、各ソースドメインの一般化エラーを効果的に削減できることを強調しています。疑似ラベルの品質を向上させます。分析によると、MultiMatch を提案します。つまり、FixMatch をマルチタスク学習フレームワークに拡張し、SSDG の高品質の疑似ラベルを生成します。具体的には、各トレーニング ドメインを 1 つのタスク (つまり、ローカル タスク) と見なし、すべてのトレーニング ドメインをまとめて (つまり、グローバル タスク)、目に見えないテスト ドメイン用の追加のタスクをトレーニングします。マルチタスク フレームワークでは、タスクごとに独立した BN と分類子を利用します。これにより、疑似ラベリング中に異なるドメインからの干渉を効果的に軽減できます。また、フレームワーク内のほとんどのパラメータは共有されており、すべてのトレーニング サンプルで十分にトレーニングできます。さらに、疑似ラベルの精度とモデルの一般化をさらに高めるために、トレーニングとテスト中にそれぞれグローバル タスクとローカル タスクからの予測を融合します。一連の実験により、提案された方法の有効性が検証され、いくつかのベンチマーク DG データセットで、既存の半教師付き方法および SSDG 方法よりも優れています。
Domain generalization (DG) aims at learning a model on source domains to well generalize on the unseen target domain. Although it has achieved great success, most of existing methods require the label information for all training samples in source domains, which is time-consuming and expensive in the real-world application. In this paper, we resort to solving the semi-supervised domain generalization (SSDG) task, where there are a few label information in each source domain. To address the task, we first analyze the theory of the multi-domain learning, which highlights that 1) mitigating the impact of domain gap and 2) exploiting all samples to train the model can effectively reduce the generalization error in each source domain so as to improve the quality of pseudo-labels. According to the analysis, we propose MultiMatch, i.e., extending FixMatch to the multi-task learning framework, producing the high-quality pseudo-label for SSDG. To be specific, we consider each training domain as a single task (i.e., local task) and combine all training domains together (i.e., global task) to train an extra task for the unseen test domain. In the multi-task framework, we utilize the independent BN and classifier for each task, which can effectively alleviate the interference from different domains during pseudo-labeling. Also, most of parameters in the framework are shared, which can be trained by all training samples sufficiently. Moreover, to further boost the pseudo-label accuracy and the model's generalization, we fuse the predictions from the global task and local task during training and testing, respectively. A series of experiments validate the effectiveness of the proposed method, and it outperforms the existing semi-supervised methods and the SSDG method on several benchmark DG datasets.
updated: Mon Apr 29 2024 06:32:01 GMT+0000 (UTC)
published: Thu Aug 11 2022 14:44:33 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)
Amazon.co.jpアソシエイト