arXiv reaDer
単一画像の超解像のための局所選択的特徴蒸留
Local-Selective Feature Distillation for Single Image Super-Resolution
畳み込みニューラルネットワーク(CNN)ベースの単一画像超解像(SISR)手法の最近の改善は、回帰損失を最小限に抑える以外に適切なトレーニングアルゴリズムを見つけるのではなく、ネットワークアーキテクチャの製造に大きく依存しています。知識蒸留(KD)を採用することで、SISRをさらに改善する道が開かれ、モデルの効率の観点からも有益です。 KDは、テスト用の追加パラメーターを使用せずに、ディープニューラルネットワーク(DNN)のパフォーマンスを向上させるモデル圧縮方法です。最近、容量とパフォーマンスのトレードオフを改善する能力が脚光を浴びています。本論文では、SISRに適した新しい特徴蒸留(FD)法を提案した。 SISRタスクで発生する既存のFitNetベースのFD法の制限を示し、ローカル機能情報に焦点を合わせるように既存のFDアルゴリズムを変更することを提案します。さらに、特定のピクセル位置に選択的に焦点を合わせて特徴情報を抽出する、教師と生徒の違いに基づくソフト特徴注意法を提案します。私たちの方法を局所選択的特徴蒸留(LSFD)と呼び、私たちの方法がSISR問題において従来のFD法よりも優れていることを確認します。
Recent improvements in convolutional neural network (CNN)-based single image super-resolution (SISR) methods rely heavily on fabricating network architectures, rather than finding a suitable training algorithm other than simply minimizing the regression loss. Adapting knowledge distillation (KD) can open a way for bringing further improvement for SISR, and it is also beneficial in terms of model efficiency. KD is a model compression method that improves the performance of Deep Neural Networks (DNNs) without using additional parameters for testing. It is getting the limelight recently for its competence at providing a better capacity-performance tradeoff. In this paper, we propose a novel feature distillation (FD) method which is suitable for SISR. We show the limitations of the existing FitNet-based FD method that it suffers in the SISR task, and propose to modify the existing FD algorithm to focus on local feature information. In addition, we propose a teacher-student-difference-based soft feature attention method that selectively focuses on specific pixel locations to extract feature information. We call our method local-selective feature distillation (LSFD) and verify that our method outperforms conventional FD methods in SISR problems.
updated: Mon Nov 22 2021 05:05:37 GMT+0000 (UTC)
published: Mon Nov 22 2021 05:05:37 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アソシエイト