arXiv reaDer
動的ネットワークプルーニングのための機能ゲート結合
Feature-Gate Coupling for Dynamic Network Pruning
ゲーティングモジュールは、特徴の表現を維持しながら、ディープニューラルネットワークの実行時の計算コストを削減するために、動的ネットワークプルーニングで広く検討されてきました。実質的な進歩にもかかわらず、既存の方法は、機能とゲート分布の間の一貫性を無視し続けており、ゲート機能の歪みにつながる可能性があります。この論文では、特徴とゲートの分布を整列させることを目的とした特徴-ゲート結合(FGC)アプローチを提案します。 FGCはプラグアンドプレイモジュールであり、自己監視の方法で繰り返される2つのステップで構成されています。最初のステップでは、FGCは、特徴空間でk最近傍法を利用して、自己監視信号として扱われるインスタンスの近隣関係を調査します。 2番目のステップでは、FGCは対照的な学習を利用して、生成された自己監視信号でゲーティングモジュールを正規化し、フィーチャとゲートスペース内のインスタンスの近隣関係の調整を行います。実験結果は、提案されたFGC法がベースラインアプローチを大幅に改善し、精度と計算のトレードオフを改善して最先端を上回っていることを検証しています。コードは公開されています。
Gating modules have been widely explored in dynamic network pruning to reduce the run-time computational cost of deep neural networks while preserving the representation of features. Despite the substantial progress, existing methods remain ignoring the consistency between feature and gate distributions, which may lead to distortion of gated features. In this paper, we propose a feature-gate coupling (FGC) approach aiming to align distributions of features and gates. FGC is a plug-and-play module, which consists of two steps carried out in an iterative self-supervised manner. In the first step, FGC utilizes the k-Nearest Neighbor method in the feature space to explore instance neighborhood relationships, which are treated as self-supervisory signals. In the second step, FGC exploits contrastive learning to regularize gating modules with generated self-supervisory signals, leading to the alignment of instance neighborhood relationships within the feature and gate spaces. Experimental results validate that the proposed FGC method improves the baseline approach with significant margins, outperforming the state-of-the-arts with better accuracy-computation trade-off. Code is publicly available.
updated: Mon Nov 29 2021 02:16:54 GMT+0000 (UTC)
published: Mon Nov 29 2021 02:16:54 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アソシエイト