arXiv reaDer
ICネットワーク:畳み込みニューラルネットワークの基本ユニットの改造
IC Networks: Remodeling the Basic Unit for Convolutional Neural Networks
畳み込みニューラルネットワーク(CNN)は、コンピュータービジョンタスクで広く使用されている人工ニューラルネットワークのクラスです。ほとんどのCNNは、特定のタイプの基本ユニットをスタックすることで優れたパフォーマンスを実現します。ネットワークの深さと幅を広げることに加えて、より効果的な基本ユニットを設計することは重要な研究トピックになっています。物理学の弾性衝突モデルに触発されて、既存のCNNに統合してパフォーマンスを向上させることができる一般的な構造を提示します。これを「層間衝突」(IC)構造と呼びます。従来の畳み込み構造と比較して、IC構造は線形畳み込み操作に非線形性と特徴の再キャリブレーションを導入し、よりきめ細かい特徴をキャプチャできます。さらに、事前にトレーニングされた基本モデルから知識を抽出することによってICネットワークのトレーニングを高速化するために、新しいトレーニング方法、つまり弱いロジット蒸留(WLD)が提案されています。 ImageNet実験では、IC構造をResNet-50に統合し、上位1エラーを22.38%から21.75%に減らします。これにより、ResNet-100の上位1エラー(21.75%)もFLOPのほぼ半分でキャッチされます。 。
Convolutional neural network (CNN) is a class of artificial neural networks widely used in computer vision tasks. Most CNNs achieve excellent performance by stacking certain types of basic units. In addition to increasing the depth and width of the network, designing more effective basic units has become an important research topic. Inspired by the elastic collision model in physics, we present a general structure which can be integrated into the existing CNNs to improve their performance. We term it the "Inter-layer Collision" (IC) structure. Compared to the traditional convolution structure, the IC structure introduces nonlinearity and feature recalibration in the linear convolution operation, which can capture more fine-grained features. In addition, a new training method, namely weak logit distillation (WLD), is proposed to speed up the training of IC networks by extracting knowledge from pre-trained basic models. In the ImageNet experiment, we integrate the IC structure into ResNet-50 and reduce the top-1 error from 22.38% to 21.75%, which also catches up the top-1 error of ResNet-100 (21.75%) with nearly half of FLOPs.
updated: Sat Feb 06 2021 03:15:43 GMT+0000 (UTC)
published: Sat Feb 06 2021 03:15:43 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アソシエイト