arXiv reaDer
コンパクトなセマンティックセグメンテーションアーキテクチャのテンプレートベースの自動検索
Template-Based Automatic Search of Compact Semantic Segmentation Architectures
 さまざまなビジョンと自然言語タスクのニューラルアーキテクチャの自動検索は、関心のある任意のデータセットの高性能構造を検出できるため、卓越したツールになりつつあります。それでも、密度の高いピクセル単位の分類など、より難しいドメインでは、現在の自動アプローチはその範囲が制限されます。既存の画像分類器への依存度が高いため、発見されたアーキテクチャがまだ含まれている追加のレイヤーを少数しか検索しない傾向があります。多数のパラメータ。対照的に、この作業では、事前トレーニング済みの分類ネットワークのほんの数ブロックから始まる軽量で正確なセグメンテーションアーキテクチャを見つけることができる新しいソリューションを提案します。このため、一連の操作のテンプレートに依存し、各テンプレートで適用するテンプレートと回数を予測しながら、接続構造とダウンサンプリング係数も生成する方法論を段階的に構築します。これらすべての決定は、ホールドアウトセットで放出されたアーキテクチャのスコアに基づいて報酬が与えられ、強化学習を使用してトレーニングされるリカレントニューラルネットワークによって行われます。発見されたアーキテクチャの1つは、CamVidで63.2%の平均IoUを達成し、270KのパラメーターしかないCityScapesで67.8%を達成しています。事前トレーニング済みのモデルと検索コードは、https://github.com/DrSleep/nas-segm-pytorchで入手できます。
Automatic search of neural architectures for various vision and natural language tasks is becoming a prominent tool as it allows to discover high-performing structures on any dataset of interest. Nevertheless, on more difficult domains, such as dense per-pixel classification, current automatic approaches are limited in their scope - due to their strong reliance on existing image classifiers they tend to search only for a handful of additional layers with discovered architectures still containing a large number of parameters. In contrast, in this work we propose a novel solution able to find light-weight and accurate segmentation architectures starting from only few blocks of a pre-trained classification network. To this end, we progressively build up a methodology that relies on templates of sets of operations, predicts which template and how many times should be applied at each step, while also generating the connectivity structure and downsampling factors. All these decisions are being made by a recurrent neural network that is rewarded based on the score of the emitted architecture on the holdout set and trained using reinforcement learning. One discovered architecture achieves 63.2% mean IoU on CamVid and 67.8% on CityScapes having only 270K parameters. Pre-trained models and the search code are available at https://github.com/DrSleep/nas-segm-pytorch.
updated: Fri May 22 2020 03:25:48 GMT+0000 (UTC)
published: Thu Apr 04 2019 06:06:32 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アソシエイト