モデルは、トレーニングに使用できる単一のドメインが1つしかないのに、多くの見えないドメインでうまく機能することを目的としているという意味で、モデルの一般化における最悪のシナリオに関心があります。このドメイン外の一般化の問題を解決するために、メタ学習ベースの敵対的ドメイン拡張を提案します。重要なアイデアは、敵対的なトレーニングを活用して、「架空の」が「挑戦的な」母集団を作成することです。この母集団から、モデルは理論的な保証で一般化することを学ぶことができます。高速で望ましいドメイン拡張を容易にするために、メタ学習スキームでモデルトレーニングをキャストし、Wasserstein Auto-Encoderを使用して、広く使用されている最悪の場合の制約を緩和します。効率的なドメイン一般化のために不確実性の定量化を統合することにより、メソッドをさらに改善します。複数のベンチマークデータセットでの広範な実験は、単一ドメインの一般化に取り組む際の優れたパフォーマンスを示しています。
We are concerned with a worst-case scenario in model generalization, in the sense that a model aims to perform well on many unseen domains while there is only one single domain available for training. We propose Meta-Learning based Adversarial Domain Augmentation to solve this Out-of-Domain generalization problem. The key idea is to leverage adversarial training to create "fictitious" yet "challenging" populations, from which a model can learn to generalize with theoretical guarantees. To facilitate fast and desirable domain augmentation, we cast the model training in a meta-learning scheme and use a Wasserstein Auto-Encoder to relax the widely used worst-case constraint. We further improve our method by integrating uncertainty quantification for efficient domain generalization. Extensive experiments on multiple benchmark datasets indicate its superior performance in tackling single domain generalization.