Disjoint Masking with Joint Distillation for Efficient Masked Image Modeling
Not on the map. Why?
Abstract (arXiv)
Masked image modeling (MIM) has shown great promise for self-supervised learning (SSL) yet been criticized for learning inefficiency. We believe the insufficient utilization of training signals should be responsible. To alleviate this issue, we introduce a conceptually simple yet learning-efficient MIM training scheme, termed Disjoint Masking with Joint Distillation (DMJD). For disjoint masking (DM), we sequentially sample multiple masked views per image in a mini-batch with the disjoint regulation to raise the usage of tokens for reconstruction in each image while keeping the masking rate of each view. For joint distillation (JD), we adopt a dual branch architecture to respectively predict invisible (masked) and visible (unmasked) tokens with superior learning targets. Rooting in orthogonal perspectives for training efficiency improvement, DM and JD cooperatively accelerate the training convergence yet not sacrificing the model generalization ability. Concretely, DM can train ViT with half of the effective training epochs (3.7 times less time-consuming) to report competitive performance. With JD, our DMJD clearly improves the linear probing classification accuracy over ConvMAE by 5.8%. On fine-grained downstream tasks like semantic segmentation, object detection, etc., our DMJD also presents superior generalization compared with state-of-the-art SSL methods. The code and model will be made public at https://github.com/mx-mark/DMJD.
Key facts. Cited by 6 papers in the map (data through 14 Oct 2026). Builds on 55 of its 67 references that are in the map. First posted on arXiv 31 Dec 2022.
Builds on
55 of 67 references.
References · 55
- 2017Attention Is All You Needcited by 60,361
- 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,744
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
- 2014Microsoft COCO: Common Objects in Contextcited by 17,608
- 2021Swin Transformer: Hierarchical Vision Transformer using Shifted Windowscited by 12,139
- 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 12,049
- 2021Masked Autoencoders Are Scalable Vision Learnerscited by 7,403
- 2017Mask R-CNNcited by 7,277
- 2021Emerging Properties in Self-Supervised Vision Transformerscited by 6,968
- 2016Layer Normalizationcited by 5,669
- 2016SGDR: Stochastic Gradient Descent with Warm Restartscited by 5,045
- 2020Training data-efficient image transformers & distillation through attentioncited by 4,586
38 more
- 2021Zero-Shot Text-to-Image Generationcited by 4,288
- 2021BEiT: BERT Pre-Training of Image Transformerscited by 2,375
- 2021An Empirical Study of Training Self-Supervised Vision Transformerscited by 1,616
- 2017Accurate, Large Minibatch SGD: Training ImageNet in 1 Hourcited by 1,492
- 2016On Large-Batch Training for Deep Learning: Generalization Gap and Sharp Minimacited by 1,352
- 2018Unified Perceptual Parsing for Scene Understandingcited by 1,271
- 2021SimMIM: A Simple Framework for Masked Image Modelingcited by 1,153
- 2021Tokens-to-Token ViT: Training Vision Transformers from Scratch on ImageNetcited by 1,101
- 2021iBOT: Image BERT Pre-Training with Online Tokenizercited by 903
- 2021Going deeper with Image Transformerscited by 764
- 2022data2vec: A General Framework for Self-supervised Learning in Speech, Vision and Languagecited by 727
- 2021Masked Feature Prediction for Self-Supervised Visual Pre-Trainingcited by 576
- 2017Large Batch Training of Convolutional Networkscited by 535
- 2022Image as a Foreign Language: BEiT Pretraining for All Vision and Vision-Language Taskscited by 511
- 2022BEiT v2: Masked Image Modeling with Vector-Quantized Visual Tokenizerscited by 303
- 2022Context Autoencoder for Self-Supervised Representation Learningcited by 297
- 2017Train longer, generalize better: closing the generalization gap in large batch training of neural networkscited by 279
- 2021PeCo: Perceptual Codebook for BERT Pre-training of Vision Transformerscited by 204
- 2021Efficient Self-supervised Vision Transformers for Representation Learningcited by 149
- 2022Should You Mask 15% in Masked Language Modeling?cited by 118
- 2022What to Hide from Your Students: Attention-Guided Masked Image Modelingcited by 118
- 2021Are Large-scale Datasets Necessary for Self-Supervised Pre-training?cited by 117
- 2021MST: Masked Self-Supervised Transformer for Visual Representationcited by 114
- 2022SemMAE: Semantic-Guided Masking for Learning Masked Autoencoderscited by 114
- 2022ConvMAE: Masked Convolution Meets Masked Autoencoderscited by 105
- 2022Contrastive Learning Rivals Masked Image Modeling in Fine-tuning via Feature Distillationcited by 93
- 2022MVP: Multimodality-guided Visual Pre-trainingcited by 92
- 2021An Image is Worth 16x16 Words, What is a Video Worth?cited by 87
- 2019MultiGrain: a unified image embedding for classes and instancescited by 85
- 2022Adversarial Masking for Self-Supervised Learningcited by 78
- 2022MILAN: Masked Image Pretraining on Language Assisted Representationcited by 69
- 2018Highly Scalable Deep Learning Training System with Mixed-Precision: Training ImageNet in Four Minutescited by 63
- 2022Uniform Masking: Enabling MAE Pre-training for Pyramid-based Vision Transformers with Localitycited by 57
- 2022SdAE: Self-distillated Masked Autoencodercited by 52
- 2022Green Hierarchical Vision Transformer for Masked Image Modelingcited by 50
- 2021Drawing Multiple Augmentation Samples Per Image During Training Efficiently Decreases Test Errorcited by 33
- 2022mc-BEiT: Multi-choice Discretization for Image BERT Pre-trainingcited by 31
- 2022Efficient Self-supervised Vision Pretraining with Local Masked Reconstructioncited by 27
50 most cited of 55 All 55 (JSON)
Built on by
Citing papers per year: 2023: 3; 2024: 1; 2025: 2.
6 papers.
- 2023CM-MaskSD: Cross-Modality Masked Self-Distillation for Referring Image Segmentationcited by 15
- 2023Masked Modeling for Self-supervised Representation Learning on Vision and Beyondcited by 13
- 2025Linguistics-aware Masked Image Modeling for Self-supervised Scene Text Recognitioncited by 6
- 2023Local Masking Meets Progressive Freezing: Crafting Efficient Vision Transformers for Self-Supervised Learningcited by 1
- 2024StyleInject: Parameter Efficient Tuning of Text-to-Image Diffusion Modelscited by 1
- 2025Automated Learning of Semantic Embedding Representations for Diffusion Modelscited by 0
- 2025Automated Learning of Semantic Embedding Representations for Diffusion Models9 May 2025
- 2025Linguistics-aware Masked Image Modeling for Self-supervised Scene Text Recognition24 Mar 2025
- 2024StyleInject: Parameter Efficient Tuning of Text-to-Image Diffusion Models25 Jan 2024
- 2023Masked Modeling for Self-supervised Representation Learning on Vision and Beyond31 Dec 2023
- 2023Local Masking Meets Progressive Freezing: Crafting Efficient Vision Transformers for Self-Supervised Learning2 Dec 2023
- 2023CM-MaskSD: Cross-Modality Masked Self-Distillation for Referring Image Segmentation19 May 2023