arXiv reaDer
2020年代のより多くのConvNets:スパース性を使用して51x51を超えるカーネルをスケールアップ
More ConvNets in the 2020s: Scaling up Kernels Beyond 51x51 using Sparsity
トランスフォーマーは、ビジョントランスフォーマー(ViT)の登場以来、コンピュータービジョンの世界で急速に輝いています。畳み込みニューラルネットワーク(CNN)の主な役割は、ますます効果的なトランスベースのモデルによって挑戦されているようです。ごく最近、いくつかの高度な畳み込みモデルが、ローカルであるが大きな注意メカニズムによって動機付けられた大きなカーネルで反撃し、魅力的なパフォーマンスと効率を示しています。そのうちの1つであるRepLKNetは、パフォーマンスを向上させてカーネルサイズを31x31にスケーリングすることができますが、Swin Transformerなどの高度なViTのスケーリング傾向と比較して、カーネルサイズが拡大し続けると、パフォーマンスは飽和し始めます。このホワイトペーパーでは、31x31を超える極端な畳み込みをトレーニングする可能性を探り、畳み込みを戦略的に拡大することでパフォーマンスのギャップを解消できるかどうかをテストします。この調査は、スパース性の観点から非常に大きなカーネルを適用するためのレシピで終わります。これにより、カーネルを61x61にスムーズにスケールアップし、パフォーマンスを向上させることができます。このレシピに基づいて構築されたSparseLargeKernel Network(SLaK)は、最先端の階層型トランスフォーマーと同等またはそれ以上のパフォーマンスを発揮する51x51カーネルを備えた純粋なCNNアーキテクチャであり、ConvNeXtやRepLKNetなどの最新のConvNetアーキテクチャです。 ImageNet分類と典型的なダウンストリームタスクについて。私たちのコードはhttps://github.com/VITA-Group/SLaKから入手できます。
Transformers have quickly shined in the computer vision world since the emergence of Vision Transformers (ViTs). The dominant role of convolutional neural networks (CNNs) seems to be challenged by increasingly effective transformer-based models. Very recently, a couple of advanced convolutional models strike back with large kernels motivated by the local but large attention mechanism, showing appealing performance and efficiency. While one of them, i.e. RepLKNet, impressively manages to scale the kernel size to 31x31 with improved performance, the performance starts to saturate as the kernel size continues growing, compared to the scaling trend of advanced ViTs such as Swin Transformer. In this paper, we explore the possibility of training extreme convolutions larger than 31x31 and test whether the performance gap can be eliminated by strategically enlarging convolutions. This study ends up with a recipe for applying extremely large kernels from the perspective of sparsity, which can smoothly scale up kernels to 61x61 with better performance. Built on this recipe, we propose Sparse Large Kernel Network (SLaK), a pure CNN architecture equipped with 51x51 kernels that can perform on par with or better than state-of-the-art hierarchical Transformers and modern ConvNet architectures like ConvNeXt and RepLKNet, on ImageNet classification as well as typical downstream tasks. Our code is available here https://github.com/VITA-Group/SLaK.
updated: Thu Jul 07 2022 23:55:52 GMT+0000 (UTC)
published: Thu Jul 07 2022 23:55:52 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アソシエイト