HI-MoE: Hierarchical Instance-Conditioned Mixture-of-Experts for Object Detection
Abstract
Mixture-of-Experts (MoE) architectures enable conditional computation by activating only a subset of model parameters for each input. Although sparse routing has been highly effective in language models and has also shown promise in vision, most vision MoE methods operate at the image or patch level. This granularity is poorly aligned with object detection, where the fundamental unit of reasoning is an object query corresponding to a candidate instance. We propose Hierarchical Instance-Conditioned Mixture-of-Experts (HI-MoE), a DETR-style detection architecture that performs routing in two stages: a lightweight scene router first selects a scene-consistent expert subset, and an instance router then assigns each object query to a small number of experts within that subset. This design aims to preserve sparse computation while better matching the heterogeneous, instance-centric structure of detection. In the current draft, experiments are concentrated on COCO with preliminary specialization analysis on LVIS. Under these settings, HI-MoE improves over a dense DINO baseline and over simpler token-level or instance-only routing variants, with especially strong gains on small objects. We also provide an initial visualization of expert specialization patterns. We present the method, ablations, and current limitations in a form intended to support further experimental validation.
Figures & tables
| Symbol | Meaning |
|---|---|
| Number of instance-level experts in each MoE block | |
| Number of active instance experts per query | |
| Number of scene-level routing logits | |
| Number of active scene routes selected by the scene router | |
| -th object query in the decoder | |
| Global scene descriptor produced from encoder features |
| Method | AP | AP s | Params |
|---|---|---|---|
| DETR | 42.0 | 20.5 | 41M |
| Deformable DETR | 48.7 | 29.6 | 40M |
| DINO | 51.3 | 32.1 | 50M |
| HI-MoE | 53.0 | 35.4 | 52M |
| Variant | Routing | AP | AP s | Params (M) | GFLOPs |
|---|---|---|---|---|---|
| Dense DINO | Dense FFN | 51.3 | 32.1 | 50 | 280 |
| Token-MoE | Flat token-level | 52.1 | 33.5 | 52 | 285 |
| Instance-only | Query router only | 52.6 | 34.2 | 52 | 282 |
| Scene-only | Scene router + shared experts | 52.4 | 33.8 | 52 | 283 |
| HI-MoE (full) | Scene + query routing | 53.0 | 35.4 | 52 | 282 |
| top- | AP | GFLOPs | Latency (ms, V100) |
|---|---|---|---|
| 1 | 52.4 | 270 | 28 |
| 2 | 53.0 | 282 | 32 |
| 4 | 53.5 | 310 | 38 |
| Experts | AP | Params |
|---|---|---|
| 4 | 52.1 | 51M |
| 8 | 52.6 | 52M |
| 16 | 53.0 | 52M |
| Placement | AP | Params |
|---|---|---|
| Encoder only | 52.3 | 52M |
| Decoder only | 52.8 | 52M |
| Encoder + Decoder | 53.0 | 52M |
| Expert ID | Small (AP s ) | Occluded (AP occ ) | Tail (AP tail ) | Dominant scene route (within-expert share) |
|---|---|---|---|---|
| E1 | 28.5 | 22.1 | 15.3 | Crowd (45% of E1 assignments) |
| E3 | 36.2 | 29.4 | 12.1 | Indoor (32% of E3 assignments) |
| E6 | 24.1 | 31.2 | 28.7 | Outdoor (51% of E6 assignments) |
| Avg | 31.2 | 27.3 | 18.4 | – |
Object Detection
Object detection locates and classifies objects in images or video, balancing localization accuracy, robustness, and inference speed.
See all 624 papers