arXiv reaDer
MixStyleによるドメインの一般化
Domain Generalization with MixStyle
畳み込みニューラルネットワーク(CNN)は、識別機能を学習する際の優れた能力を示していますが、多くの場合、目に見えないドメインへの一般化は不十分です。ドメインの一般化は、ソースドメインのセットから、見えないドメインに一般化できるモデルを学習することで、この問題に対処することを目的としています。この論文では、ソースドメイン間でトレーニングサンプルのインスタンスレベルの特徴統計を確率的に混合することに基づいて、新しいアプローチを提案します。 MixStyleと呼ばれる私たちの方法は、視覚領域が画像スタイル(たとえば、写真とスケッチ画像)に密接に関連しているという観察に動機付けられています。このようなスタイル情報は、提案されたスタイルミキシングが行われるCNNの最下層によってキャプチャされます。トレーニングインスタンスのスタイルを混合すると、新しいドメインが暗黙的に合成され、ソースドメインのドメインの多様性が高まり、トレーニングされたモデルの一般化が可能になります。 MixStyleはミニバッチトレーニングに完全に適合し、実装が非常に簡単です。 MixStyleの有効性は、カテゴリ分類、インスタンス検索、強化学習などの幅広いタスクで実証されています。
Though convolutional neural networks (CNNs) have demonstrated remarkable ability in learning discriminative features, they often generalize poorly to unseen domains. Domain generalization aims to address this problem by learning from a set of source domains a model that is generalizable to any unseen domain. In this paper, a novel approach is proposed based on probabilistically mixing instance-level feature statistics of training samples across source domains. Our method, termed MixStyle, is motivated by the observation that visual domain is closely related to image style (e.g., photo vs.~sketch images). Such style information is captured by the bottom layers of a CNN where our proposed style-mixing takes place. Mixing styles of training instances results in novel domains being synthesized implicitly, which increase the domain diversity of the source domains, and hence the generalizability of the trained model. MixStyle fits into mini-batch training perfectly and is extremely easy to implement. The effectiveness of MixStyle is demonstrated on a wide range of tasks including category classification, instance retrieval and reinforcement learning.
updated: Mon Apr 05 2021 16:58:09 GMT+0000 (UTC)
published: Mon Apr 05 2021 16:58:09 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アソシエイト