arXiv reaDer
ThreshNet:接続を減らすためにしきい値メカニズムを使用する効率的なDenseNet
ThreshNet: An Efficient DenseNet using Threshold Mechanism to Reduce Connections
コンピュータビジョンタスクにおけるニューラルネットワークの継続的な開発により、ますます多くのネットワークアーキテクチャが目覚ましい成功を収めています。最先端のニューラルネットワークアーキテクチャの1つとして、DenseNetは、モデルの深さの問題を解決するためにすべての特徴マップをショートカットします。このネットワークアーキテクチャは、低MAC(乗算と累積)で優れた精度を備えていますが、推論に時間がかかりすぎます。この問題を解決するために、HarDNetはフィーチャマップ間の接続を減らし、残りの接続を高調波に似せます。ただし、この圧縮方法では、モデルの精度が低下し、MACとモデルのサイズが増加する可能性があります。このネットワークアーキテクチャは、メモリアクセス時間を短縮するだけであり、全体的なパフォーマンスを改善する必要があります。したがって、接続方法をさらに最適化するために、しきい値メカニズムを使用した新しいネットワークアーキテクチャを提案します。 ThreshNetの機能マップを圧縮するために、さまざまな畳み込み層のさまざまな数の接続が破棄されます。提案されたネットワークアーキテクチャは、画像分類のパフォーマンスを評価するために、CIFAR-10、CIFAR-100、およびSVHNの3つのデータセットを使用しました。実験結果は、ThreshNetがDenseNetと比較して推論時間を最大60%短縮し、これらのデータセットでHarDNetと比較してトレーニング速度を最大35%短縮し、エラー率を20%削減することを示しています。
With the continuous development of neural networks in computer vision tasks, more and more network architectures have achieved outstanding success. As one of the most advanced neural network architectures, DenseNet shortcuts all feature maps to solve the problem of model depth. Although this network architecture has excellent accuracy at low MACs (multiplications and accumulations), it takes excessive inference time. To solve this problem, HarDNet reduces the connections between feature maps, making the remaining connections resemble harmonic waves. However, this compression method may result in decreasing model accuracy and increasing MACs and model size. This network architecture only reduces the memory access time, its overall performance still needs to be improved. Therefore, we propose a new network architecture using threshold mechanism to further optimize the method of connections. Different numbers of connections for different convolutional layers are discarded to compress the feature maps in ThreshNet. The proposed network architecture used three datasets, CIFAR-10, CIFAR-100, and SVHN, to evaluate the performance for image classifications. Experimental results show that ThreshNet achieves up to 60% reduction in inference time compared to DenseNet, and up to 35% faster training speed and 20% reduction in error rate compared to HarDNet on these datasets.
updated: Sun Jan 09 2022 13:52:16 GMT+0000 (UTC)
published: Sun Jan 09 2022 13:52:16 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アソシエイト