arXiv reaDer
効率的なモバイルネットワーク設計のためのボトルネック構造の再考
Rethinking Bottleneck Structure for Efficient Mobile Network Design
最近、反転残差ブロックがモバイルネットワークのアーキテクチャ設計を支配しています。これは、反転残差の学習と線形ボトルネックの使用という2つの設計ルールを導入することで、古典的な残差ボトルネックを変更します。このホワイトペーパーでは、そのような設計変更の必要性を再考し、情報の損失や勾配の混乱のリスクをもたらす可能性があることを発見しました。したがって、構造を反転させ、砂時計ブロックと呼ばれる新しいボトルネック設計を提示することを提案します。これは、高次元でアイデンティティマッピングと空間変換を実行し、情報の損失と勾配の混乱を効果的に緩和します。広範な実験は、一般的な信念とは異なり、このようなボトルネック構造は、モバイルネットワークの逆のものよりも有益であることを示しています。 ImageNet分類では、パラメーターと計算を増やすことなく、反転残差ブロックを砂時計ブロックに置き換えるだけで、分類精度をMobileNetV2よりも1.7%以上向上させることができます。 Pascal VOC 2007テストセットでは、オブジェクト検出においても0.9%のmAP改善が見られます。砂時計ブロックをニューラルアーキテクチャ検索メソッドDARTSの検索スペースに追加して、砂時計ブロックの有効性をさらに検証します。 25%のパラメーター削減により、分類精度は以前のDARTSモデルに比べて0.13%向上しました。コードは、https://github.com/zhoudaquan/rethinking_bottleneck_designにあります。
The inverted residual block is dominating architecture design for mobile networks recently. It changes the classic residual bottleneck by introducing two design rules: learning inverted residuals and using linear bottlenecks. In this paper, we rethink the necessity of such design changes and find it may bring risks of information loss and gradient confusion. We thus propose to flip the structure and present a novel bottleneck design, called the sandglass block, that performs identity mapping and spatial transformation at higher dimensions and thus alleviates information loss and gradient confusion effectively. Extensive experiments demonstrate that, different from the common belief, such bottleneck structure is more beneficial than the inverted ones for mobile networks. In ImageNet classification, by simply replacing the inverted residual block with our sandglass block without increasing parameters and computation, the classification accuracy can be improved by more than 1.7% over MobileNetV2. On Pascal VOC 2007 test set, we observe that there is also 0.9% mAP improvement in object detection. We further verify the effectiveness of the sandglass block by adding it into the search space of neural architecture search method DARTS. With 25% parameter reduction, the classification accuracy is improved by 0.13% over previous DARTS models. Code can be found at: https://github.com/zhoudaquan/rethinking_bottleneck_design.
updated: Fri Nov 27 2020 16:02:39 GMT+0000 (UTC)
published: Sun Jul 05 2020 08:55:26 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アソシエイト