EasyDeepLearn

What is a solid default augmentation recipe for training a modern vision model?

medium

Answer

  • Random resized crop, horizontal flip, color jitter (brightness/contrast/saturation/hue), and one of RandAugment / TrivialAugment for stronger transforms.
  • Add Random Erasing / Cutout.
  • For classification, layer in Mixup (α=0.2) and CutMix (α=1.0).
  • Label smoothing 0.1.
  • Combined with AdamW + cosine schedule + EMA, this recipe gives you ~+2-4% over vanilla training on ImageNet and is the modern baseline (DeiT / ConvNeXt).

How to say it out loud

I'd start with random resized crop and horizontal flip — those two always pay for themselves. Then RandAugment or TrivialAugment for the stronger transforms, Random Erasing, and if it's classification, Mixup and CutMix on top. Label smoothing at 0.1. That's essentially the DeiT recipe, and it buys you two to four points on ImageNet over training with no augmentation at all. The caveat I'd add is that this is tuned for long schedules — three hundred epochs. On a short fine-tune, that much augmentation underfits, and I'd strip it back to crop, flip, and light jitter.

Check yourself — multiple choice
  • Only random crop
  • RandomResizedCrop + flip + color jitter + RandAugment + RandomErasing + Mixup/CutMix + label smoothing 0.1
  • No augmentation
  • Only rotation

Modern vision recipe: crop + flip + jitter + RandAug + erase + Mixup/CutMix + smoothing.

What the interviewer asks next

#augmentation#training#computer-vision

Practise Deep Learning

214 interview questions in this topic.

Related questions