arXiv reaDer
ドメイン適応のための相互学習:サンプルサイクルを備えた自己蒸留画像デヘイズネットワーク
Mutual Learning for Domain Adaptation: Self-distillation Image Dehazing Network with Sample-cycle
ディープラーニングベースの方法は、画像の曇り除去に大きな成果をもたらしました。ただし、既存の曇り除去ネットワークのほとんどは、シミュレートされたぼんやりとした画像を使用したモデルのトレーニングに集中しているため、ドメインシフトのために、実際のぼんやりとした画像に適用すると、一般化のパフォーマンスが低下します。本論文では、ドメイン適応のための相互学習のヘイズ除去フレームワークを提案する。具体的には、まず合成ドメインの教師ネットワークと実ドメインの学生ネットワークの2つのシャムネットワークを考案し、次にEMAと共同損失を活用して相互学習方式でそれらを最適化します。さらに、密度増強(HDA)モジュールに基づくサンプルサイクル戦略を設計し、学生ネットワークによって提供される疑似実世界の画像ペアをトレーニングに導入して、一般化パフォーマンスをさらに向上させます。合成データセットと実世界のデータセットの両方での広範な実験は、提案された相互学習フレームワークが、主観的および客観的な評価の点で最先端のヘイズ除去技術よりも優れていることを示しています。
Deep learning-based methods have made significant achievements for image dehazing. However, most of existing dehazing networks are concentrated on training models using simulated hazy images, resulting in generalization performance degradation when applied on real-world hazy images because of domain shift. In this paper, we propose a mutual learning dehazing framework for domain adaption. Specifically, we first devise two siamese networks: a teacher network in the synthetic domain and a student network in the real domain, and then optimize them in a mutual learning manner by leveraging EMA and joint loss. Moreover, we design a sample-cycle strategy based on density augmentation (HDA) module to introduce pseudo real-world image pairs provided by the student network into training for further improving the generalization performance. Extensive experiments on both synthetic and real-world dataset demonstrate that the propose mutual learning framework outperforms state-of-the-art dehazing techniques in terms of subjective and objective evaluation.
updated: Thu Mar 17 2022 16:32:14 GMT+0000 (UTC)
published: Thu Mar 17 2022 16:32:14 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アソシエイト