Domain adaptation: covariate shift, DANN, and few-shot learning

Prerequisites

Before reading this article, make sure you are comfortable with:

• Transfer learning: pretraining, feature extraction, and fine-tuning
• Bias-variance tradeoff: why models fail to generalize and how model complexity relates to data size

Interactive visualization: aligning source and target domains

Use the adaptation controls to watch source and target feature distributions move into a shared space. Notice how the decision boundary becomes useful for both domains only after alignment.

The domain shift problem

You trained on product photos from a studio. Clean backgrounds, consistent lighting, high resolution. Now you deploy on blurry phone photos taken by customers. Performance drops from 95% to 72%.

This is like moving to a new city where the rules are slightly different. You know how to drive, but the road signs look different, traffic flows in unexpected ways, and your GPS keeps recalculating. Your core skills still apply, but the surface details have changed.

Domain shift: source vs target distributions

graph LR
  A["Source distribution
Studio photos
Clean, consistent"] --> C["Same task
Same labels
Different features"]
  B["Target distribution
Phone photos
Noisy, varied"] --> C
  C --> D["Goal: learn features
that work for both"]

The core issue is that the training distribution (source domain) differs from the deployment distribution (target domain). The labels have not changed: a shoe is still a shoe. But the pixel patterns are different enough that the model struggles. We need strategies to bridge this gap.

Covariate shift vs dataset shift

Covariate shift: The input distribution changes ($p_S(x) \neq p_T(x)$), but the labeling function stays the same ($p(y|x)$ is identical). A cat is still a cat whether photographed with a DSLR or a phone camera. The relationship between features and labels hasn’t changed; the features themselves just look different.

Dataset shift (more general): Both $p(x)$ and $p(y|x)$ may change. Reviews from different time periods might use the same words differently, so the mapping from text to sentiment actually changes.

Label shift: $p(y)$ changes but $p(x|y)$ stays the same. A disease classifier trained where prevalence is 10% deployed where it’s 1%.

Most practical domain adaptation focuses on covariate shift, where the goal is to learn features that are invariant to the domain while remaining discriminative for the task.

Types of distribution shift

graph TD
  A["Distribution Shift"] --> B["Covariate Shift
P(x) changes
P(y|x) stays same"]
  A --> C["Label Shift
P(y) changes
P(x|y) stays same"]
  A --> D["Concept Drift
P(y|x) changes
The meaning of
features shifts"]
  B --> E["Example: DSLR vs phone
Same cat, different pixels"]
  C --> F["Example: disease prevalence
10% in training, 1% in deployment"]
  D --> G["Example: word meaning change
over time"]

Covariate shift is the most common and the most studied. The solutions below primarily target it, though some generalize to other shift types.

Feature extraction vs fine-tuning

Source vs target domain accuracy by adaptation method

The simplest approach to domain adaptation is transfer learning: take a pretrained model and adapt it to your target domain. Two strategies:

Feature extraction: Freeze the pretrained model. Use its output as a fixed feature vector. Train only a new classifier head on target data. This is safe when you have very little target data (say, under 100 examples per class), because you’re only training a small number of parameters and can’t overfit badly.

Full fine-tuning: Unfreeze the entire model and train on target data with a small learning rate. This is more powerful but riskier. If target data is scarce, the model can forget useful pretrained features (catastrophic forgetting) or overfit to the target training set.

The right choice depends on two factors: how much target data you have, and how different the source and target domains are.

flowchart TD
  START["Need to adapt model to new domain"] --> Q1{"How much target labeled data?"}
  Q1 -->|"< 100 per class"| Q2{"How different are domains?"}
  Q1 -->|"100-1000 per class"| FT_PARTIAL["Fine-tune top layers
Freeze bottom layers"]
  Q1 -->|"> 1000 per class"| FT_FULL["Full fine-tuning
with small lr"]

  Q2 -->|"Similar"| FE["Feature extraction
(freeze all, train head)"]
  Q2 -->|"Very different"| DA["Domain adaptation
(DANN, CORAL)"]

  FT_PARTIAL --> LR["Use layer-wise
learning rate decay"]

  style START fill:#9775fa,color:#fff
  style FE fill:#51cf66,color:#fff
  style FT_PARTIAL fill:#74c0fc,color:#fff
  style FT_FULL fill:#ffa94d,color:#fff
  style DA fill:#ff6b6b,color:#fff

Layer-wise learning rate decay

When fine-tuning, not all layers should learn at the same rate. Earlier layers capture general features (edges, textures) that transfer well. Later layers capture task-specific features that need more adaptation. Layer-wise learning rate decay assigns a smaller learning rate to earlier layers.

Given a base learning rate $\eta$ and a decay factor $\gamma \in (0, 1)$, the learning rate for layer $l$ (counting from the top) is:

where $L$ is the total number of layers. The top layer (closest to the output) gets the full learning rate $\eta$. Each layer below it gets $\gamma$ times the rate of the layer above.

This prevents catastrophic forgetting of low-level features while allowing high-level features to adapt. The decay factor $\gamma$ is typically between 0.1 and 0.5. Smaller $\gamma$ means more aggressive freezing of early layers.

DANN: domain adversarial neural network

Domain Adversarial Neural Network (DANN, 2016) takes a more principled approach. The idea: train a feature extractor that produces representations which are (1) useful for the task and (2) indistinguishable between source and target domains.

The architecture has three components:

1. Feature extractor $G_f$: maps input $x$ to features $f = G_f(x)$
2. Label predictor $G_y$: maps features to task labels $\hat{y} = G_y(f)$
3. Domain classifier $G_d$: maps features to domain labels $\hat{d} = G_d(f)$ (source vs target)

The training objective:

The feature extractor minimizes the label prediction loss while maximizing the domain classification loss. In other words, it tries to create features that help predict labels but confuse the domain classifier.

The minus sign is implemented through a gradient reversal layer (GRL). During forward pass, the GRL is an identity function. During backward pass, it multiplies the gradient by $-\lambda$. This elegantly handles the adversarial objective with standard gradient descent.

flowchart LR
  X["Input x"] --> GF["Feature extractor
Gf"]
  GF --> F["Features f"]
  F --> GY["Label predictor
Gy"]
  GY --> YHAT["ŷ (task label)"]

  F --> GRL["Gradient reversal
layer (×−λ)"]
  GRL --> GD["Domain classifier
Gd"]
  GD --> DHAT["d̂ (source/target)"]

  YHAT --> LY["Label loss ℒy
(minimize)"]
  DHAT --> LD["Domain loss ℒd
(maximize via GRL)"]

  style GRL fill:#ff6b6b,color:#fff
  style LY fill:#51cf66,color:#fff
  style LD fill:#ffa94d,color:#fff

The theory behind DANN connects to Ben-David’s domain adaptation bound. The target error is bounded by the source error plus the domain divergence (measured by the ability of a classifier to distinguish domains). By minimizing both terms, DANN minimizes the bound on target error.

CORAL: aligning statistics

CORrelation ALignment (CORAL) takes a simpler approach: align the second-order statistics (covariance matrices) of source and target features.

Given source features with covariance $C_S$ and target features with covariance $C_T$, the CORAL loss is:

where $d$ is the feature dimension and $\|\cdot\|_F$ is the Frobenius norm.

The $\frac{1}{4d^2}$ normalization makes the loss scale-invariant with respect to feature dimension. You can add this loss to any standard training objective as a regularization term:

CORAL is simple to implement and adds minimal computational overhead. It doesn’t require a separate domain classifier or gradient reversal. The limitation is that it only aligns second-order statistics. If the domain shift involves higher-order structure (e.g., different cluster shapes), CORAL may not be enough.

Deep CORAL applies this idea to features at intermediate layers of a deep network, not just the final features.

Comparing adaptation strategies

graph TD
  A["Fine-tuning
Retrain top layers
on target data"] --> D["Needs labeled
target data"]
  B["Domain Adversarial
DANN
Fool domain classifier"] --> E["Works with
unlabeled target data"]
  C["Distribution Matching
CORAL
Align feature statistics"] --> F["Simple, minimal
overhead"]
  D --> G["Best when you have
100+ labeled examples"]
  E --> H["Best for significant
domain gap"]
  F --> I["Best as lightweight
regularizer"]

Fine-tuning is the default when you have labeled target data. DANN is more powerful when labels are scarce but unlabeled target data is plentiful. CORAL is the simplest to add and works well as a regularization term alongside any other objective.

Few-shot learning

Few-shot learning handles the extreme case: you have only a handful of labeled examples per class in the target domain. The standard setup is N-way K-shot: classify among $N$ classes, with only $K$ examples per class.

Prototypical networks: Compute a prototype (mean embedding) for each class from the $K$ examples. Classify a query by finding the nearest prototype in embedding space.

For class $c$ with support examples $S_c$:

Classification uses softmax over negative distances:

where $d$ is typically squared Euclidean distance.

Meta-learning (learning to learn): Train the model across many few-shot tasks (episodes) sampled from a large dataset. Each episode simulates the few-shot scenario. The model learns to quickly adapt to new classes from few examples. MAML (Model-Agnostic Meta-Learning) is the canonical example: it finds model parameters that can be adapted to any new task with just a few gradient steps.

Transfer scenarios comparison

Example 1: layer-wise learning rate decay

A 5-layer network with base learning rate $\eta = 0.001$ and decay factor $\gamma = 0.3$.

Layers are numbered 1 (bottom/earliest) to 5 (top/closest to output).

Layer 1 learns ~123x slower than layer 5. The bottom layers (which capture general features like edges and textures) are nearly frozen, while the top layers (task-specific features) adapt freely.

In practice, you’d implement this in your optimizer by creating parameter groups. In PyTorch:

params = []
for i, layer in enumerate(model.layers):
    lr = base_lr * (decay ** (num_layers - 1 - i))
    params.append({"params": layer.parameters(), "lr": lr})
optimizer = torch.optim.Adam(params)

If the decay were $\gamma = 0.1$ instead, layer 1 would get $\eta_1 = 0.001 \times 0.1^4 = 0.0000001$, essentially frozen. With $\gamma = 0.5$, it would get $0.001 \times 0.5^4 = 0.0000625$, still learning but slowly.

Example 2: CORAL loss

Source features have covariance matrix $C_S$ and target features have covariance matrix $C_T$.

Feature dimension $d = 2$.

Step 1: Compute the difference

Step 2: Frobenius norm squared

The Frobenius norm squared is the sum of squares of all elements:

Step 3: CORAL loss

What does this number mean? The source features have positive correlation between dimensions (off-diagonal = 1), while target features are uncorrelated (off-diagonal = 0). The source also has higher variance in both dimensions. The CORAL loss penalizes these statistical differences.

If we add this to a classification loss with weight $\alpha = 1.0$:

During training, the gradient of $\mathcal{L}_{\text{CORAL}}$ pushes the feature extractor to produce features whose source and target covariances align. Over time, $C_S$ and $C_T$ should converge, and the CORAL loss should approach zero.

Example 3: few-shot 5-way 1-shot classification

We have 5 classes, 1 support example per class. A query image arrives and we need to classify it.

The encoder $f$ maps each image to an embedding. The 5 class prototypes (just the single support embedding for each class) are already computed. We compute the Euclidean distance from the query embedding to each prototype.

Given distances from the query to each class prototype:

These are distances, so smaller means more similar. Prototypical networks use negative distances in a softmax:

Step 1: Compute $\exp(-d_c)$ for each class

Step 2: Sum

Step 3: Softmax probabilities

Prediction: Class 1 with probability 0.2631.

The model is not very confident. The margin between class 1 (0.263) and class 5 (0.238) is small. This is typical in 1-shot settings. With 5-shot (5 examples per class), prototypes would be more robust (averaging 5 embeddings reduces noise), and confidence would increase. The distances would also typically separate more because the prototypes better represent the class centers.

Practical fine-tuning guide

1. Always start with the simplest approach. Try feature extraction (frozen backbone + linear head) first. If that works well enough, stop. If not, try fine-tuning the top few layers. Only resort to full fine-tuning or domain adaptation methods if simpler approaches fail.

2. Use a smaller learning rate for fine-tuning than for training from scratch. A common starting point: $\frac{1}{10}$ of the original pretraining learning rate.

3. Watch for overfitting. With small target datasets, validation loss will start increasing quickly. Use early stopping, regularization (dropout, weight decay), and data augmentation.

4. Gradual unfreezing is an alternative to layer-wise lr decay. Start with all layers frozen except the head. Train for a few epochs. Unfreeze the top layer. Train more. Unfreeze the next layer. This gives you fine-grained control and is less sensitive to learning rate choices.

5. Domain adaptation methods (DANN, CORAL) are most valuable when you have unlabeled target data. If you have labeled target data, fine-tuning is usually simpler and works just as well.

Summary

Domain shift is inevitable in real applications. The right adaptation strategy depends on how much labeled target data you have and how different the domains are. Feature extraction is safe with little data. Fine-tuning with layer-wise learning rate decay works well with moderate data. DANN and CORAL are useful when you have unlabeled target data and significant domain shift. Few-shot methods handle the extreme case of just a handful of examples.

The common thread: good representations transfer. The better your pretrained features, the less adaptation you need.

What comes next

The next article on distributed representations and latent spaces dives into the geometry of learned representations. You’ll see how word embedding analogies work, why latent spaces have linear structure, how to measure disentanglement, and how latent arithmetic in GANs and VAEs enables controlled generation.