arXiv reaDer
Memory-Efficient Hierarchical Neural Architecture Search for Image Denoising
 最近、ニューラルアーキテクチャ検索(NAS)メソッドは多くの注目を集めており、いくつかの高度な視覚タスクで手動で設計されたアーキテクチャよりも優れています。この論文では、画像のノイズ除去に効果的なニューラルネットワークアーキテクチャを自動的に設計するためにNASを採用するための取り組みであるHiNAS(Hierarchical NAS)を提案します。 HiNASは勾配ベースの検索戦略を採用し、適応型受容野での操作を採用して、柔軟な階層検索スペースを構築します。検索段階では、HiNASは異なる機能レベルでセルを共有してメモリを節約し、NASの崩壊の問題を回避し、検索速度を大幅に加速するために早期停止戦略を採用します。提案されたHiNASは、メモリと計算の両方の効率が高く、単一のGPUを使用して検索するのに約4.5時間しかかかりません。 2つの異なるデータセット、つまり加法性ホワイトガウスノイズデータセットBSD500と現実的なノイズデータセットSIM1800で提案されたHiNASの有効性を評価します。実験結果は、HiNASで見つかったアーキテクチャのパラメーターが少なく、推論速度が速い一方で、最先端の方法と比較して非常に競争力のあるパフォーマンスを実現していることを示しています。また、NASで見つかったアーキテクチャの分析も紹介します。 HiNASはまた、画像の降雨の実験でも良好なパフォーマンスを示します。
Recently, neural architecture search (NAS) methods have attracted much attention and outperformed manually designed architectures on a few high-level vision tasks. In this paper, we propose HiNAS (Hierarchical NAS), an effort towards employing NAS to automatically design effective neural network architectures for image denoising. HiNAS adopts gradient based search strategies and employs operations with adaptive receptive field to build an flexible hierarchical search space. During the search stage, HiNAS shares cells across different feature levels to save memory and employ an early stopping strategy to avoid the collapse issue in NAS, and considerably accelerate the search speed. The proposed HiNAS is both memory and computation efficient, which takes only about 4.5 hours for searching using a single GPU. We evaluate the effectiveness of our proposed HiNAS on two different datasets, namely an additive white Gaussian noise dataset BSD500, and a realistic noise dataset SIM1800. Experimental results show that the architecture found by HiNAS has fewer parameters and enjoys a faster inference speed, while achieving highly competitive performance compared with state-of-the-art methods. We also present analysis on the architectures found by NAS. HiNAS also shows good performance on experiments for image de-raining.
updated: Fri Feb 07 2020 04:41:32 GMT+0000 (UTC)
published: Wed Sep 18 2019 06:49:19 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)アソシエイト