atlas

Research map / Label-efficient and robust learning / Domain adaptation and generalization

Mean Teacher DETR with Masked Feature Alignment: A Robust Domain Adaptive Detection Transformer Framework

AAAI 2024 · Label-efficient and robust learning › Domain adaptation and generalization · approach: Geometric alignment and synthetic-to-real adaptation · 21 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes MTM, a two-stage unsupervised domain adaptive DETR framework that pretrains on source and CycleGAN target-like images with masked domain-query and token-wise feature alignment, then self-trains with mean teacher and Object Queries Knowledge Transfer.

Problem

Unsupervised domain adaptive object detection with DETR suffers from performance fluctuation and training stagnation in one-stage feature alignment, and from unreliable pretrained teachers and unstable gains in mean-teacher two-stage methods; target-like images are unexplored for DETR adaptation.

Approach

Pretrain Deformable DETR on labeled source and CycleGAN target-like images with masked (randomly dropped) domain-query and token-wise feature discriminators to avoid stagnation, then self-train with mean teacher where student object queries are enhanced by attending to teacher queries (OQKT).

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use MTM instead of the strongest listed baselineDeformable DETR · Cityscapes → Foggy CityscapesmAP: 46.8 -> 48.9Helped
Use MTM instead of the strongest listed baselineDeformable DETR · Cityscapes → BDD100KmAP: 32.6 -> 37.3Helped
Use MTM instead of MTTransDeformable DETR · Sim10k → CityscapesAP on car: 57.9 -> 58.1Helped
Add labeled target-like images to source-only pretraining (ablation)Deformable DETR · Cityscapes → Foggy Cityscapes (pretraining)mAP: 28.5 -> 42.3Helped
Replace non-masked MTWFA with masked MTWFA (ablation)Target-like pretraining · Cityscapes → Foggy Cityscapes (pretraining)mAP: 42.7 -> 43.6Helped
Replace non-masked MDQFA with masked MDQFA (ablation)Target-like pretraining · Cityscapes → Foggy Cityscapes (pretraining)mAP: 44.9 -> 45.7Helped
Replace both non-masked alignment methods with masked methods (ablation)Target-like pretraining · Cityscapes → Foggy Cityscapes (pretraining)mAP: 45.3 -> 47.2Helped
Set θMask=0.40 and η=1/2 (ablation)MTM pretraining · Cityscapes → Foggy Cityscapes (pretraining)mAP: 45.6 -> 47.2Helped
Use mean teacher self-training (ablation)Deformable DETR · Cityscapes → Foggy Cityscapes (self-training)mAP: 28.5 -> 35.8Helped
Add OQKT to mean teacher (ablation)Deformable DETR self-training · Cityscapes → Foggy Cityscapes (self-training)mAP: 35.8 -> 38.3Helped
Add OQKT to mean teacher with MTM pretraining (ablation)MTM self-training · Cityscapes → Foggy CityscapesmAP: 47.7 -> 47.9Helped
Add masked MDQFA (ablation)MTM self-training with mean teacher and OQKT · Cityscapes → Foggy CityscapesmAP: 47.9 -> 48.0Helped

Surprising findings

Limitations

Experimental setup

Datasets: Cityscapes Foggy Cityscapes BDD100K Sim10K ImageNet

Benchmarks: Cityscapes to Foggy Cityscapes Cityscapes to BDD100K Sim10K to Cityscapes

Metrics: mAP@0.5 AP on car

Baselines: Deformable DETR (source only) SFA O2Net MTTrans DA-DETR

Builds on: Deformable DETR CycleGAN Mean Teacher MTTrans SFA

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy