教師なし視覚表現学習のための対照的な学習方法は、驚くべきレベルの転送パフォーマンスに達しています。現在の方法はインスタンスレベルの口実タスクでのみトレーニングされており、高密度のピクセル予測を必要とするダウンストリームタスクには最適ではない可能性がある表現につながるため、対照学習の力はまだ完全には解き放たれていないと主張します。この論文では、密な特徴表現を学習するためのピクセルレベルの口実タスクを紹介します。最初のタスクは、ピクセルレベルで対照学習を直接適用します。さらに、最先端のアプローチを大幅に上回り、より良い結果を生み出すピクセルから伝搬への一貫性タスクを提案します。具体的には、Pascal VOCオブジェクト検出(C4)、COCOオブジェクト検出(FPN / C4)、および2.6 APのResNet-50バックボーンネットワークを使用したCityscapesセマンティックセグメンテーションに転送すると、60.2 AP、41.4 / 40.5 mAP、77.2mIoUを実現します。 0.8 / 1.0mAPおよび1.0mIoUは、インスタンスレベルの対比学習に基づいて構築された以前の最良の方法よりも優れています。さらに、ピクセルレベルの口実タスクは、通常のバックボーンネットワークだけでなく、高密度のダウンストリームタスクに使用されるヘッドネットワークの事前トレーニングにも効果的であり、インスタンスレベルの対照的な方法を補完します。これらの結果は、ピクセルレベルで口実タスクを定義する可能性が高いことを示しており、教師なし視覚表現学習の新しい道を示唆しています。コードはhttps://github.com/zdaxie/PixProで入手できます。
Contrastive learning methods for unsupervised visual representation learning have reached remarkable levels of transfer performance. We argue that the power of contrastive learning has yet to be fully unleashed, as current methods are trained only on instance-level pretext tasks, leading to representations that may be sub-optimal for downstream tasks requiring dense pixel predictions. In this paper, we introduce pixel-level pretext tasks for learning dense feature representations. The first task directly applies contrastive learning at the pixel level. We additionally propose a pixel-to-propagation consistency task that produces better results, even surpassing the state-of-the-art approaches by a large margin. Specifically, it achieves 60.2 AP, 41.4 / 40.5 mAP and 77.2 mIoU when transferred to Pascal VOC object detection (C4), COCO object detection (FPN / C4) and Cityscapes semantic segmentation using a ResNet-50 backbone network, which are 2.6 AP, 0.8 / 1.0 mAP and 1.0 mIoU better than the previous best methods built on instance-level contrastive learning. Moreover, the pixel-level pretext tasks are found to be effective for pre-training not only regular backbone networks but also head networks used for dense downstream tasks, and are complementary to instance-level contrastive methods. These results demonstrate the strong potential of defining pretext tasks at the pixel level, and suggest a new path forward in unsupervised visual representation learning. Code is available at https://github.com/zdaxie/PixPro.