arXiv reaDer
DDS2M: ハイパースペクトル画像復元のための自己教師ありノイズ除去拡散空間スペクトル モデル
DDS2M: Self-Supervised Denoising Diffusion Spatio-Spectral Model for Hyperspectral Image Restoration
拡散モデルは、特にノイズのロバスト性の点で、画像復元の印象的なパフォーマンスにより、最近注目を集めています。ただし、既存の拡散ベースの方法は、大量のトレーニング データでトレーニングされ、分布内で非常にうまく機能しますが、分布シフトの影響を非常に受けやすい可能性があります。これは、データ不足のハイパースペクトル イメージ (HSI) の復元には特に不適切です。この問題に取り組むために、この作業では、HSI 復元のための自己管理型拡散モデル、つまりノイズ除去拡散空間スペクトル モデル (DDS2M) を提案しました。拡散プロセス。追加のトレーニング データを使用せずに、劣化した HSI のみを使用します。 VS2M では、変分推論ベースの損失関数がカスタマイズされ、トレーニングされていない空間ネットワークとスペクトル ネットワークが事後分布を学習できるようになります。これは、サンプリング チェーンの遷移として機能し、拡散プロセスを逆転させるのに役立ちます。 DDS2M は、その自己管理型の性質と拡散プロセスの恩恵を受けて、既存の拡散ベースの方法と比較して、さまざまな HSI に対するより強力な一般化機能と、既存の HSI 復元方法と比較したノイズに対する優れたロバスト性を享受しています。 HSI のノイズ除去、ノイズの多い HSI の完了、およびさまざまな HSI での超解像に関する広範な実験は、既存のタスク固有の最先端技術に対する DDS2M の優位性を示しています。
Diffusion models have recently received a surge of interest due to their impressive performance for image restoration, especially in terms of noise robustness. However, existing diffusion-based methods are trained on a large amount of training data and perform very well in-distribution, but can be quite susceptible to distribution shift. This is especially inappropriate for data-starved hyperspectral image (HSI) restoration. To tackle this problem, this work puts forth a self-supervised diffusion model for HSI restoration, namely Denoising Diffusion Spatio-Spectral Model (DDS2M), which works by inferring the parameters of the proposed Variational Spatio-Spectral Module (VS2M) during the reverse diffusion process, solely using the degraded HSI without any extra training data. In VS2M, a variational inference-based loss function is customized to enable the untrained spatial and spectral networks to learn the posterior distribution, which serves as the transitions of the sampling chain to help reverse the diffusion process. Benefiting from its self-supervised nature and the diffusion process, DDS2M enjoys stronger generalization ability to various HSIs compared to existing diffusion-based methods and superior robustness to noise compared to existing HSI restoration methods. Extensive experiments on HSI denoising, noisy HSI completion and super-resolution on a variety of HSIs demonstrate DDS2M's superiority over the existing task-specific state-of-the-arts.
updated: Sun Mar 12 2023 14:57:04 GMT+0000 (UTC)
published: Sun Mar 12 2023 14:57:04 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アソシエイト