arXiv reaDer
自己回帰コーディング (RandSAC) を使用したランダム セグメントによる自己監視
Self-supervision through Random Segments with Autoregressive Coding (RandSAC)
自然言語 (GPT とそのバリアント) での自己教師あり自己回帰表現学習の成功と、Vision Transformers (ViTs) を使用した最近のビジュアル アーキテクチャ設計の進歩に触発されて、このホワイト ペーパーでは、さまざまな設計の選択が成功に及ぼす影響を探ります。視覚的特徴学習にそのようなトレーニング戦略を適用する方法。具体的には、自己回帰コーディングを使用したランダム セグメント (RandSAC) と呼ばれる新しい戦略を紹介します。 RandSAC では、パッチ表現 (イメージ トークン) を階層的に配置されたセグメントにグループ化します。各セグメント内では、トークンはBERTと同様に並行して予測されますが、セグメント全体の予測はGPTと同様に順次です。セグメントのランダム化されたシリアル化により、パフォーマンスが大幅に向上し、特徴学習に効果的な空間的に長い (セグメント全体) および短い (セグメント内) 予測での分布が得られることを示します。これらの設計選択の妥当性を説明し、多数のデータセット (CIFAR10、CIFAR100、ImageNet など) の代替案を検討します。私たちの事前トレーニング戦略は通常の Transformer で機能しますが、概念的には単純ですが非常に効果的な、エンコーダーの機能レイヤーへの学習可能なスキップ接続を可能にするデコーダーへの追加も提案します。これにより、パフォーマンスがさらに向上します。
Inspired by the success of self-supervised autoregressive representation learning in natural language (GPT and its variants), and advances in recent visual architecture design with Vision Transformers (ViTs), in this paper, we explore the effect various design choices have on the success of applying such training strategies for visual feature learning. Specifically, we introduce a novel strategy that we call Random Segments with Autoregressive Coding (RandSAC). In RandSAC, we group patch representations (image tokens) into hierarchically arranged segments; within each segment, tokens are predicted in parallel, similar to BERT, while across segment predictions are sequential, similar to GPT. We illustrate that randomized serialization of the segments significantly improves the performance and results in distribution over spatially-long (across-segments) and -short (within-segment) predictions which are effective for feature learning. We illustrate the pertinence of these design choices and explore alternatives on a number of datasets (e.g., CIFAR10, CIFAR100, ImageNet). While our pre-training strategy works with a vanilla Transformer, we also propose a conceptually simple, but highly effective, addition to the decoder that allows learnable skip-connections to encoder's feature layers, which further improves the performance.
updated: Wed Oct 26 2022 03:59:43 GMT+0000 (UTC)
published: Tue Mar 22 2022 21:28:55 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アソシエイト