用主成分替代随机像素掩码,提升视觉表征学习效果
From Pixels to Components: Eigenvector Masking for Visual Representation Learning
- 在图像主成分空间中掩码部分分量,而非原始像素
- 在ImageNet上分类准确率比像素掩码提升1.8%
- 适合需要高阶语义特征的自监督学习场景
预测图像中被遮蔽部分的可见信息是一种强大的自监督视觉表征学习方法。然而,传统的随机像素块掩码存在局限性,可能阻碍有意义高层特征的学习。本文提出一种新掩码策略:先对数据进行主成分分析(PCA),再随机掩码占数据方差固定比例的若干主成分,学习任务变为从可见成分重建被掩码成分。相比局部像素块,主成分携带更多全局信息。实验证明,该方法在图像分类任务中优于像素掩码,在ImageNet上准确率提升1.8%。该方法简单且数据驱动,为掩码图像建模提供了稳健替代方案。
原文摘要 · Abstract (English)
Predicting masked from visible parts of an image is a powerful self-supervised approach for visual representation learning. However, the common practice of masking random patches of pixels exhibits certain failure modes, which can prevent learning meaningful high-level features, as required for downstream tasks. We propose an alternative masking strategy that operates on a suitable transformation of the data rather than on the raw pixels. Specifically, we perform principal component analysis and then randomly mask a subset of components, which accounts for a fixed ratio of the data variance. The learning task then amounts to reconstructing the masked components from the visible ones. Compared to local patches of pixels, the principal components of images carry more global information. We thus posit that predicting masked from visible components involves more high-level features, allowing our masking strategy to extract more useful representations. This is corroborated by our empirical findings which demonstrate improved image classification performance for component over pixel masking. Our method thus constitutes a simple and robust data-driven alternative to traditional masked image modeling approaches.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。