arXiv reaDer
二重回帰学習による軽量超解像に向けて
Towards Lightweight Super-Resolution with Dual Regression Learning
ディープニューラルネットワークは、低解像度(LR)画像から高解像度(HR)画像へのマッピングを学習することにより、画像の超解像(SR)タスクで卓越したパフォーマンスを発揮しました。ただし、SRの問題は通常、不適切な問題であり、既存の方法にはいくつかの制限があります。まず、同じLR画像にダウンサンプリングできる多くの異なるHR画像が存在する可能性があるため、SRの可能なマッピングスペースは非常に大きくなる可能性があります。その結果、このような広い空間から有望なSRマッピングを直接学ぶことは困難です。第二に、有望なSR性能を生み出すために、非常に高い計算コストで非常に大きなモデルを開発することはしばしば避けられません。実際には、モデル圧縮技術を使用して、モデルの冗長性を減らしてコンパクトなモデルを取得できます。それにもかかわらず、SRマッピングスペースが非常に大きいため、既存のモデル圧縮方法で冗長コンポーネントを正確に識別することは困難です。最初の課題を軽減するために、可能なSRマッピングのスペースを削減するための二重回帰学習スキームを提案します。具体的には、LRからHR画像へのマッピングに加えて、ダウンサンプリングカーネルを推定し、LR画像を再構築するための追加の二重回帰マッピングを学習します。このように、デュアルマッピングは、可能なマッピングのスペースを削減するための制約として機能します。 2番目の課題に対処するために、チャネルプルーニングに基づいてレイヤーレベルとチャネルレベルの両方でモデルの冗長性を減らすための軽量の二重回帰圧縮方法を提案します。具体的には、まず、各層の冗長性を判断するために、二重回帰損失を最小化するチャネル番号検索方法を開発します。検索されたチャネル番号を前提として、二重回帰法をさらに活用して、チャネルの重要性を評価し、冗長なチャネルを削除します。広範な実験は、正確で効率的なSRモデルを取得する上での私たちの方法の有効性を示しています。
Deep neural networks have exhibited remarkable performance in image super-resolution (SR) tasks by learning a mapping from low-resolution (LR) images to high-resolution (HR) images. However, the SR problem is typically an ill-posed problem and existing methods would come with several limitations. First, the possible mapping space of SR can be extremely large since there may exist many different HR images that can be downsampled to the same LR image. As a result, it is hard to directly learn a promising SR mapping from such a large space. Second, it is often inevitable to develop very large models with extremely high computational cost to yield promising SR performance. In practice, one can use model compression techniques to obtain compact models by reducing model redundancy. Nevertheless, it is hard for existing model compression methods to accurately identify the redundant components due to the extremely large SR mapping space. To alleviate the first challenge, we propose a dual regression learning scheme to reduce the space of possible SR mappings. Specifically, in addition to the mapping from LR to HR images, we learn an additional dual regression mapping to estimate the downsampling kernel and reconstruct LR images. In this way, the dual mapping acts as a constraint to reduce the space of possible mappings. To address the second challenge, we propose a lightweight dual regression compression method to reduce model redundancy in both layer-level and channel-level based on channel pruning. Specifically, we first develop a channel number search method that minimizes the dual regression loss to determine the redundancy of each layer. Given the searched channel numbers, we further exploit the dual regression manner to evaluate the importance of channels and prune the redundant ones. Extensive experiments show the effectiveness of our method in obtaining accurate and efficient SR models.
updated: Wed Jul 27 2022 21:54:54 GMT+0000 (UTC)
published: Sat Jul 16 2022 12:46:10 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アソシエイト