arXiv reaDer
タスク転移学習によるレインストリークの除去
Removing Rain Streaks via Task Transfer Learning
ペアになった実世界のトレーニング データを収集することは難しいため、現在、画像のデレインは、Photoshop レンダリングなどによって生成された合成データを使用した教師あり学習によって支配されています。ただし、実際の雨のシーンへの一般化は、合成データと現実世界のデータの間にギャップがあるため、通常は制限されます。この論文では、最初に、教師ありデレイン モデルが実際の雨のケースにうまく一般化できない理由を統計的に調査し、合成データと実際の雨データの実質的な違いを見つけます。私たちの研究に触発されて、他の接続されたタスクから好ましいデレイン表現を学習することにより、雨を取り除くことを提案します。連結タスクでは、実データのラベルを簡単に取得できます。したがって、私たちの核となるアイデアは、タスク転送を通じて実際のデータから表現を学習し、脱線一般化を改善することです。したがって、私たちは学習戦略をタスク転移学習と呼んでいます。接続されたタスクが複数ある場合は、知識の蒸留によってモデルのサイズを縮小することを提案します。接続されたタスクの事前トレーニング済みモデルは教師として扱われ、すべての知識が学生ネットワークに蒸留されるため、モデルのサイズを縮小しながら、すべての接続されたタスクからの効果的な事前表現を保持します。最後に、スチューデント ネットワークは少数の対になった合成雨データを使用して微調整され、事前トレーニング済みの事前表現をガイドして雨を除去します。広範な実験により、提案されたタスク転送学習戦略が驚くほど成功し、最先端の教師あり学習方法と比較して有利であり、合成データに対する他の半教師あり脱線方法を明らかに上回っていることが示されています。特に、実世界のシーンへの優れた一般化を示しています。
Due to the difficulty in collecting paired real-world training data, image deraining is currently dominated by supervised learning with synthesized data generated by e.g., Photoshop rendering. However, the generalization to real rainy scenes is usually limited due to the gap between synthetic and real-world data. In this paper, we first statistically explore why the supervised deraining models cannot generalize well to real rainy cases, and find the substantial difference of synthetic and real rainy data. Inspired by our studies, we propose to remove rain by learning favorable deraining representations from other connected tasks. In connected tasks, the label for real data can be easily obtained. Hence, our core idea is to learn representations from real data through task transfer to improve deraining generalization. We thus term our learning strategy as task transfer learning. If there are more than one connected tasks, we propose to reduce model size by knowledge distillation. The pretrained models for the connected tasks are treated as teachers, all their knowledge is distilled to a student network, so that we reduce the model size, meanwhile preserve effective prior representations from all the connected tasks. At last, the student network is fine-tuned with minority of paired synthetic rainy data to guide the pretrained prior representations to remove rain. Extensive experiments demonstrate that proposed task transfer learning strategy is surprisingly successful and compares favorably with state-of-the-art supervised learning methods and apparently surpass other semi-supervised deraining methods on synthetic data. Particularly, it shows superior generalization over them to real-world scenes.
updated: Sun Aug 28 2022 03:32:17 GMT+0000 (UTC)
published: Sun Aug 28 2022 03:32:17 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アソシエイト