arXiv reaDer
超軽量画像超解像の微分可能なニューラル アーキテクチャ探索
Differentiable Neural Architecture Search for Extremely Lightweight Image Super-Resolution
Single Image Super-Resolution (SISR) タスクは、ディープ ニューラル ネットワークで大幅なパフォーマンスを達成しています。ただし、SISR タスクの CNN ベースの方法の多数のパラメーターには、重い計算が必要です。いくつかの効率的な SISR モデルが最近提案されていますが、ほとんどは手作りであり、柔軟性に欠けています。この作業では、軽量 SISR モデルを検索するために、セル レベルとネットワーク レベルの両方で新しい微分可能なニューラル アーキテクチャ検索 (NAS) アプローチを提案します。具体的には、セルレベルの探索空間を情報蒸留機構に基づいて設計し、軽量演算の組み合わせに着目し、より軽量で精度の高いSR構造の構築を目指します。ネットワーク レベルの検索空間は、セル間の機能の接続を考慮するように設計されており、どの情報フローがセルに最も利益をもたらしてパフォーマンスを向上させるかを見つけることを目的としています。既存の強化学習 (RL) または進化的アルゴリズム (EA) ベースの SISR タスクの NAS メソッドとは異なり、検索パイプラインは完全に微分可能であり、軽量の SISR モデルはセルレベルとネットワークレベルの両方で効率的に検索できます。シングル GPU。実験では、PSNR、SSIM、およびモデルの複雑さに関して、ベンチマーク データセットで、×2 の場合は 68G マルチ追加、×4 SR タスクの場合は 18G マルチ追加のみで、私たちの方法が最先端のパフォーマンスを達成できることが示されています。
Single Image Super-Resolution (SISR) tasks have achieved significant performance with deep neural networks. However, the large number of parameters in CNN-based met-hods for SISR tasks require heavy computations. Although several efficient SISR models have been recently proposed, most are handcrafted and thus lack flexibility. In this work, we propose a novel differentiable Neural Architecture Search (NAS) approach on both the cell-level and network-level to search for lightweight SISR models. Specifically, the cell-level search space is designed based on an information distillation mechanism, focusing on the combinations of lightweight operations and aiming to build a more lightweight and accurate SR structure. The network-level search space is designed to consider the feature connections among the cells and aims to find which information flow benefits the cell most to boost the performance. Unlike the existing Reinforcement Learning (RL) or Evolutionary Algorithm (EA) based NAS methods for SISR tasks, our search pipeline is fully differentiable, and the lightweight SISR models can be efficiently searched on both the cell-level and network-level jointly on a single GPU. Experiments show that our methods can achieve state-of-the-art performance on the benchmark datasets in terms of PSNR, SSIM, and model complexity with merely 68G Multi-Adds for ×2 and 18G Multi-Adds for ×4 SR tasks.
updated: Mon Dec 19 2022 06:48:43 GMT+0000 (UTC)
published: Sun May 09 2021 13:30:16 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アソシエイト