arXiv reaDer
統一された基盤モデルに向けて:対になっていない画像とテキストでトランスフォーマーを共同で事前トレーニングする
Towards a Unified Foundation Model: Jointly Pre-Training Transformers on Unpaired Images and Text
このホワイトペーパーでは、ビジョンのみのタスクとテキストのみのタスクの両方に適応できる統合された基盤モデルを構築する可能性を探ります。 BERTとViTから始めて、モダリティ固有のトークナイザー、共有トランスフォーマーエンコーダー、およびタスク固有の出力ヘッドで構成される統合トランスフォーマーを設計します。ペアになっていない画像とテキストで提案されたモデルを共同で効率的に事前トレーニングするために、2つの新しい手法を提案します。(i)教師として別々にトレーニングされたBERTモデルとViTモデルを採用し、知識蒸留を適用して、ジョイントに追加の正確な監視信号を提供します。トレーニング; (ii)画像とテキストの事前トレーニング損失からのパラメータ更新のバランスをとるための新しい勾配マスキング戦略を提案します。画像分類タスクと自然言語理解タスクでそれぞれ微調整することにより、共同で事前トレーニングされたトランスフォーマーを評価します。実験は、結果として得られた統合基礎変圧器がビジョンのみとテキストのみのタスクの両方で驚くほどうまく機能することを示し、提案された知識蒸留と勾配マスキング戦略は、個別に訓練されたモデルのレベルに近づくためにパフォーマンスを効果的に高めることができます。
In this paper, we explore the possibility of building a unified foundation model that can be adapted to both vision-only and text-only tasks. Starting from BERT and ViT, we design a unified transformer consisting of modality-specific tokenizers, a shared transformer encoder, and task-specific output heads. To efficiently pre-train the proposed model jointly on unpaired images and text, we propose two novel techniques: (i) We employ the separately-trained BERT and ViT models as teachers and apply knowledge distillation to provide additional, accurate supervision signals for the joint training; (ii) We propose a novel gradient masking strategy to balance the parameter updates from the image and text pre-training losses. We evaluate the jointly pre-trained transformer by fine-tuning it on image classification tasks and natural language understanding tasks, respectively. The experiments show that the resultant unified foundation transformer works surprisingly well on both the vision-only and text-only tasks, and the proposed knowledge distillation and gradient masking strategy can effectively lift the performance to approach the level of separately-trained models.
updated: Tue Dec 14 2021 00:20:55 GMT+0000 (UTC)
published: Tue Dec 14 2021 00:20: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アソシエイト