MultiMAE: Multi-modal Multi-task Masked Autoencoders
Not on the map. Why?
Abstract (arXiv)
We propose a pre-training strategy called Multi-modal Multi-task Masked Autoencoders (MultiMAE). It differs from standard Masked Autoencoding in two key aspects: I) it can optionally accept additional modalities of information in the input besides the RGB image (hence "multi-modal"), and II) its training objective accordingly includes predicting multiple outputs besides the RGB image (hence "multi-task"). We make use of masking (across image patches and input modalities) to make training MultiMAE tractable as well as to ensure cross-modality predictive coding is indeed learned by the network. We show this pre-training strategy leads to a flexible, simple, and efficient framework with improved transfer results to downstream tasks. In particular, the same exact pre-trained network can be flexibly used when additional information besides RGB images is available or when no information other than RGB is available - in all configurations yielding competitive to or significantly better results than the baselines. To avoid needing training datasets with multiple modalities and tasks, we train MultiMAE entirely using pseudo labeling, which makes the framework widely applicable to any RGB dataset. The experiments are performed on multiple transfer tasks (image classification, semantic segmentation, depth estimation) and datasets (ImageNet, ADE20K, Taskonomy, Hypersim, NYUv2). The results show an intriguingly impressive capability by the model in cross-modal/task predictive coding and transfer.
Key facts. Cited by 291 papers in the map (data through 14 Oct 2026). Builds on 78 of its 102 references that are in the map. First posted on arXiv 4 Apr 2022.
Builds on
78 of 102 references.
References · 78
- 2017Attention Is All You Needcited by 60,361
- 2018BERT: Pre-training of Deep Bidirectional Transformers for Language Understandingcited by 40,399
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
- 2019PyTorch: An Imperative Style, High-Performance Deep Learning Librarycited by 20,305
- 2014Microsoft COCO: Common Objects in Contextcited by 17,608
- 2021Swin Transformer: Hierarchical Vision Transformer using Shifted Windowscited by 12,139
- 2021Masked Autoencoders Are Scalable Vision Learnerscited by 7,403
- 2021Emerging Properties in Self-Supervised Vision Transformerscited by 6,968
- 2016Layer Normalizationcited by 5,669
- 2017mixup: Beyond Empirical Risk Minimizationcited by 5,183
- 2016SGDR: Stochastic Gradient Descent with Warm Restartscited by 5,045
- 2016Feature Pyramid Networks for Object Detectioncited by 5,036
38 more
- 2020Bootstrap your own latent: A new approach to self-supervised Learningcited by 4,750
- 2020Training data-efficient image transformers & distillation through attentioncited by 4,586
- 2022A ConvNet for the 2020scited by 3,579
- 2019CutMix: Regularization Strategy to Train Strong Classifiers with Localizable Featurescited by 2,884
- 2019Benchmarking Neural Network Robustness to Common Corruptions and Perturbationscited by 2,877
- 2021BEiT: BERT Pre-Training of Image Transformerscited by 2,375
- 2021Masked-attention Mask Transformer for Universal Image Segmentationcited by 2,351
- 2019ViLBERT: Pretraining Task-Agnostic Visiolinguistic Representations for Vision-and-Language Taskscited by 2,222
- 2020The Many Faces of Robustness: A Critical Analysis of Out-of-Distribution Generalizationcited by 1,850
- 2021Vision Transformers for Dense Predictioncited by 1,803
- 2021An Empirical Study of Training Self-Supervised Vision Transformerscited by 1,616
- 2019LXMERT: Learning Cross-Modality Encoder Representations from Transformerscited by 1,553
- 2019Towards Robust Monocular Depth Estimation: Mixing Datasets for Zero-shot Cross-dataset Transfercited by 1,495
- 2017Accurate, Large Minibatch SGD: Training ImageNet in 1 Hourcited by 1,492
- 2016Context Encoders: Feature Learning by Inpaintingcited by 1,428
- 2019Natural Adversarial Examplescited by 1,408
- 2020Rethinking Semantic Segmentation from a Sequence-to-Sequence Perspective with Transformerscited by 1,369
- 2019Self-training with Noisy Student improves ImageNet classificationcited by 1,311
- 2018Unified Perceptual Parsing for Scene Understandingcited by 1,271
- 2021ViLT: Vision-and-Language Transformer Without Convolution or Region Supervisioncited by 1,230
- 2021SimMIM: A Simple Framework for Masked Image Modelingcited by 1,153
- 2019Learning Robust Global Representations by Penalizing Local Predictive Powercited by 1,013
- 2016Deep Networks with Stochastic Depthcited by 979
- 2014Deep Visual-Semantic Alignments for Generating Image Descriptionscited by 968
- 2019VL-BERT: Pre-training of Generic Visual-Linguistic Representationscited by 954
- 2021iBOT: Image BERT Pre-Training with Online Tokenizercited by 903
- 2021Segmenter: Transformer for Semantic Segmentationcited by 855
- 2021MDETR -- Modulated Detection for End-to-End Multi-Modal Understandingcited by 819
- 2022data2vec: A General Framework for Self-supervised Learning in Speech, Vision and Languagecited by 727
- 2018Taskonomy: Disentangling Task Transfer Learningcited by 634
- 2021Perceiver IO: A General Architecture for Structured Inputs & Outputscited by 629
- 2020Hypersim: A Photorealistic Synthetic Dataset for Holistic Indoor Scene Understandingcited by 604
- 2020Simple Copy-Paste is a Strong Data Augmentation Method for Instance Segmentationcited by 596
- 2021Masked Feature Prediction for Self-Supervised Visual Pre-Trainingcited by 576
- 2014Predicting Depth, Surface Normals and Semantic Labels with a Common Multi-Scale Convolutional Architecturecited by 483
- 2017Look, Listen and Learncited by 458
- 2020Rethinking Few-Shot Image Classification: a Good Embedding Is All You Need?cited by 437
- 2021VATT: Transformers for Multimodal Self-Supervised Learning from Raw Video, Audio and Textcited by 429
50 most cited of 78 All 78 (JSON)
Built on by
Citing papers per year: 2022: 32; 2023: 77; 2024: 67; 2025: 66; 2026: 49.
291 papers.
Sort:
By area:
- 2023ImageBind: One Embedding Space To Bind Them Allcited by 1,175
- 2022InternVideo: General Video Foundation Models via Generative and Discriminative Learningcited by 447
- 2022SatMAE: Pre-training Transformers for Temporal and Multi-Spectral Satellite Imagerycited by 408
- 2022Images Speak in Images: A Generalist Painter for In-Context Visual Learningcited by 318
- 2022Scaling Language-Image Pre-training via Maskingcited by 317
- 2025MapAnything: Universal Feed-Forward Metric 3D Reconstructioncited by 314
- 2022Real-World Robot Learning with Masked Visual Pre-trainingcited by 245
- 2022Masked World Models for Visual Controlcited by 170
- 2022CroCo v2: Improved Cross-view Completion Pre-training for Stereo Matching and Optical Flowcited by 166
- 2022CroCo: Self-Supervised Pre-training for 3D Vision Tasks by Cross-View Completioncited by 160
- 2022Contrastive Audio-Visual Masked Autoencodercited by 146
- 20234M: Massively Multimodal Masked Modelingcited by 120
38 more
- 2022Vision-Language Pre-training: Basics, Recent Advances, and Future Trendscited by 117
- 2023Delivering Arbitrary-Modal Semantic Segmentationcited by 117
- 2024MMEarth: Exploring Multi-Modal Pretext Tasks For Geospatial Representation Learningcited by 77
- 2023UniPAD: A Universal Pre-training Paradigm for Autonomous Drivingcited by 73
- 2024Sigma: Siamese Mamba Network for Multi-Modal Semantic Segmentationcited by 71
- 2024Improving 2D Feature Representations by 3D-Aware Fine-Tuningcited by 71
- 2024A Touch, Vision, and Language Dataset for Multimodal Alignmentcited by 69
- 2022Masked Vision and Language Modeling for Multi-modal Representation Learningcited by 67
- 2023Identifiability Results for Multimodal Contrastive Learningcited by 61
- 2023DFormer: Rethinking RGBD Representation Learning for Semantic Segmentationcited by 60
- 2022MAViL: Masked Audio-Video Learnerscited by 59
- 2022A Survey on Masked Autoencoder for Self-supervised Learning in Vision and Beyondcited by 58
- 2023Hard Patches Mining for Masked Image Modelingcited by 57
- 2023PonderV2: Pave the Way for 3D Foundation Model with A Universal Pre-training Paradigmcited by 56
- 2022Towards Good Practices for Missing Modality Robust Action Recognitioncited by 54
- 2023PiMAE: Point Cloud and Image Interactive Masked Autoencoders for 3D Object Detectioncited by 53
- 2024Masked Autoencoders for Microscopy are Scalable Learners of Cellular Biologycited by 52
- 20244M-21: An Any-to-Any Vision Model for Tens of Tasks and Modalitiescited by 52
- 2022Audiovisual Masked Autoencoderscited by 49
- 2024What to align in multimodal contrastive learning?cited by 47
- 2024Bridging Remote Sensors with Multisensor Geospatial Foundation Modelscited by 46
- 2024Lift3D Foundation Policy: Lifting 2D Large-Scale Pretrained Models for Robust 3D Robotic Manipulationcited by 46
- 2023TaskExpert: Dynamically Assembling Multi-Task Representations with Memorial Mixture-of-Expertscited by 45
- 2024Point Cloud Matters: Rethinking the Impact of Different Observation Spaces on Robot Learningcited by 45
- 2024Transferable Tactile Transformers for Representation Learning Across Diverse Sensors and Taskscited by 45
- 2023Traj-MAE: Masked Autoencoders for Trajectory Predictioncited by 42
- 2023An Unbiased Look at Datasets for Visuo-Motor Pre-Trainingcited by 42
- 2022Masked Event Modeling: Self-Supervised Pretraining for Event Camerascited by 41
- 2024vid-TLDR: Training Free Token merging for Light-weight Video Transformercited by 41
- 2025Matrix3D: Large Photogrammetry Model All-in-Onecited by 40
- 2024Cubify Anything: Scaling Indoor 3D Object Detectioncited by 39
- 2023Self-Distilled Masked Auto-Encoders are Efficient Video Anomaly Detectorscited by 37
- 2025TAPNext: Tracking Any Point (TAP) as Next Token Predictioncited by 37
- 2022Unleashing Vanilla Vision Transformer with Masked Image Modeling for Object Detectioncited by 33
- 2022XKD: Cross-modal Knowledge Distillation with Domain Alignment for Video Representation Learningcited by 31
- 2025DFormerv2: Geometry Self-Attention for RGBD Semantic Segmentationcited by 31
- 2024Cross-View Completion Models are Zero-shot Correspondence Estimatorscited by 30
- 2023Robust Multimodal Learning with Missing Modalities via Parameter-Efficient Adaptationcited by 29
50 most cited of 291 All 291 (JSON)
- 2026Toward a Multimodal Foundation Model for Plasma State Reconstruction10 Oct 2026
- 2026On the Geometry of Multimodal Saturation: Riemannian VICReg5 Oct 2026
- 2026DTFormer: Text-Guided Semantic Alignment for RGB-D Segmentation4 Oct 2026
- 2026SynCo: Learning Cross-Modal Synergy by Contrasting Interaction Residuals26 Sep 2026
- 2026RGBD20K: A Large-Scale Benchmark for RGB-D Semantic Segmentation24 Sep 2026
- 2026Depth-Guided Contrastive Learning for 2D Representations with 3D Spatial Awareness23 Sep 2026
- 2026Reading the Whole Heart: Latent-Attention Masked Autoencoders for Multimodal Cardiac Representation Learning10 Sep 2026
- 2026Multimodal-Multiresolution Foundation Model for Lunar Remote Sensing8 Sep 2026
- 2026MEOX: Compact Multimodal Mixture-of-Experts for Earth Observation4 Sep 2026
- 2026DART: Depth-as-Target Pretraining for Surgical Vision Foundation Models3 Sep 2026
- 2026GAAT: Geometry-Aware Alignment Transformer for Multimodal UAV Perception28 Aug 2026
- 2026DINOcular: Self-Supervised Visuospatial Representations27 Aug 2026
38 more
- 2026Drift Variation Autoencoder: Unifying Generation and Representation Learning through Conditional Posterior Flow Matching25 Aug 2026
- 2026Expressivity In Multimodal Contrastive Learning17 Aug 2026
- 2026CM-MAE: A Physics-Guided Cross-Modal Self-Supervised Learning Framework for Vision-Wireless Applications16 Aug 2026
- 2026LunarFM: A Shared Multimodal Representation of the Moon's Surface24 Jul 2026
- 2026Unified Video Dense Prediction from Disjoint Data23 Jul 2026
- 2026Multimodality as Supervision: Self-Supervised Specialization to the Test Environment via Multimodality16 Jul 2026
- 2026Kepler-Encoder-v0.1: Towards a Multimodal Embedding Model for Robots15 Jul 2026
- 2026Weaving Light and Time: Unified Harmonic-Geometric Representation Learning for Dense RGB-Event Parsing10 Jul 2026
- 2026Moonstone: A Multimodal Foundation Model and Benchmark for Lunar Remote Sensing3 Jul 2026
- 2026Learning Spectral and Polarimetric Clues for One-to-Multimodal Novel View Synthesis2 Jul 2026
- 2026OctoSense: Self-Supervised Learning for Multimodal Robot Perception25 Jun 2026
- 2026EyeMVP: OCT-Informed Fundus Representation Learning via Paired CFP--OCT Pretraining13 Jun 2026
- 2026FLORO: A Multimodal Geospatial Foundation Model for Ecological Remote Sensing Across Sensors and Scales27 May 2026
- 2026OmniGF: A Dual-Branch Vision-Language Framework for Unified Gaze Following26 May 2026
- 2026Enhancing Gaze Reasoning in Vision Foundation Models for Gaze Following21 May 2026
- 2026AVI-HT: Adaptive Vision-IMU Fusion for 3D Hand Tracking20 May 2026
- 2026Learning Structural Latent Points for Efficient Visual Representations in Robotic Manipulation20 May 2026
- 2026Better Together: Evaluating the Complementarity of Earth Embedding Models18 May 2026
- 2026RelFlexformer: Efficient Attention 3D-Transformers for Integrable Relative Positional Encodings11 May 2026
- 2026Assessing the Potential of Masked Autoencoder Foundation Models in Predicting Downhole Metrics from Surface Drilling Data16 Apr 2026
- 2026SegRGB-X: General RGB-X Semantic Segmentation Model30 Mar 2026
- 2026From None to All: Self-Supervised 3D Reconstruction via Novel View Synthesis29 Mar 2026
- 2026Le MuMo JEPA: Multi-Modal Self-Supervised Representation Learning with Learnable Fusion Tokens25 Mar 2026
- 2026Self-Distillation of Hidden Layers for Self-Supervised Representation Learning16 Mar 2026
- 2026GLEAM: A Multimodal Imaging Dataset and HAMM for Glaucoma Classification13 Mar 2026
- 2026From Semantics to Pixels: Coarse-to-Fine Masked Autoencoders for Hierarchical Visual Understanding10 Mar 2026
- 2026SIGMAE: A Spectral-Index-Guided Foundation Model for Multispectral Remote Sensing8 Mar 2026
- 2026Unifying Heterogeneous Multi-Modal Remote Sensing Detection Via Language-Pivoted Pretraining2 Mar 2026
- 2026Nonparametric Teaching of Attention Learners24 Feb 2026
- 2026LiDAR-Anchored Collaborative Distillation for Robust 2D Representations13 Feb 2026
- 2026MMEarth-Bench: Global Model Adaptation via Multimodal Test-Time Training6 Feb 2026
- 2026Self-Supervised Learning with a Multi-Task Latent Space Objective5 Feb 2026
- 2026Self-Soupervision: Cooking Model Soups without Labels2 Feb 2026
- 2026Semi-Supervised Masked Autoencoders: Unlocking Vision Transformer Potential with Limited Data27 Jan 2026
- 2026On the Role of Depth in Surgical Vision Foundation Models: An Empirical Study of RGB-D Pre-training26 Jan 2026
- 2026SpaRRTa: A Synthetic Benchmark for Evaluating Spatial Intelligence in Visual Foundation Models16 Jan 2026
- 2026Efficient Prediction of Dense Visual Embeddings via Distillation and RGB-D Transformers1 Jan 2026
- 2025Multi-Modal Soccer Scene Analysis with Masked Pre-Training22 Dec 2025
50 newest of 291 All 291 (JSON)
- 2024DiffusionMTL: Learning Multi-Task Denoising Diffusion Model from Partially Annotated Datacited by 18
- 2025MMGen: Unified Multi-modal Image Generation and Understanding in One Gocited by 11
- 2025JointDiT: Enhancing RGB-Depth Joint Modeling with Diffusion Transformerscited by 11
- 2025PDDM: Pseudo Depth Diffusion Model for RGB-PD Semantic Segmentation Based in Complex Indoor Scenescited by 2
- 2026Drift Variation Autoencoder: Unifying Generation and Representation Learning through Conditional Posterior Flow Matchingcited by 0
- 2026Drift Variation Autoencoder: Unifying Generation and Representation Learning through Conditional Posterior Flow Matching25 Aug 2026
- 2025JointDiT: Enhancing RGB-Depth Joint Modeling with Diffusion Transformers1 May 2025
- 2025MMGen: Unified Multi-modal Image Generation and Understanding in One Go26 Mar 2025
- 2025PDDM: Pseudo Depth Diffusion Model for RGB-PD Semantic Segmentation Based in Complex Indoor Scenes24 Mar 2025
- 2024DiffusionMTL: Learning Multi-Task Denoising Diffusion Model from Partially Annotated Data22 Mar 2024
- 2022Vision-Language Pre-training: Basics, Recent Advances, and Future Trendscited by 117
- 2023Prismer: A Vision-Language Model with Multi-Task Expertscited by 28
- 2023EVE: Efficient Vision-Language Pre-training with Masked Prediction and Modality-Aware MoEcited by 25
- 2025HAWAII: Hierarchical Visual Knowledge Transfer for Efficient Vision-Language Modelscited by 5
- 2026Expressivity In Multimodal Contrastive Learningcited by 0
- 2026Expressivity In Multimodal Contrastive Learning17 Aug 2026
- 2025HAWAII: Hierarchical Visual Knowledge Transfer for Efficient Vision-Language Models23 Jun 2025
- 2023EVE: Efficient Vision-Language Pre-training with Masked Prediction and Modality-Aware MoE23 Aug 2023
- 2023Prismer: A Vision-Language Model with Multi-Task Experts4 Mar 2023
- 2022Vision-Language Pre-training: Basics, Recent Advances, and Future Trends17 Oct 2022
- 2022Masked World Models for Visual Controlcited by 170
- 2022Masked World Models for Visual Control28 Jun 2022
- 2023ImageBind: One Embedding Space To Bind Them Allcited by 1,175
- 2022InternVideo: General Video Foundation Models via Generative and Discriminative Learningcited by 447
- 2022SatMAE: Pre-training Transformers for Temporal and Multi-Spectral Satellite Imagerycited by 408
- 2022Images Speak in Images: A Generalist Painter for In-Context Visual Learningcited by 318
- 2022Scaling Language-Image Pre-training via Maskingcited by 317
- 2025MapAnything: Universal Feed-Forward Metric 3D Reconstructioncited by 314
- 2022Real-World Robot Learning with Masked Visual Pre-trainingcited by 245
- 2022CroCo v2: Improved Cross-view Completion Pre-training for Stereo Matching and Optical Flowcited by 166
- 2022CroCo: Self-Supervised Pre-training for 3D Vision Tasks by Cross-View Completioncited by 160
- 2022Contrastive Audio-Visual Masked Autoencodercited by 146
- 20234M: Massively Multimodal Masked Modelingcited by 120
- 2023Delivering Arbitrary-Modal Semantic Segmentationcited by 117
12 most cited of 267 All 267 (JSON)
- 2026Toward a Multimodal Foundation Model for Plasma State Reconstruction10 Oct 2026
- 2026On the Geometry of Multimodal Saturation: Riemannian VICReg5 Oct 2026
- 2026DTFormer: Text-Guided Semantic Alignment for RGB-D Segmentation4 Oct 2026
- 2026SynCo: Learning Cross-Modal Synergy by Contrasting Interaction Residuals26 Sep 2026
- 2026RGBD20K: A Large-Scale Benchmark for RGB-D Semantic Segmentation24 Sep 2026
- 2026Depth-Guided Contrastive Learning for 2D Representations with 3D Spatial Awareness23 Sep 2026
- 2026Reading the Whole Heart: Latent-Attention Masked Autoencoders for Multimodal Cardiac Representation Learning10 Sep 2026
- 2026Multimodal-Multiresolution Foundation Model for Lunar Remote Sensing8 Sep 2026
- 2026DART: Depth-as-Target Pretraining for Surgical Vision Foundation Models3 Sep 2026
- 2026GAAT: Geometry-Aware Alignment Transformer for Multimodal UAV Perception28 Aug 2026
- 2026DINOcular: Self-Supervised Visuospatial Representations27 Aug 2026
- 2026CM-MAE: A Physics-Guided Cross-Modal Self-Supervised Learning Framework for Vision-Wireless Applications16 Aug 2026
12 newest of 267 All 267 (JSON)