arXiv reaDer
圧縮センシングのためのメモリ増強ディープアンフォールディングネットワーク
Memory-Augmented Deep Unfolding Network for Compressive Sensing
切り捨てられた最適化手法をディープニューラルネットワークにマッピングするディープアンフォールディングネットワーク(DUN)は、その優れた解釈可能性と高性能により、圧縮センシング(CS)でますます注目を集めています。 DUNの各段階は、最適化の1回の反復に対応します。人間の脳の記憶処理の観点からDUNを理解することにより、既存のDUNには2つの問題があることがわかります。 1つは、短期記憶と見なすことができる2つの隣接するステージごとの情報が、通常、深刻に失われることです。もう1つは、前のステージが現在のステージに影響を与えることを保証する明示的なメカニズムではありません。つまり、メモリは簡単に忘れられます。これらの問題を解決するために、この論文では、CS用の永続メモリを備えた新しいDUNを提案します。これは、メモリ拡張ディープアンフォールディングネットワーク(MADUN)と呼ばれます。 2種類の記憶増強メカニズム、すなわち高スループット短期記憶(HSM)とクロスステージ長期記憶(CLM)を組み合わせることにより、記憶増強近位マッピングモジュール(MAPMM)を設計します。 HSMを利用して、DUNがマルチチャネルの短期記憶を送信できるようにします。これにより、隣接するステージ間の情報損失が大幅に削減されます。 CLMは、カスケードステージ全体での深い情報の依存関係を開発するために使用されます。これにより、ネットワーク表現機能が大幅に強化されます。自然画像とMR画像に関する広範なCS実験は、情報を維持およびバランスさせる強力な能力により、MADUNが既存の最先端の方法を大幅に上回っていることを示しています。ソースコードはhttps://github.com/jianzhangcs/MADUN/で入手できます。
Mapping a truncated optimization method into a deep neural network, deep unfolding network (DUN) has attracted growing attention in compressive sensing (CS) due to its good interpretability and high performance. Each stage in DUNs corresponds to one iteration in optimization. By understanding DUNs from the perspective of the human brain's memory processing, we find there exists two issues in existing DUNs. One is the information between every two adjacent stages, which can be regarded as short-term memory, is usually lost seriously. The other is no explicit mechanism to ensure that the previous stages affect the current stage, which means memory is easily forgotten. To solve these issues, in this paper, a novel DUN with persistent memory for CS is proposed, dubbed Memory-Augmented Deep Unfolding Network (MADUN). We design a memory-augmented proximal mapping module (MAPMM) by combining two types of memory augmentation mechanisms, namely High-throughput Short-term Memory (HSM) and Cross-stage Long-term Memory (CLM). HSM is exploited to allow DUNs to transmit multi-channel short-term memory, which greatly reduces information loss between adjacent stages. CLM is utilized to develop the dependency of deep information across cascading stages, which greatly enhances network representation capability. Extensive CS experiments on natural and MR images show that with the strong ability to maintain and balance information our MADUN outperforms existing state-of-the-art methods by a large margin. The source code is available at https://github.com/jianzhangcs/MADUN/.
updated: Tue Oct 19 2021 07:03:12 GMT+0000 (UTC)
published: Tue Oct 19 2021 07:03:12 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アソシエイト