arXiv reaDer
小さな深層学習のためのネットワーク拡張
Network Augmentation for Tiny Deep Learning
小さなニューラルネットワークのパフォーマンスを向上させるための新しいトレーニング方法であるネットワーク拡張(NetAug)を紹介します。既存の正則化手法(データ拡張、ドロップアウトなど)は、過剰適合を克服するためにノイズを追加することにより、大規模なニューラルネットワーク(ResNet50など)で多くの成功を収めています。ただし、これらの手法は小さなニューラルネットワークのパフォーマンスに悪影響を与えることがわかりました。小さなモデルのトレーニングは大きなモデルとは異なると主張します。小さなモデルは容量が限られているため、過剰適合ではなく過適合に悩まされる傾向があるため、データを拡張するのではなく、モデルを拡張する必要があります。この問題を軽減するために、NetAugは、データセットまたはネットワークにノイズを挿入する代わりに、ネットワークを拡張します(逆ドロップアウト)。小さなモデルをより大きなモデルに配置し、独立したモデルとして機能することに加えて、より大きなモデルのサブモデルとして機能し、追加の監視を取得するように促します。テスト時には、小さなモデルのみが推論に使用され、推論のオーバーヘッドはゼロになります。画像分類とオブジェクト検出に対するNetAugの有効性を示します。 NetAugは、小さなモデルのパフォーマンスを一貫して向上させ、ImageNetで最大2.1%、自動車で4.3%の精度向上を実現します。 Pascal VOCでは、NetAugは同じ計算コストで2.96%のmAP改善を提供します。
We introduce Network Augmentation (NetAug), a new training method for improving the performance of tiny neural networks. Existing regularization techniques (e.g., data augmentation, dropout) have shown much success on large neural networks (e.g., ResNet50) by adding noise to overcome over-fitting. However, we found these techniques hurt the performance of tiny neural networks. We argue that training tiny models are different from large models: rather than augmenting the data, we should augment the model, since tiny models tend to suffer from under-fitting rather than over-fitting due to limited capacity. To alleviate this issue, NetAug augments the network (reverse dropout) instead of inserting noise into the dataset or the network. It puts the tiny model into larger models and encourages it to work as a sub-model of larger models to get extra supervision, in addition to functioning as an independent model. At test time, only the tiny model is used for inference, incurring zero inference overhead. We demonstrate the effectiveness of NetAug on image classification and object detection. NetAug consistently improves the performance of tiny models, achieving up to 2.1% accuracy improvement on ImageNet, and 4.3% on Cars. On Pascal VOC, NetAug provides 2.96% mAP improvement with the same computational cost.
updated: Sun Oct 17 2021 18:48:41 GMT+0000 (UTC)
published: Sun Oct 17 2021 18:48:41 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アソシエイト