Robust Deep Learning Under Distribution Shift
summary
Proposes a practical pipeline that detects and corrects label shift by using an existing classifier as a low‑dimensional probe: estimate an invertible confusion matrix on source data, solve a linear system to recover the target label marginal from the mean softmax outputs, and then apply importance-weighted retraining or plug‑in reweighting. Shows a consistent estimator with error bounds depending on confusion‑matrix condition number and sample sizes, finds detection works with ~100s–1Ks examples, and warns domain‑adversarial methods break when class marginals change.
AI-written summary · about
chapters (32)
- 0:00Intro
- 1:39Outline
- 3:02Standard assumptions
- 4:47Adversarial Misspellings (Char-Level Attack)
- 6:07Curated Training Task Fail to Represent Reality
- 7:57Feedback Loops
- 8:47Impossibility absent assumptions
- 11:06Detecting and correcting for label shift with black box predictors
- 11:20Motivation 1: Pneumonia prediction
- 12:08Epidemic
- 13:17Motivation 2: Image Classification
- 14:22The test-Item effect
- 15:59Domain Adaptation - Formal Setup
- 17:31Label Shift (aka Target Shift)
- 18:47Contrast with Covariate Shift
- 20:35Black Box Shift Estimation (BBSE)
- 22:28Confusion matrices
- 22:59Applying the label shift assumption...
- 25:41Consistency
- 26:06Error bound
- 29:16Detection
- 31:31Estimation error (MNIST)
- 32:46Black Box Shift Correction (CIFAR10 w IW-ERM)
- 35:34A General Pipeline for Detecting Shift
- 39:35Non-adversarial image perturbations
- 41:29Covariate shift + model misspecification
- 41:53Implicit bias of SGD on linear networks w. linearly separable data
- 43:12Impact of IW on ERM decays over MLP training
- 45:21Weight-Invariance after 1000 epochs
- 46:08L2 Regularization v Dropout
- 46:53Deep DA / Domain-Adversarial Nets
- 50:02Synthetic experiments
related talks
42mBeyond Adversarial Approaches on Distributional Robustness
01h 01mDistribution Shift as Underspecification, and What We Might Do About It
48mA New Paradigm for Learning with Distribution Shift
37mDomain Adaptation-- 20 years of theory chasing practice
39mPseudo-Labeling for Covariate Shift Adaptation
33mWeak supervision: a new paradigm for unreliable data
20mGreenfield vs. Brownfield Data Labeling
23mHow to detect silent failures in ML models
