arXiv reaDer
ItNet: iterative neural networks with small graphs for accurate, efficient and anytime semantic segmentation
ディープニューラルネットワークは通常、モバイルなどの低電力デバイスで使用するために圧縮および高速化する必要があります。最近、メモリ内計算を利用することにより、低電力で高スループットと低遅延を提供する超並列ハードウェアアクセラレータが開発されました。ただし、これらの利点を活用するには、ニューラルネットワークの計算グラフを、通常はサイズがかなり制限されているこれらのハードウェアシステムの計算中のメモリに適合させる必要があります。この研究では、計算グラフの観点からメモリフットプリントが小さいクラスのネットワークモデルを紹介します。この目的のために、グラフは、単一のネットワークビルディングブロックを繰り返し実行することによってループを含むように設計されています。さらに、これらのいわゆる反復ニューラルネットワークの精度と遅延の間のトレードオフは、トレーニングと推論の両方で複数の中間出力を追加することによって改善されます。 CamVidおよびCityscapesデータセットでのセマンティックセグメンテーションの最新の結果を示します。これらは、計算リソースの観点から特に要求が厳しいものです。アブレーション研究では、中間ネットワーク出力によるネットワークトレーニングの改善と、反復による重みの共有とネットワークサイズの間のトレードオフが調査されます。
Deep neural networks have usually to be compressed and accelerated for their usage in low-power, e.g. mobile, devices. Recently, massively-parallel hardware accelerators were developed that offer high throughput and low latency at low power by utilizing in-memory computation. However, to exploit these benefits the computational graph of a neural network has to fit into the in-computation memory of these hardware systems that is usually rather limited in size. In this study, we introduce a class of network models that have a small memory footprint in terms of their computational graphs. To this end, the graph is designed to contain loops by iteratively executing a single network building block. Furthermore, the trade-off between accuracy and latency of these so-called iterative neural networks is improved by adding multiple intermediate outputs during both training and inference. We show state-of-the-art results for semantic segmentation on the CamVid and Cityscapes datasets that are especially demanding in terms of computational resources. In ablation studies, the improvement of network training by intermediate network outputs as well as the trade-off between weight sharing over iterations and the network size are investigated.
updated: Tue Aug 03 2021 16:40:36 GMT+0000 (UTC)
published: Thu Jan 21 2021 15:56:29 GMT+0000 (UTC)
参考文献 (このサイトで利用可能なもの) / References (only if available on this site)
被参照文献 (このサイトで利用可能なものを新しい順に) / Citations (only if available on this site, in order of most recent)アソシエイト