ディープ ニューラル ネットワークは、さまざまな視覚タスクで大きな成功を収めています。ただし、最先端のモデルは、目に見えないテスト データセットに適用すると、ドメインのシフトによりパフォーマンスが低下することがよくあります。この論文では、ドメインシフトに対する特徴マップのチャネルの堅牢性を強化するという新しい観点から、ドメイン一般化のための新しいアプローチを紹介します。ソース ドメインでトレーニングされたモデルには、さまざまなドメイン間で不安定な活性化を示す相当数のチャネルが含まれており、これらのチャネルはドメイン固有の特徴を捕捉し、目に見えないターゲット ドメインにさらされると異常に動作する傾向があることが観察されています。この問題に対処するために、ドメイン シフトに対するチャネルの堅牢性を継続的に強化する DomainDrop フレームワークを提案します。このフレームワークでは、順方向伝播中に各ネットワーク層の特徴マップで不安定なチャネルを識別してドロップするためにドメイン弁別器が使用されます。私たちのフレームワークが一般化限界を効果的に下げることができることを理論的に証明します。いくつかのベンチマークに関する広範な実験により、私たちのフレームワークが他の競合手法と比較して最先端のパフォーマンスを達成していることが示されています。私たちのコードは https://github.com/lingeringlight/DomainDrop で入手できます。
Deep Neural Networks have exhibited considerable success in various visual tasks. However, when applied to unseen test datasets, state-of-the-art models often suffer performance degradation due to domain shifts. In this paper, we introduce a novel approach for domain generalization from a novel perspective of enhancing the robustness of channels in feature maps to domain shifts. We observe that models trained on source domains contain a substantial number of channels that exhibit unstable activations across different domains, which are inclined to capture domain-specific features and behave abnormally when exposed to unseen target domains. To address the issue, we propose a DomainDrop framework to continuously enhance the channel robustness to domain shifts, where a domain discriminator is used to identify and drop unstable channels in feature maps of each network layer during forward propagation. We theoretically prove that our framework could effectively lower the generalization bound. Extensive experiments on several benchmarks indicate that our framework achieves state-of-the-art performance compared to other competing methods. Our code is available at https://github.com/lingeringlight/DomainDrop.