A Theory of Label Propagation for Subpopulation Shift
Not on the map. Why?
Abstract (arXiv)
One of the central problems in machine learning is domain adaptation. Unlike past theoretical work, we consider a new model for subpopulation shift in the input or representation space. In this work, we propose a provably effective framework for domain adaptation based on label propagation. In our analysis, we use a simple but realistic expansion assumption, proposed in citetwei2021theoretical. Using a teacher classifier trained on the source domain, our algorithm not only propagates to the target domain but also improves upon the teacher. By leveraging existing generalization bounds, we also obtain end-to-end finite-sample guarantees on the entire algorithm. In addition, we extend our theoretical framework to a more general setting of source-to-target transfer based on a third unlabeled dataset, which can be easily applied in various learning scenarios. Inspired by our theory, we adapt consistency-based semi-supervised learning methods to domain adaptation settings and gain significant improvements.
Key facts. Cited by 48 papers in the map (data through 7 Oct 2026). Builds on 49 of its 85 references that are in the map. First posted on arXiv 22 Feb 2021.
Builds on
49 of 85 references.
References · 49
- 2020A Simple Framework for Contrastive Learning of Visual Representationscited by 11,979
- 2020Bootstrap your own latent: A new approach to self-supervised Learningcited by 4,712
- 2015Domain-Adversarial Training of Neural Networkscited by 3,023
- 2020Unsupervised Learning of Visual Features by Contrasting Cluster Assignmentscited by 2,845
- 2020FixMatch: Simplifying Semi-Supervised Learning with Consistency and Confidencecited by 2,080
- 2019Invariant Risk Minimizationcited by 1,832
- 2015Learning Transferable Features with Deep Adaptation Networkscited by 1,323
- 2017Adversarial Discriminative Domain Adaptationcited by 1,309
- 2019Distributionally Robust Neural Networks for Group Shifts: On the Importance of Regularization for Worst-Case Generalizationcited by 1,232
- 2017Deep Hashing Network for Unsupervised Domain Adaptationcited by 1,181
- 2019A Comprehensive Survey on Transfer Learningcited by 1,154
- 2017CyCADA: Cycle-Consistent Adversarial Domain Adaptationcited by 1,079
37 more
- 2019Unsupervised Data Augmentation for Consistency Trainingcited by 1,057
- 2020In Search of Lost Domain Generalizationcited by 1,042
- 2017Virtual Adversarial Training: A Regularization Method for Supervised and Semi-Supervised Learningcited by 918
- 2017Learning to Generalize: Meta-Learning for Domain Generalizationcited by 762
- 2020Out-of-Distribution Generalization via Risk Extrapolation (REx)cited by 737
- 2016Deep Transfer Learning with Joint Adaptation Networkscited by 609
- 2019Bridging Theory and Algorithm for Domain Adaptationcited by 332
- 2015Domain Generalization for Object Recognition with Multi-task Autoencoderscited by 253
- 2019Sliced Wasserstein Discrepancy for Unsupervised Domain Adaptationcited by 235
- 2020How Neural Networks Extrapolate: From Feedforward to Graph Neural Networkscited by 218
- 2018A DIRT-T Approach to Unsupervised Domain Adaptationcited by 215
- 2020Representation Learning via Invariant Causal Mechanismscited by 197
- 2018Multi-Adversarial Domain Adaptationcited by 195
- 2020Invariant Risk Minimization Gamescited by 188
- 2020Theoretical Analysis of Self-Training with Deep Networks on Unlabeled Datacited by 179
- 2020BREEDS: Benchmarks for Subpopulation Shiftcited by 155
- 2020Understanding Self-Training for Gradual Domain Adaptationcited by 149
- 2018Deep Co-Training for Semi-Supervised Image Recognitioncited by 145
- 2020Learning explanations that are hard to varycited by 132
- 2019Multi-adversarial Faster-RCNN for Unrestricted Object Detectioncited by 111
- 2018Deep Cocktail Network: Multi-source Unsupervised Domain Adaptation with Category Shiftcited by 109
- 2020A Review of Single-Source Deep Unsupervised Visual Domain Adaptationcited by 96
- 2019HoMM: Higher-order Moment Matching for Unsupervised Domain Adaptationcited by 89
- 2014Domain-Adversarial Neural Networkscited by 78
- 2019Multi-source Domain Adaptation for Semantic Segmentationcited by 74
- 2020Self-training Avoids Using Spurious Features Under Domain Shiftcited by 69
- 2019Unsupervised Domain Adaptation using Feature-Whitening and Consensus Losscited by 67
- 2019Domain Adaptation with Asymmetrically-Relaxed Distribution Alignmentcited by 61
- 2019Improved Sample Complexities for Deep Networks and Robust Classification via an All-Layer Margincited by 56
- 2019On Learning Invariant Representation for Domain Adaptationcited by 38
- 2020Fundamental Limits and Tradeoffs in Invariant Representation Learningcited by 38
- 2020Rethinking Distributional Matching Based Domain Adaptationcited by 33
- 2019Multi-level Domain Adaptive learning for Cross-Domain Detectioncited by 30
- 2016Generative Adversarial Nets from a Density Ratio Estimation Perspectivecited by 24
- 2020Learning Causal Models Onlinecited by 18
- 2014Adaptation Algorithm and Theory Based on Generalized Discrepancycited by 9
- 2019Butterfly: One-step Approach towards Wildly Unsupervised Domain Adaptationcited by 8
Built on by
Citing papers per year: 2021: 8; 2022: 15; 2023: 11; 2024: 5; 2025: 6; 2026: 3.
48 papers.
Sort:
By area:
- 2022Fine-Tuning can Distort Pretrained Features and Underperform Out-of-Distributioncited by 722
- 2021Provable Guarantees for Self-Supervised Deep Learning with Spectral Contrastive Losscited by 311
- 2021Cycle Self-Training for Domain Adaptationcited by 120
- 2021Towards a Theoretical Framework of Out-of-Distribution Generalizationcited by 94
- 2022Connect, Not Collapse: Explaining Contrastive Learning for Unsupervised Domain Adaptationcited by 66
- 2022Divide and Contrast: Source-free Domain Adaptation via Adaptive Contrastive Learningcited by 53
- 2024Theoretical Analysis of Weak-to-Strong Generalizationcited by 46
- 2022Beyond Separability: Analyzing the Linear Transferability of Contrastive Representations to Related Subpopulationscited by 44
- 2021Near-Optimal Linear Regression under Distribution Shiftcited by 40
- 2022Invariant and Transportable Representations for Anti-Causal Domain Shiftscited by 37
- 2021If your data distribution shifts, use self-learningcited by 36
- 2022Sample Efficiency of Data Augmentation Consistency Regularizationcited by 28
36 more
- 2022Transferring Fairness under Distribution Shifts via Fair Consistency Regularizationcited by 21
- 2022First Steps Toward Understanding the Extrapolation of Nonlinear Models to Unseen Domainscited by 20
- 2021Self-training Converts Weak Learners to Strong Learners in Mixture Modelscited by 19
- 2022Localized Adversarial Domain Generalizationcited by 19
- 2023Pseudo-Labeling for Kernel Ridge Regression under Covariate Shiftcited by 17
- 2023GeoNet: Benchmarking Unsupervised Adaptation across Geographiescited by 17
- 2022Understanding new tasks through the lens of training data via exponential tiltingcited by 11
- 2024The Road to Artificial SuperIntelligence: A Comprehensive Survey of Superalignmentcited by 11
- 2024Controllable Prompt Tuning For Balancing Group Distributional Robustnesscited by 10
- 2025Discrepancies are Virtue: Weak-to-Strong Generalization through Lens of Intrinsic Dimensioncited by 10
- 2022Label Propagation with Weak Supervisioncited by 9
- 2022Can domain adaptation make object recognition work for everyone?cited by 8
- 2023Cluster-aware Semi-supervised Learning: Relational Knowledge Distillation Provably Learns Clusteringcited by 7
- 2025Mitigating Spurious Correlation via Distributionally Robust Learning with Hierarchical Ambiguity Setscited by 7
- 2023Complementary Benefits of Contrastive Learning and Self-Training Under Distribution Shiftcited by 5
- 2023DAC-MR: Data Augmentation Consistency Based Meta-Regularization for Meta-Learningcited by 4
- 2023Understanding the Robustness of Multi-modal Contrastive Learning to Distribution Shiftcited by 4
- 2021Encoding Hierarchical Information in Neural Networks helps in Subpopulation Shiftcited by 3
- 2024Optimizing importance weighting in the presence of sub-population shiftscited by 3
- 2025Analyzing Similarity Metrics for Data Selection for Language Model Pretrainingcited by 3
- 2025Mitigating Confounding in Speech-Based Dementia Detection through Weight Maskingcited by 3
- 2023Pushing the Accuracy-Group Robustness Frontier with Introspective Self-playcited by 2
- 2023On the Stability and Generalization of Triplet Learningcited by 2
- 2023Controlling Federated Learning for Covertnesscited by 2
- 2024Bridging Domains with Approximately Shared Featurescited by 2
- 2022GOCA: Guided Online Cluster Assignment for Self-Supervised Video Representation Learningcited by 1
- 2023IW-GAE: Importance Weighted Group Accuracy Estimation for Improved Calibration and Model Selection in Unsupervised Domain Adaptationcited by 1
- 2021Can semi-supervised learning reduce the amount of manual labelling required for effective radio galaxy morphology classification?cited by 0
- 2022Dual-Correction Adaptation Network for Noisy Knowledge Transfercited by 0
- 2022Cross-client Label Propagation for Transductive and Semi-Supervised Federated Learningcited by 0
- 2023Chaos to Order: A Label Propagation Perspective on Source-Free Domain Adaptationcited by 0
- 2025References Indeed Matter? Reference-Free Preference Optimization for Conversational Query Reformulationcited by 0
- 2025Exploiting Vision Language Model for Training-Free 3D Point Cloud OOD Detection via Graph Score Propagationcited by 0
- 2026Learning from Limited Labels: Transductive Graph Label Propagation for Indian Music Analysiscited by 0
- 2026Black-Box Assisted Regression: Phase Transitions and Minimax Optimalitycited by 0
- 2026Learning the Robustness Mechanism with Bilevel Optimizationcited by 0
- 2026Learning the Robustness Mechanism with Bilevel Optimization28 Sep 2026
- 2026Black-Box Assisted Regression: Phase Transitions and Minimax Optimality24 Jun 2026
- 2026Learning from Limited Labels: Transductive Graph Label Propagation for Indian Music Analysis7 Jan 2026
- 2025Mitigating Spurious Correlation via Distributionally Robust Learning with Hierarchical Ambiguity Sets3 Oct 2025
- 2025Exploiting Vision Language Model for Training-Free 3D Point Cloud OOD Detection via Graph Score Propagation27 Jun 2025
- 2025Mitigating Confounding in Speech-Based Dementia Detection through Weight Masking5 Jun 2025
- 2025References Indeed Matter? Reference-Free Preference Optimization for Conversational Query Reformulation10 May 2025
- 2025Discrepancies are Virtue: Weak-to-Strong Generalization through Lens of Intrinsic Dimension7 Feb 2025
- 2025Analyzing Similarity Metrics for Data Selection for Language Model Pretraining4 Feb 2025
- 2024The Road to Artificial SuperIntelligence: A Comprehensive Survey of Superalignment21 Dec 2024
- 2024Optimizing importance weighting in the presence of sub-population shifts18 Oct 2024
- 2024Theoretical Analysis of Weak-to-Strong Generalization25 May 2024
36 more
- 2024Bridging Domains with Approximately Shared Features11 Mar 2024
- 2024Controllable Prompt Tuning For Balancing Group Distributional Robustness5 Mar 2024
- 2023Complementary Benefits of Contrastive Learning and Self-Training Under Distribution Shift6 Dec 2023
- 2023IW-GAE: Importance Weighted Group Accuracy Estimation for Improved Calibration and Model Selection in Unsupervised Domain Adaptation16 Oct 2023
- 2023Understanding the Robustness of Multi-modal Contrastive Learning to Distribution Shift8 Oct 2023
- 2023Controlling Federated Learning for Covertness17 Aug 2023
- 2023Cluster-aware Semi-supervised Learning: Relational Knowledge Distillation Provably Learns Clustering20 Jul 2023
- 2023DAC-MR: Data Augmentation Consistency Based Meta-Regularization for Meta-Learning13 May 2023
- 2023GeoNet: Benchmarking Unsupervised Adaptation across Geographies27 Mar 2023
- 2023Pseudo-Labeling for Kernel Ridge Regression under Covariate Shift20 Feb 2023
- 2023On the Stability and Generalization of Triplet Learning20 Feb 2023
- 2023Pushing the Accuracy-Group Robustness Frontier with Introspective Self-play11 Feb 2023
- 2023Chaos to Order: A Label Propagation Perspective on Source-Free Domain Adaptation20 Jan 2023
- 2022First Steps Toward Understanding the Extrapolation of Nonlinear Models to Unseen Domains21 Nov 2022
- 2022Divide and Contrast: Source-free Domain Adaptation via Adaptive Contrastive Learning12 Nov 2022
- 2022Cross-client Label Propagation for Transductive and Semi-Supervised Federated Learning12 Oct 2022
- 2022Label Propagation with Weak Supervision7 Oct 2022
- 2022GOCA: Guided Online Cluster Assignment for Self-Supervised Video Representation Learning20 Jul 2022
- 2022Dual-Correction Adaptation Network for Noisy Knowledge Transfer10 Jul 2022
- 2022Invariant and Transportable Representations for Anti-Causal Domain Shifts4 Jul 2022
- 2022Transferring Fairness under Distribution Shifts via Fair Consistency Regularization26 Jun 2022
- 2022Understanding new tasks through the lens of training data via exponential tilting26 May 2022
- 2022Localized Adversarial Domain Generalization9 May 2022
- 2022Can domain adaptation make object recognition work for everyone?23 Apr 2022
- 2022Beyond Separability: Analyzing the Linear Transferability of Contrastive Representations to Related Subpopulations6 Apr 2022
- 2022Connect, Not Collapse: Explaining Contrastive Learning for Unsupervised Domain Adaptation1 Apr 2022
- 2022Sample Efficiency of Data Augmentation Consistency Regularization24 Feb 2022
- 2022Fine-Tuning can Distort Pretrained Features and Underperform Out-of-Distribution21 Feb 2022
- 2021Encoding Hierarchical Information in Neural Networks helps in Subpopulation Shift20 Dec 2021
- 2021Can semi-supervised learning reduce the amount of manual labelling required for effective radio galaxy morphology classification?8 Nov 2021
- 2021Self-training Converts Weak Learners to Strong Learners in Mixture Models25 Jun 2021
- 2021Near-Optimal Linear Regression under Distribution Shift23 Jun 2021
- 2021Towards a Theoretical Framework of Out-of-Distribution Generalization8 Jun 2021
- 2021Provable Guarantees for Self-Supervised Deep Learning with Spectral Contrastive Loss8 Jun 2021
- 2021If your data distribution shifts, use self-learning27 Apr 2021
- 2021Cycle Self-Training for Domain Adaptation5 Mar 2021
- 2022Fine-Tuning can Distort Pretrained Features and Underperform Out-of-Distributioncited by 722
- 2021Provable Guarantees for Self-Supervised Deep Learning with Spectral Contrastive Losscited by 311
- 2021Cycle Self-Training for Domain Adaptationcited by 120
- 2021Towards a Theoretical Framework of Out-of-Distribution Generalizationcited by 94
- 2022Connect, Not Collapse: Explaining Contrastive Learning for Unsupervised Domain Adaptationcited by 66
- 2022Divide and Contrast: Source-free Domain Adaptation via Adaptive Contrastive Learningcited by 53
- 2024Theoretical Analysis of Weak-to-Strong Generalizationcited by 46
- 2022Beyond Separability: Analyzing the Linear Transferability of Contrastive Representations to Related Subpopulationscited by 44
- 2021Near-Optimal Linear Regression under Distribution Shiftcited by 40
- 2022Invariant and Transportable Representations for Anti-Causal Domain Shiftscited by 37
- 2021If your data distribution shifts, use self-learningcited by 36
- 2022Sample Efficiency of Data Augmentation Consistency Regularizationcited by 28
12 most cited of 45 All 45 (JSON)
- 2026Learning the Robustness Mechanism with Bilevel Optimization28 Sep 2026
- 2026Black-Box Assisted Regression: Phase Transitions and Minimax Optimality24 Jun 2026
- 2026Learning from Limited Labels: Transductive Graph Label Propagation for Indian Music Analysis7 Jan 2026
- 2025Mitigating Spurious Correlation via Distributionally Robust Learning with Hierarchical Ambiguity Sets3 Oct 2025
- 2025Mitigating Confounding in Speech-Based Dementia Detection through Weight Masking5 Jun 2025
- 2025Discrepancies are Virtue: Weak-to-Strong Generalization through Lens of Intrinsic Dimension7 Feb 2025
- 2025Analyzing Similarity Metrics for Data Selection for Language Model Pretraining4 Feb 2025
- 2024Optimizing importance weighting in the presence of sub-population shifts18 Oct 2024
- 2024Theoretical Analysis of Weak-to-Strong Generalization25 May 2024
- 2024Bridging Domains with Approximately Shared Features11 Mar 2024
- 2024Controllable Prompt Tuning For Balancing Group Distributional Robustness5 Mar 2024
- 2023Complementary Benefits of Contrastive Learning and Self-Training Under Distribution Shift6 Dec 2023
12 newest of 45 All 45 (JSON)