arXiv reaDer
Graph-guided Architecture Search for Real-time Semantic Segmentation
軽量なセマンティックセグメンテーションネットワークを設計する場合、研究者はしばしば性能と速度のトレードオフを見つける必要があるが、ニューラルネットワークの解釈可能性は限られているため、常に経験的なものになる。このような退屈で機械的な試行から研究者を解放するために、我々はリアルタイムのセマンティックセグメンテーションネットワークを自動的に探するためのGraph-guided Architecture Search (GAS)パイプラインを提案する。簡略化された探索空間を用い、繰り返し可能なセルを積み上げてネットワークを形成する従来の研究とは異なり、セルレベルの多様性とレイテンシ指向の制約により、軽量なモデルを効果的に探索できる新しい探索空間を持つ探索機構を導入する。具体的には、セルレベルの多様性を実現するために、セルに依存しない方法でセル共有制約を除去する。次に、グラフ畳み込みネットワーク(GCN)をセル間の通信メカニズムとしてシームレスに統合する。最後に、探索プロセスにレイテンシ指向の制約を与えることで、速度と性能のバランスをとっている。CityscapesとCamVidのデータセットを用いた広範な実験により、GASが精度と速度のトレードオフを実現していることが実証された。特に、Cityscapesデータセットでは、GASはTitan Xp上で108.4FPSの速度で73.5%のmIoUという新たな最高性能を達成した。
Designing a lightweight semantic segmentation network often requires researchers to find a trade-off between performance and speed, which is always empirical due to the limited interpretability of neural networks. In order to release researchers from these tedious mechanical trials, we propose a Graph-guided Architecture Search (GAS) pipeline to automatically search real-time semantic segmentation networks. Unlike previous works that use a simplified search space and stack a repeatable cell to form a network, we introduce a novel search mechanism with new search space where a lightweight model can be effectively explored through the cell-level diversity and latencyoriented constraint. Specifically, to produce the cell-level diversity, the cell-sharing constraint is eliminated through the cell-independent manner. Then a graph convolution network (GCN) is seamlessly integrated as a communication mechanism between cells. Finally, a latency-oriented constraint is endowed into the search process to balance the speed and performance. Extensive experiments on Cityscapes and CamVid datasets demonstrate that GAS achieves the new state-of-the-art trade-off between accuracy and speed. In particular, on Cityscapes dataset, GAS achieves the new best performance of 73.5% mIoU with speed of 108.4 FPS on Titan Xp.
updated: Wed Apr 01 2020 04:38:55 GMT+0000 (UTC)
published: Sun Sep 15 2019 12:45:55 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)アソシエイト