arXiv reaDer
コンパクトな単一画像の曇り除去のための対照学習
Contrastive Learning for Compact Single Image Dehazing
単一画像の曇り除去は、深刻な情報の縮退のため、適切な問題です。ただし、既存の深層学習ベースのデヘイズ方法は、ネガティブな情報が活用されていない間、デヘイズネットワークのトレーニングをガイドするためのポジティブサンプルとして明確な画像のみを採用します。さらに、それらのほとんどは、深さと幅を増やしてヘイズ除去ネットワークを強化することに焦点を当てており、計算とメモリの重要な要件につながります。この論文では、コントラスト学習に基づいて構築された新しい対照正則化(CR)を提案し、ぼんやりとした画像と鮮明な画像の両方の情報をそれぞれネガティブサンプルとポジティブサンプルとして活用します。 CRは、復元された画像が鮮明な画像に近づき、表現空間のかすんでいる画像から遠くに押し出されるようにします。さらに、パフォーマンスとメモリストレージ間のトレードオフを考慮して、オートエンコーダのような(AE)フレームワークに基づくコンパクトなヘイズ除去ネットワークを開発します。これには、適応型ミックスアップ操作と動的機能拡張モジュールが含まれます。これらは、情報フローを適応的に維持し、受容野を拡大してネットワークの変換機能を向上させることでそれぞれメリットを得ることができます。オートエンコーダーと対照的な正則化を備えたデヘイズネットワークをAECR-Netと呼びます。合成データセットと実世界のデータセットに関する広範な実験は、AECR-Netが最先端のアプローチを上回っていることを示しています。コードはhttps://github.com/GlassyWu/AECR-Netでリリースされています。
Single image dehazing is a challenging ill-posed problem due to the severe information degeneration. However, existing deep learning based dehazing methods only adopt clear images as positive samples to guide the training of dehazing network while negative information is unexploited. Moreover, most of them focus on strengthening the dehazing network with an increase of depth and width, leading to a significant requirement of computation and memory. In this paper, we propose a novel contrastive regularization (CR) built upon contrastive learning to exploit both the information of hazy images and clear images as negative and positive samples, respectively. CR ensures that the restored image is pulled to closer to the clear image and pushed to far away from the hazy image in the representation space. Furthermore, considering trade-off between performance and memory storage, we develop a compact dehazing network based on autoencoder-like (AE) framework. It involves an adaptive mixup operation and a dynamic feature enhancement module, which can benefit from preserving information flow adaptively and expanding the receptive field to improve the network's transformation capability, respectively. We term our dehazing network with autoencoder and contrastive regularization as AECR-Net. The extensive experiments on synthetic and real-world datasets demonstrate that our AECR-Net surpass the state-of-the-art approaches. The code is released in https://github.com/GlassyWu/AECR-Net.
updated: Mon Apr 19 2021 14:56:21 GMT+0000 (UTC)
published: Mon Apr 19 2021 14:56:21 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アソシエイト