arXiv reaDer
畳み込みネットワーク用の BERT の設計: スパースおよび階層マスク モデリング
Designing BERT for Convolutional Networks: Sparse and Hierarchical Masked Modeling
BERT スタイルの事前トレーニングまたはマスクされた画像モデリングの成功を畳み込みネットワーク (convnet) に拡張する際の 2 つの主要な障害を特定して克服します。 (ii) BERT 事前トレーニングの単一規模の性質は、convnet の階層構造と矛盾しています。 (i) については、マスクされていないピクセルを 3D 点群の疎なボクセルとして扱い、疎な畳み込みを使用してエンコードします。これは、2D マスク モデリングにスパース畳み込みを初めて使用したものです。 (ii)については、マルチスケールでエンコードされた特徴から画像を再構築するための階層デコーダーを開発します。スパース マスク モデリング (SparK) と呼ばれる私たちの方法は一般的です。バックボーンを変更することなく、任意の畳み込みモデルで直接使用できます。従来のモデル (ResNet) と最新のモデル (ConvNeXt) の両方で検証しました。3 つのダウンストリーム タスクで、最先端の対照学習とトランスフォーマー ベースのマスク モデリングの両方を、同様に大きなマージン (約 +1.0%) で上回っています。オブジェクトの検出とインスタンスのセグメンテーションの改善はより大幅 (最大 +3.5%) であり、学習した機能の強力な転送可能性を検証しています。また、より大きなモデルでより多くのゲインを観察することにより、その好ましいスケーリング動作を見つけます。このすべての証拠は、convnets での生成的な事前トレーニングの有望な未来を明らかにしています。コードとモデルは https://github.com/keyu-tian/SparK で公開されています。
We identify and overcome two key obstacles in extending the success of BERT-style pre-training, or the masked image modeling, to convolutional networks (convnets): (i) convolution operation cannot handle irregular, random-masked input images; (ii) the single-scale nature of BERT pre-training is inconsistent with convnet's hierarchical structure. For (i), we treat unmasked pixels as sparse voxels of 3D point clouds and use sparse convolution to encode. This is the first use of sparse convolution for 2D masked modeling. For (ii), we develop a hierarchical decoder to reconstruct images from multi-scale encoded features. Our method called Sparse masKed modeling (SparK) is general: it can be used directly on any convolutional model without backbone modifications. We validate it on both classical (ResNet) and modern (ConvNeXt) models: on three downstream tasks, it surpasses both state-of-the-art contrastive learning and transformer-based masked modeling by similarly large margins (around +1.0%). Improvements on object detection and instance segmentation are more substantial (up to +3.5%), verifying the strong transferability of features learned. We also find its favorable scaling behavior by observing more gains on larger models. All this evidence reveals a promising future of generative pre-training on convnets. Codes and models are released at https://github.com/keyu-tian/SparK.
updated: Tue Jan 10 2023 08:02:09 GMT+0000 (UTC)
published: Mon Jan 09 2023 18:59:50 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アソシエイト