arXiv reaDer
畳み込みとコンテキストトランスフォーマーを使用したスペクトル圧縮イメージングの再構成
Spectral Compressive Imaging Reconstruction Using Convolution and Contextual Transformer
スペクトル圧縮イメージング(SCI)は、高次元のハイパースペクトル画像を2D測定にエンコードし、アルゴリズムを使用して空間スペクトルデータキューブを再構築することができます。現在、SCIの主なボトルネックは再構成アルゴリズムであり、最先端の(SOTA)再構成方法は、一般に、再構成時間が長い、および/または詳細の回復が不十分であるという問題に直面しています。本論文では、畳み込みの誘導バイアス能力と変圧器の強力なモデリング能力を同時に獲得することができ、再構成の品質を向上させるのに役立つ、新しいハイブリッドネットワークモジュール、すなわちCCoT(Convolution and Contextual Transformer)ブロックを提案します。細かい部分を復元します。提案されたCCoTブロックを、一般化された交互投影アルゴリズムに基づく深い展開フレームワークに統合し、さらにGAP-CCoTネットワークを提案します。最後に、GAP-CCoTアルゴリズムをSCI再構成に適用します。広範な合成データと実際のデータの実験を通じて、提案されたモデルは、既存のSOTAアルゴリズムよりも高い再構成品質(シミュレートされたベンチマークデータセットのPSNRで> 2dB)と短い実行時間を大幅に達成します。コードとモデルは一般に公開されます。
Spectral compressive imaging (SCI) is able to encode the high-dimensional hyperspectral image to a 2D measurement, and then uses algorithms to reconstruct the spatio-spectral data-cube. At present, the main bottleneck of SCI is the reconstruction algorithm, and the state-of-the-art (SOTA) reconstruction methods generally face the problem of long reconstruction time and/or poor detail recovery. In this paper, we propose a novel hybrid network module, namely CCoT (Convolution and Contextual Transformer) block, which can acquire the inductive bias ability of convolution and the powerful modeling ability of transformer simultaneously,and is conducive to improving the quality of reconstruction to restore fine details. We integrate the proposed CCoT block into deep unfolding framework based on the generalized alternating projection algorithm, and further propose the GAP-CCoT network. Finally, we apply the GAP-CCoT algorithm to SCI reconstruction. Through the experiments of extensive synthetic and real data, our proposed model achieves higher reconstruction quality (>2dB in PSNR on simulated benchmark datasets) and shorter running time than existing SOTA algorithms by a large margin. The code and models will be released to the public.
updated: Wed Mar 09 2022 13:52:27 GMT+0000 (UTC)
published: Sat Jan 15 2022 06:30:03 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アソシエイト