arXiv reaDer
推論時間ラベル保存ターゲット射影によるドメインの一般化
Domain Generalization via Inference-time Label-Preserving Target Projections
さまざまな統計を持つ目に見えないターゲットドメイン上の一連のソースドメインでトレーニングされた機械学習モデルの一般化は、困難な問題です。この問題を解決するために多くのアプローチが提案されていますが、それらはトレーニング中にソースデータを利用するだけであり、推論時に単一のターゲット例が利用可能であるという事実を利用していません。これを動機として、単なる分類を超えて、推論時に対象サンプルを有効に活用する方法を提案します。私たちの方法には3つの要素があります-(i)ソースデータのラベル保存機能またはメトリック変換により、ソースサンプルはドメインに関係なくクラスに従ってクラスター化されます(ii)これらの機能でトレーニングされた生成モデル(iii)学習されたメトリックを使用して生成モデルの入力空間の最適化問題を解くことにより、推論中にソースフィーチャ多様体上のターゲットポイントのラベルを保持する投影。最後に、投影されたターゲットが分類器で使用されます。投影されたターゲットフィーチャはソースマニホールドから取得され、設計上実際のターゲットと同じラベルを持っているため、分類器は実際のターゲットよりも優れたパフォーマンスを発揮することが期待されます。私たちの方法は、複数のデータセットとタスクで最先端のドメイン一般化方法よりも優れていることを示しています。
Generalization of machine learning models trained on a set of source domains on unseen target domains with different statistics, is a challenging problem. While many approaches have been proposed to solve this problem, they only utilize source data during training but do not take advantage of the fact that a single target example is available at the time of inference. Motivated by this, we propose a method that effectively uses the target sample during inference beyond mere classification. Our method has three components - (i) A label-preserving feature or metric transformation on source data such that the source samples are clustered in accordance with their class irrespective of their domain (ii) A generative model trained on the these features (iii) A label-preserving projection of the target point on the source-feature manifold during inference via solving an optimization problem on the input space of the generative model using the learned metric. Finally, the projected target is used in the classifier. Since the projected target feature comes from the source manifold and has the same label as the real target by design, the classifier is expected to perform better on it than the true target. We demonstrate that our method outperforms the state-of-the-art Domain Generalization methods on multiple datasets and tasks.
updated: Mon Jul 19 2021 01:27:08 GMT+0000 (UTC)
published: Mon Mar 01 2021 17:10:38 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アソシエイト