DAMEX: Dataset-aware Mixture-of-Experts for visual understanding of mixture-of-datasets
On the map: Mixture-of-experts → Vision mixture-of-experts
Abstract (arXiv)
Construction of a universal detector poses a crucial question: How can we most effectively train a model on a large mixture of datasets? The answer lies in learning dataset-specific features and ensembling their knowledge but do all this in a single model. Previous methods achieve this by having separate detection heads on a common backbone but that results in a significant increase in parameters. In this work, we present Mixture-of-Experts as a solution, highlighting that MoEs are much more than a scalability tool. We propose Dataset-Aware Mixture-of-Experts, DAMEX where we train the experts to become an `expert' of a dataset by learning to route each dataset tokens to its mapped expert. Experiments on Universal Object-Detection Benchmark show that we outperform the existing state-of-the-art by average +10.2 AP score and improve over our non-MoE baseline by average +2.0 AP score. We also observe consistent gains while mixing datasets with (1) limited availability, (2) disparate domains and (3) divergent label sets. Further, we qualitatively show that DAMEX is robust against expert representation collapse.
Key facts. Cited by 23 papers in the map (data through 14 Oct 2026). Placed in Vision mixture-of-experts, one of 12 directions in Mixture-of-experts. Builds on 38 of its 47 references that are in the map. First posted on arXiv 8 Nov 2023.
Builds on
38 of 47 references.
Closest ancestors · 2
Same direction, newest first
- 2022Towards Understanding Mixture of Experts in Deep Learning4 Aug
- 2021Scaling Vision with Sparse Mixture of Experts10 Jun
Other references · 36
- 2014Microsoft COCO: Common Objects in Contextcited by 17,608
- 2015Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networkscited by 9,639
- 2023Segment Anythingcited by 8,234
- 2017Outrageously Large Neural Networks: The Sparsely-Gated Mixture-of-Experts Layercited by 3,509
- 2021Switch Transformers: Scaling to Trillion Parameter Models with Simple and Efficient Sparsitycited by 3,073
- 2020GShard: Scaling Giant Models with Conditional Computation and Automatic Shardingcited by 1,631
- 2019Do ImageNet Classifiers Generalize to ImageNet?cited by 1,515
- 2019Towards Robust Monocular Depth Estimation: Mixing Datasets for Zero-shot Cross-dataset Transfercited by 1,495
- 2022DINO: DETR with Improved DeNoising Anchor Boxes for End-to-End Object Detectioncited by 1,319
- 2021YOLOX: Exceeding YOLO Series in 2021cited by 1,071
- 2021GLaM: Efficient Scaling of Language Models with Mixture-of-Expertscited by 897
- 2021Florence: A New Foundation Model for Computer Visioncited by 804
24 more
- 2017DOTA: A Large-scale Dataset for Object Detection in Aerial Imagescited by 623
- 2022DAB-DETR: Dynamic Anchor Boxes are Better Queries for DETRcited by 573
- 2022DN-DETR: Accelerate DETR Training by Introducing Query DeNoisingcited by 488
- 2022Unified-IO: A Unified Model for Vision, Language, and Multi-Modal Taskscited by 421
- 2015WIDER FACE: A Face Detection Benchmarkcited by 298
- 2020VarifocalNet: An IoU-aware Dense Object Detectorcited by 231
- 2018Cross-Domain Weakly-Supervised Object Detection through Progressive Domain Adaptationcited by 218
- 2022Omnivore: A Single Model for Many Visual Modalitiescited by 215
- 2020Scaled-YOLOv4: Scaling Cross Stage Partial Networkcited by 211
- 2022Tutel: Adaptive Mixture-of-Experts at Scalecited by 201
- 2018Why do deep convolutional networks generalize so poorly to small image transformations?cited by 198
- 2021Dynamic Head: Unifying Object Detection Heads with Attentionscited by 187
- 2022On the Representation Collapse of Sparse Mixture of Expertscited by 157
- 2021Simple multi-dataset detectioncited by 127
- 2019Hierarchical Deep Stereo Matching on High-resolution Imagescited by 126
- 2021MSeg: A Composite Dataset for Multi-domain Semantic Segmentationcited by 122
- 2020Generalized Focal Loss V2: Learning Reliable Localization Quality Estimation for Dense Object Detectioncited by 104
- 2019Towards Universal Object Detection by Domain Attentioncited by 89
- 2020Object Detection with a Unified Label Space from Multiple Datasetscited by 65
- 2021Towards General Purpose Vision Systemscited by 46
- 2017Deep Lesion Graphs in the Wild: Relationship Learning and Organization of Significant Radiology Image Findings in a Diverse Large-scale Lesion Databasecited by 42
- 2021M6-10T: A Sharing-Delinking Paradigm for Efficient Multi-Trillion Parameter Pretrainingcited by 31
- 2016Multiview RGB-D Dataset for Object Instance Detectioncited by 26
- 2020Universal-RCNN: Universal Object Detector via Transferable Graph R-CNNcited by 18
Built on by
Citing papers per year: 2024: 5; 2025: 7; 2026: 11.
23 papers.
Sort:
By area:
- 2024SM3Det: A Unified Model for Multi-Modal Remote Sensing Object Detectioncited by 22
- 2024XTrack: Multimodal Training Boosts RGB-X Video Object Trackerscited by 21
- 2024PLeaS -- Merging Models with Permutations and Least Squarescited by 16
- 2024Routing Experts: Learning to Route Dynamic Experts in Multi-modal Large Language Modelscited by 14
- 2026Sparse Models, Sparse Safety: Unsafe Routes in Mixture-of-Experts LLMscited by 9
- 2025Fast-Slow-Thinking: Complex Task Solving with Large Language Modelscited by 7
- 2025Long-Tailed Distribution-Aware Router For Mixture-of-Experts in Large Vision-Language Modelcited by 7
- 2025GEMINUS: Dual-aware Global and Scene-Adaptive Mixture-of-Experts for End-to-End Autonomous Drivingcited by 5
- 2025MAPEX: Modality-Aware Pruning of Experts for Remote Sensing Foundation Modelscited by 3
- 2026HI-MoE: Hierarchical Instance-Conditioned Mixture-of-Experts for Object Detectioncited by 3
- 2026UniRoute: Unified Routing Mixture-of-Experts for Modality-Adaptive Remote Sensing Change Detectioncited by 2
- 2024Multi-Stage Multi-Modal Pre-Training for Automatic Speech Recognitioncited by 1
11 more
- 2025COME: Dual Structure-Semantic Learning with Collaborative MoE for Universal Lesion Detection Across Heterogeneous Ultrasound Datasetscited by 1
- 2025Tokenize Once, Recommend Anywhere: Unified Item Tokenization for Multi-domain LLM-based Recommendationcited by 1
- 2025Expert Switching for Robust AAV Landing: A Dual-Detector Framework in Simulationcited by 0
- 2026Understanding Task Aggregation for Generalizable Ultrasound Foundation Modelscited by 0
- 2026Bidirectional Learning of Facial Action Units and Expressions via Structured Semantic Mapping across Heterogeneous Datasetscited by 0
- 2026Multi-Domain Learning with Global Expert Mappingcited by 0
- 2026HMR-Net: Hierarchical Modular Routing for Cross-Domain Object Detection in Aerial Imagescited by 0
- 2026Teacher-Guided Routing for Sparse Vision Mixture-of-Expertscited by 0
- 2026SARIF: Segment Anything for Robust Image Forensicscited by 0
- 2026FaceMoE: Mixture of Experts for Low-Resolution Face Recognitioncited by 0
- 2026RouteRec: Behavior-Guided Sparse Routing for Sequential Recommendationcited by 0
- 2026RouteRec: Behavior-Guided Sparse Routing for Sequential Recommendation30 Sep 2026
- 2026FaceMoE: Mixture of Experts for Low-Resolution Face Recognition30 Jun 2026
- 2026SARIF: Segment Anything for Robust Image Forensics19 Jun 2026
- 2026Teacher-Guided Routing for Sparse Vision Mixture-of-Experts23 Apr 2026
- 2026HMR-Net: Hierarchical Modular Routing for Cross-Domain Object Detection in Aerial Images20 Apr 2026
- 2026Multi-Domain Learning with Global Expert Mapping20 Apr 2026
- 2026Bidirectional Learning of Facial Action Units and Expressions via Structured Semantic Mapping across Heterogeneous Datasets12 Apr 2026
- 2026HI-MoE: Hierarchical Instance-Conditioned Mixture-of-Experts for Object Detection6 Apr 2026
- 2026Understanding Task Aggregation for Generalizable Ultrasound Foundation Models18 Mar 2026
- 2026Sparse Models, Sparse Safety: Unsafe Routes in Mixture-of-Experts LLMs9 Feb 2026
- 2026UniRoute: Unified Routing Mixture-of-Experts for Modality-Adaptive Remote Sensing Change Detection21 Jan 2026
- 2025Expert Switching for Robust AAV Landing: A Dual-Detector Framework in Simulation16 Dec 2025
11 more
- 2025Tokenize Once, Recommend Anywhere: Unified Item Tokenization for Multi-domain LLM-based Recommendation17 Nov 2025
- 2025COME: Dual Structure-Semantic Learning with Collaborative MoE for Universal Lesion Detection Across Heterogeneous Ultrasound Datasets13 Aug 2025
- 2025GEMINUS: Dual-aware Global and Scene-Adaptive Mixture-of-Experts for End-to-End Autonomous Driving19 Jul 2025
- 2025MAPEX: Modality-Aware Pruning of Experts for Remote Sensing Foundation Models10 Jul 2025
- 2025Long-Tailed Distribution-Aware Router For Mixture-of-Experts in Large Vision-Language Model2 Jul 2025
- 2025Fast-Slow-Thinking: Complex Task Solving with Large Language Models11 Apr 2025
- 2024SM3Det: A Unified Model for Multi-Modal Remote Sensing Object Detection30 Dec 2024
- 2024Routing Experts: Learning to Route Dynamic Experts in Multi-modal Large Language Models19 Jul 2024
- 2024PLeaS -- Merging Models with Permutations and Least Squares2 Jul 2024
- 2024XTrack: Multimodal Training Boosts RGB-X Video Object Trackers28 May 2024
- 2024Multi-Stage Multi-Modal Pre-Training for Automatic Speech Recognition28 Mar 2024
- 2026Sparse Models, Sparse Safety: Unsafe Routes in Mixture-of-Experts LLMscited by 9
- 2026HI-MoE: Hierarchical Instance-Conditioned Mixture-of-Experts for Object Detectioncited by 3
- 2025COME: Dual Structure-Semantic Learning with Collaborative MoE for Universal Lesion Detection Across Heterogeneous Ultrasound Datasetscited by 1
- 2025Tokenize Once, Recommend Anywhere: Unified Item Tokenization for Multi-domain LLM-based Recommendationcited by 1
- 2026Multi-Domain Learning with Global Expert Mappingcited by 0
- 2026Teacher-Guided Routing for Sparse Vision Mixture-of-Expertscited by 0
- 2026FaceMoE: Mixture of Experts for Low-Resolution Face Recognitioncited by 0
- 2026RouteRec: Behavior-Guided Sparse Routing for Sequential Recommendationcited by 0
- 2026RouteRec: Behavior-Guided Sparse Routing for Sequential Recommendation30 Sep 2026
- 2026FaceMoE: Mixture of Experts for Low-Resolution Face Recognition30 Jun 2026
- 2026Teacher-Guided Routing for Sparse Vision Mixture-of-Experts23 Apr 2026
- 2026Multi-Domain Learning with Global Expert Mapping20 Apr 2026
- 2026HI-MoE: Hierarchical Instance-Conditioned Mixture-of-Experts for Object Detection6 Apr 2026
- 2026Sparse Models, Sparse Safety: Unsafe Routes in Mixture-of-Experts LLMs9 Feb 2026
- 2025Tokenize Once, Recommend Anywhere: Unified Item Tokenization for Multi-domain LLM-based Recommendation17 Nov 2025
- 2025COME: Dual Structure-Semantic Learning with Collaborative MoE for Universal Lesion Detection Across Heterogeneous Ultrasound Datasets13 Aug 2025
- 2024SM3Det: A Unified Model for Multi-Modal Remote Sensing Object Detectioncited by 22
- 2024XTrack: Multimodal Training Boosts RGB-X Video Object Trackerscited by 21
- 2024PLeaS -- Merging Models with Permutations and Least Squarescited by 16
- 2025MAPEX: Modality-Aware Pruning of Experts for Remote Sensing Foundation Modelscited by 3
- 2026UniRoute: Unified Routing Mixture-of-Experts for Modality-Adaptive Remote Sensing Change Detectioncited by 2
- 2024Multi-Stage Multi-Modal Pre-Training for Automatic Speech Recognitioncited by 1
- 2025Expert Switching for Robust AAV Landing: A Dual-Detector Framework in Simulationcited by 0
- 2026Understanding Task Aggregation for Generalizable Ultrasound Foundation Modelscited by 0
- 2026Bidirectional Learning of Facial Action Units and Expressions via Structured Semantic Mapping across Heterogeneous Datasetscited by 0
- 2026HMR-Net: Hierarchical Modular Routing for Cross-Domain Object Detection in Aerial Imagescited by 0
- 2026SARIF: Segment Anything for Robust Image Forensicscited by 0
- 2026SARIF: Segment Anything for Robust Image Forensics19 Jun 2026
- 2026HMR-Net: Hierarchical Modular Routing for Cross-Domain Object Detection in Aerial Images20 Apr 2026
- 2026Bidirectional Learning of Facial Action Units and Expressions via Structured Semantic Mapping across Heterogeneous Datasets12 Apr 2026
- 2026Understanding Task Aggregation for Generalizable Ultrasound Foundation Models18 Mar 2026
- 2026UniRoute: Unified Routing Mixture-of-Experts for Modality-Adaptive Remote Sensing Change Detection21 Jan 2026
- 2025Expert Switching for Robust AAV Landing: A Dual-Detector Framework in Simulation16 Dec 2025
- 2025MAPEX: Modality-Aware Pruning of Experts for Remote Sensing Foundation Models10 Jul 2025
- 2024SM3Det: A Unified Model for Multi-Modal Remote Sensing Object Detection30 Dec 2024
- 2024PLeaS -- Merging Models with Permutations and Least Squares2 Jul 2024
- 2024XTrack: Multimodal Training Boosts RGB-X Video Object Trackers28 May 2024
- 2024Multi-Stage Multi-Modal Pre-Training for Automatic Speech Recognition28 Mar 2024
Similar, not cited
Close in wording, no citation either way
- 2026Domain-Specialized Object Detection via Model-Level Mixtures of Experts
- 2024Mixture of Experts in Image Classification: What's the Sweet Spot?
- 2024Multilinear Mixture of Experts: Scalable Expert Specialization through Factorization
- 2025Efficient Data Driven Mixture-of-Expert Extraction from Trained Networks
- 2026Mixture-of-Experts Models in Vision: Routing, Optimization, and Generalization
- 2023MoCaE: Mixture of Calibrated Experts Significantly Improves Object Detection
- 2020Biased Mixtures Of Experts: Enabling Computer Vision Inference Under Data Transfer Limitations
- 2025MixtureKit: A General Framework for Composing, Training, and Visualizing Mixture-of-Experts Models