arXiv reaDer
信頼できる銀行による水中画像復元のための対照的な半教師あり学習
Contrastive Semi-supervised Learning for Underwater Image Restoration via Reliable Bank
最近の水中画像復元技術の目覚ましい成果にもかかわらず、ラベル付きデータの欠如はさらなる進歩の大きなハードルとなっています。この作業では、ラベルのないデータをネットワークトレーニングに組み込むために、平均教師ベースの半教師付き水中画像復元 (Semi-UIR) フレームワークを提案します。ただし、単純な平均教師法には 2 つの主な問題があります。(1) 教師の予測が間違っていると、トレーニングで使用される整合性損失が無効になる可能性があります。 (2) L1 距離を使用すると、ネットワークが間違ったラベルをオーバーフィットする可能性があり、その結果、確認バイアスが発生します。上記の問題に対処するために、最初に信頼できるバンクを導入して、「史上最高の」出力を疑似グラウンド トゥルースとして保存します。出力の品質を評価するために、最も信頼できる NR-IQA メソッドを選択するために、単調性プロパティに基づいて経験的分析を行います。さらに、確認バイアスの問題を考慮して、対照的な正則化を組み込んで、間違ったラベルへの過剰適合を防ぎます。完全参照および非参照水中ベンチマークの両方での実験結果は、我々のアルゴリズムが量的および質的に SOTA メソッドよりも明らかに改善されていることを示しています。コードは https://github.com/Huang-ShiRui/Semi-UIR で公開されています。
Despite the remarkable achievement of recent underwater image restoration techniques, the lack of labeled data has become a major hurdle for further progress. In this work, we propose a mean-teacher based Semi-supervised Underwater Image Restoration (Semi-UIR) framework to incorporate the unlabeled data into network training. However, the naive mean-teacher method suffers from two main problems: (1) The consistency loss used in training might become ineffective when the teacher's prediction is wrong. (2) Using L1 distance may cause the network to overfit wrong labels, resulting in confirmation bias. To address the above problems, we first introduce a reliable bank to store the "best-ever" outputs as pseudo ground truth. To assess the quality of outputs, we conduct an empirical analysis based on the monotonicity property to select the most trustworthy NR-IQA method. Besides, in view of the confirmation bias problem, we incorporate contrastive regularization to prevent the overfitting on wrong labels. Experimental results on both full-reference and non-reference underwater benchmarks demonstrate that our algorithm has obvious improvement over SOTA methods quantitatively and qualitatively. Code has been released at https://github.com/Huang-ShiRui/Semi-UIR.
updated: Tue Mar 21 2023 02:21:03 GMT+0000 (UTC)
published: Thu Mar 16 2023 06:14:18 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アソシエイト