arXiv reaDer
半教師ありビジョントランスフォーマー
Semi-Supervised Vision Transformers
半教師あり画像分類のためのビジョントランスフォーマーのトレーニングを研究します。トランスフォーマーは最近、多数の教師あり学習タスクで印象的なパフォーマンスを示しました。驚くべきことに、ラベル付けされたデータの小さなセットしか利用できない場合、VisionTransformerのパフォーマンスは畳み込みニューラルネットワークよりも大幅に劣ることを示しています。この観察に触発されて、トランスフォーマーストリーム、畳み込みストリーム、およびこれらのストリーム間で知識を共有するために慎重に設計された融合モジュールを含む、共同の半教師あり学習フレームワーク、Semiformerを紹介します。畳み込みストリームは、限定されたラベル付きデータでトレーニングされ、さらに、ラベルなしデータでの変圧器ストリームのトレーニングを監視するための疑似ラベルを生成するために使用されます。 ImageNetでの広範な実験は、Semiformerが75.5%のトップ1精度を達成し、最先端を明確なマージンで上回っていることを示しています。さらに、Semiformerは、最新のトランスフォーマーおよび畳み込みニューラルアーキテクチャのほとんどと互換性のある一般的なフレームワークであることを示します。コードはhttps://github.com/wengzejia1/Semiformerで入手できます。
We study the training of Vision Transformers for semi-supervised image classification. Transformers have recently demonstrated impressive performance on a multitude of supervised learning tasks. Surprisingly, we show Vision Transformers perform significantly worse than Convolutional Neural Networks when only a small set of labeled data is available. Inspired by this observation, we introduce a joint semi-supervised learning framework, Semiformer, which contains a transformer stream, a convolutional stream and a carefully designed fusion module for knowledge sharing between these streams. The convolutional stream is trained on limited labeled data and further used to generate pseudo labels to supervise the training of the transformer stream on unlabeled data. Extensive experiments on ImageNet demonstrate that Semiformer achieves 75.5% top-1 accuracy, outperforming the state-of-the-art by a clear margin. In addition, we show, among other things, Semiformer is a general framework that is compatible with most modern transformer and convolutional neural architectures. Code is available at https://github.com/wengzejia1/Semiformer.
updated: Sun Jul 17 2022 08:25:22 GMT+0000 (UTC)
published: Mon Nov 22 2021 09:28:13 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アソシエイト