atlas

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

Decoupled Training: Return of Frustratingly Easy Multi-Domain Learning

AAAI 2024 · Label-efficient and robust learning › Domain adaptation and generalization · approach: Mixture and distillation for domain shift · 4 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 Decoupled Training (D-Train), a hyperparameter-free tri-phase general-to-specific training strategy on a shared-bottom model (pre-train shared, post-train multi-heads, fine-tune heads with frozen backbone) for multi-domain learning.

Problem

Multi-domain learning must handle dataset bias across domains and domain domination (imbalanced or long-tailed sample counts), while recent MDL methods add parameters and hyperparameters through complex architectures or losses.

Approach

Train a shared-bottom model in three phases: pre-train on all domains to warm up a root model, post-train with domain-specific heads, then freeze the backbone and fine-tune each domain's head independently so that head-dominant domains cannot dominate the others.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Use D-Train instead of MMoEMulti-domain learning with ResNet-50 · Office-HomeAvg. Acc.: 85.2 -> 86.0Helped
Use D-Train instead of CDAN-MDLMulti-domain learning with ResNet-101 · DomainNetAvg. Acc.: 76.2 -> 77.3Helped
Use D-Train instead of CDAN-MDLMulti-domain learning with ResNet-101 · DomainNetWorst Acc.: 71.0 -> 72.2Helped
Use D-Train instead of Jointly TrainMulti-domain learning with DenseNet-121 · FMoWAvg. Acc.: 60.4 -> 61.0Helped
Use D-Train instead of MulANNMulti-domain learning with DenseNet-121 · FMoWWorst Acc.: 57.5 -> 58.3Helped
Add D-Train fine-tuningPLE · Amazon Product ReviewAUC_d: 77.62 -> 78.57Helped
Add D-Train fine-tuningPLE · Amazon Product ReviewAUC_s: 77.46 -> 77.56Helped
Add D-Train fine-tuningMMoE · Amazon Product ReviewAUC_d: 76.01 -> 77.70Helped
Add D-Train fine-tuningMMoE · Amazon Product ReviewAUC_s: 76.13 -> 77.05Helped
Add the fine-tuning phase (ablation)D-Train · DomainNetAvg. Acc.: 75.9 -> 77.3Helped
Add the post-training phase (ablation)D-Train · DomainNetAvg. Acc.: 76.6 -> 77.3Helped
Add the pre-training phase (ablation)D-Train · DomainNetAvg. Acc.: 77.0 -> 77.3Helped

Surprising findings

Limitations

Experimental setup

Datasets: Office-Home DomainNet FMoW Amazon Product Review Tencent DSP online ads data

Benchmarks: Office-Home DomainNet FMoW Amazon Product Review

Metrics: accuracy worst-domain accuracy AUC online cost lift GMV lift

Baselines: Separately Train Jointly Train MulANN DANN-MDL ASP-MDL CDAN-MDL Shared Bottom MoE MMoE PLE

Builds on: shared-bottom architecture DANN CDAN MMoE PLE

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