arXiv reaDer
MemNet:Augment-Trim学習によるメモリ効率ガイド付きニューラルアーキテクチャ検索
MemNet: Memory-Efficiency Guided Neural Architecture Search with Augment-Trim learning
自動ニューラルアーキテクチャ検索に関する最近の研究では、手作りのニューラルアーキテクチャと同等以上の優れたパフォーマンスが実証されています。ただし、既存のネットワークアーキテクチャのほとんどは、浅い機能と深い機能の間の残余の並列構造と連結ブロックを使用して大規模なネットワークを構築する傾向があります。これには、重みと機能マップの両方を格納するために大量のメモリが必要です。設計された大規模なネットワークモデルで推論を実行するための十分なメモリがない可能性があるため、モバイルデバイスや組み込みデバイスにとってこれは困難です。このギャップを埋めるために、パフォーマンスだけでなくメモリ要件も最適化する拡張トリム学習ベースのニューラルネットワーク検索フレームワークであるMemNetを提案します。具体的には、検索プロセスをナビゲートするためのメモリ消費量の上限を強制するランク付けスコアに基づくメモリ消費量を採用しています。実験結果は、最先端の効率的な設計方法と比較して、MemNetが競争力のある精度を達成し、必要な総メモリを平均24.17%節約できるアーキテクチャを見つけることができることを示しています。
Recent studies on automatic neural architectures search have demonstrated significant performance, competitive to or even better than hand-crafted neural architectures. However, most of the existing network architecture tend to use residual, parallel structures and concatenation block between shallow and deep features to construct a large network. This requires large amounts of memory for storing both weights and feature maps. This is challenging for mobile and embedded devices since they may not have enough memory to perform inference with the designed large network model. To close this gap, we propose MemNet, an augment-trim learning-based neural network search framework that optimizes not only performance but also memory requirement. Specifically, it employs memory consumption based ranking score which forces an upper bound on memory consumption for navigating the search process. Experiment results show that, as compared to the state-of-the-art efficient designing methods, MemNet can find an architecture which can achieve competitive accuracy and save an average of 24.17% on the total memory needed.
updated: Wed Jun 10 2020 20:12:57 GMT+0000 (UTC)
published: Mon Jul 22 2019 20:49:53 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アソシエイト