Mask Hierarchical Features For Self-Supervised Learning

Fenggang Liu,Yangguang Li,Feng Liang,Jilan Xu,Bin Huang,Jing Shao
2023-04-01
Abstract:This paper shows that Masking the Deep hierarchical features is an efficient self-supervised method, denoted as MaskDeep. MaskDeep treats each patch in the representation space as an independent instance. We mask part of patches in the representation space and then utilize sparse visible patches to reconstruct high semantic image representation. The intuition of MaskDeep lies in the fact that models can reason from sparse visible patches semantic to the global semantic of the image. We further propose three designs in our framework: 1) a Hierarchical Deep-Masking module to concern the hierarchical property of patch representations, 2) a multi-group strategy to improve the efficiency without any extra computing consumption of the encoder and 3) a multi-target strategy to provide more description of the global semantic. Our MaskDeep brings decent improvements. Trained on ResNet50 with 200 epochs, MaskDeep achieves state-of-the-art results of 71.2% Top1 accuracy linear classification on ImageNet. On COCO object detection tasks, MaskDeep outperforms the self-supervised method SoCo, which specifically designed for object detection. When trained with 100 epochs, MaskDeep achieves 69.6% Top1 accuracy, which surpasses current methods trained with 200 epochs, such as HCSC, by 0.4% .
Computer Vision and Pattern Recognition,Artificial Intelligence
What problem does this paper attempt to address?
### What problem does this paper attempt to solve? This paper aims to solve the problem of how to effectively use local information to construct global image representations in self - supervised learning. Specifically, the authors propose a new self - supervised learning method - MaskDeep. By masking deep - level features in the feature space and using sparsely visible local features to reconstruct high - semantic image representations. #### Main problems: 1. **Improve the effectiveness of self - supervised learning**: Existing self - supervised learning methods such as contrastive learning and masked image modeling (MIM) perform well on some tasks, but there is still room for improvement. In particular, how to improve model performance while reducing computational overhead is an important challenge. 2. **Integrate local and global information**: Many existing methods focus either on global information or on local information, and lack a method that can effectively integrate both simultaneously. The goal of this paper is to enable the model to infer global semantics from sparse local features through reasonable framework design. #### Solutions: - **MaskDeep**: This method masks patches in the representation space and uses sparsely visible patches to reconstruct high - level image representations. The core idea is to infer global image content through reasoning about the relationships between local features. - **Hierarchical Deep - Masking Module**: To consider the hierarchy of feature representations, a hierarchical deep - masking module based on the FPN module is introduced to generate a hierarchical representation space. - **Multi - group Strategy**: By sampling sparse local features multiple times without increasing the computational consumption of the encoder, the training efficiency is improved. - **Multi - target Strategy**: By having the joint patch representation predict multiple global targets, more supervision signals are provided to maintain the stability of training. ### Experimental results: - On the ImageNet linear classification task, after 200 epochs of training, MaskDeep achieved a Top - 1 accuracy of 71.2%, surpassing current self - supervised learning methods. - On the COCO object detection task, MaskDeep also performed well, outperforming SoCo, a self - supervised method specifically designed for object detection. In summary, this paper solves the problem of how to efficiently use local information to construct global image representations by introducing MaskDeep and achieves significant performance improvements on multiple downstream tasks.