Out-of-Distribution Learning with Human Feedback
On the map: LLM alignment / RLHF / preference optimisation → Reward modeling and RLHF
Abstract (arXiv)
Out-of-distribution (OOD) learning often relies heavily on statistical approaches or predefined assumptions about OOD data distributions, hindering their efficacy in addressing multifaceted challenges of OOD generalization and OOD detection in real-world deployment environments. This paper presents a novel framework for OOD learning with human feedback, which can provide invaluable insights into the nature of OOD shifts and guide effective model adaptation. Our framework capitalizes on the freely available unlabeled data in the wild that captures the environmental test-time OOD distributions under both covariate and semantic shifts. To harness such data, our key idea is to selectively provide human feedback and label a small number of informative samples from the wild data distribution, which are then used to train a multi-class classifier and an OOD detector. By exploiting human feedback, we enhance the robustness and reliability of machine learning models, equipping them with the capability to handle OOD scenarios with greater precision. We provide theoretical insights on the generalization error bounds to justify our algorithm. Extensive experiments show the superiority of our method, outperforming the current state-of-the-art by a significant margin.
Key facts. Cited by 6 papers in the map (data through 7 Oct 2026). Placed in Reward modeling and RLHF, one of 10 directions in LLM alignment / RLHF / preference optimisation. Builds on 106 of its 150 references that are in the map. First posted on arXiv 14 Aug 2024.
Builds on
106 of 150 references.
References · 106
- 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,456
- 2017mixup: Beyond Empirical Risk Minimizationcited by 5,155
- 2015Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learningcited by 4,416
- 2016Wide Residual Networkscited by 3,057
- 2015Domain-Adversarial Training of Neural Networkscited by 3,023
- 2016A Baseline for Detecting Misclassified and Out-of-Distribution Examples in Neural Networkscited by 2,362
- 2021Learning to Prompt for Vision-Language Modelscited by 2,269
- 2013Describing Textures in the Wildcited by 2,110
- 2019Invariant Risk Minimizationcited by 1,832
- 2018A Simple Unified Framework for Detecting Out-of-Distribution Samples and Adversarial Attackscited by 1,472
- 2022Conditional Prompt Learning for Vision-Language Modelscited by 1,427
- 2020WILDS: A Benchmark of in-the-Wild Distribution Shiftscited by 1,366
38 more
- 2015LSUN: Construction of a Large-scale Image Dataset using Deep Learning with Humans in the Loopcited by 1,350
- 2017Enhancing The Reliability of Out-of-distribution Image Detection in Neural Networkscited by 1,325
- 2020Energy-based Out-of-distribution Detectioncited by 1,296
- 2019Distributionally Robust Neural Networks for Group Shifts: On the Importance of Regularization for Worst-Case Generalizationcited by 1,232
- 2016Deep CORAL: Correlation Alignment for Deep Domain Adaptationcited by 1,194
- 2020In Search of Lost Domain Generalizationcited by 1,042
- 2017Deeper, Broader and Artier Domain Generalizationcited by 983
- 2018Deep Anomaly Detection with Outlier Exposurecited by 939
- 2021Generalized Out-of-Distribution Detection: A Surveycited by 796
- 2017Learning to Generalize: Meta-Learning for Domain Generalizationcited by 762
- 2021Domain Generalization: A Surveycited by 760
- 2020Out-of-Distribution Generalization via Risk Extrapolation (REx)cited by 737
- 2019Deep Batch Active Learning by Diverse, Uncertain Gradient Lower Boundscited by 602
- 2022Out-of-Distribution Detection with Deep Nearest Neighborscited by 596
- 2021Domain Generalization with MixStylecited by 563
- 2019A Simple Baseline for Bayesian Uncertainty in Deep Learningcited by 555
- 2020A Survey of Deep Active Learningcited by 547
- 2022Continual Test-Time Domain Adaptationcited by 520
- 2021ReAct: Out-of-distribution Detection With Rectified Activationscited by 486
- 2020Self-Challenging Improves Cross-Domain Generalizationcited by 426
- 2022ViM: Out-Of-Distribution with Virtual-logit Matchingcited by 404
- 2021On the Importance of Gradients for Detecting Distributional Shifts in the Wildcited by 338
- 2018Why ReLU networks yield high-confidence predictions far away from the training data and how to mitigate the problemcited by 323
- 2022VOS: Learning What You Don't Know by Virtual Outlier Synthesiscited by 300
- 2022OpenOOD: Benchmarking Generalized Out-of-Distribution Detectioncited by 267
- 2021Gradient Matching for Domain Generalizationcited by 258
- 2020Being Bayesian, Even Just a Bit, Fixes Overconfidence in ReLU Networkscited by 227
- 2022Extremely Simple Activation Shaping for Out-of-Distribution Detectioncited by 212
- 2021SelfReg: Self-supervised Contrastive Regularization for Domain Generalizationcited by 189
- 2020Invariant Risk Minimization Gamescited by 188
- 2021Fishr: Invariant Gradient Variances for Out-of-Distribution Generalizationcited by 186
- 2020Adaptive Risk Minimization: Learning to Adapt to Domain Shiftcited by 183
- 2017Domain Generalization by Marginal Transfer Learningcited by 180
- 2023OpenOOD v1.5: Enhanced Benchmark for Out-of-Distribution Detectioncited by 178
- 2018Adversarial Active Learning for Deep Networks: a Margin Based Approachcited by 155
- 2020Understanding Self-Training for Gradual Domain Adaptationcited by 149
- 2023Non-Parametric Outlier Synthesiscited by 129
- 2022How to Exploit Hyperspherical Embeddings for Out-of-Distribution Detection?cited by 128
50 most cited of 106 All 106 (JSON)
Built on by
Citing papers per year: 2024: 3; 2025: 0; 2026: 3.
6 papers.
- 2024HaloScope: Harnessing Unlabeled LLM Generations for Hallucination Detectioncited by 96
- 2024Process Reward Model with Q-Value Rankingscited by 78
- 2024Out-of-Distribution Detection: A Task-Oriented Survey of Recent Advancescited by 24
- 2026Unknown Aware AI-Generated Content Attributioncited by 0
- 2026Toward Robust In-Context Learning: Leveraging Out-of-distribution Proxies for Target Inaccessible Demonstration Retrievalcited by 0
- 2026CHILD: Human-in-the-Loop OOD Detection for Safe Clinical Deploymentcited by 0
- 2026CHILD: Human-in-the-Loop OOD Detection for Safe Clinical Deployment7 Sep 2026
- 2026Toward Robust In-Context Learning: Leveraging Out-of-distribution Proxies for Target Inaccessible Demonstration Retrieval13 Apr 2026
- 2026Unknown Aware AI-Generated Content Attribution1 Jan 2026
- 2024Process Reward Model with Q-Value Rankings15 Oct 2024
- 2024HaloScope: Harnessing Unlabeled LLM Generations for Hallucination Detection26 Sep 2024
- 2024Out-of-Distribution Detection: A Task-Oriented Survey of Recent Advances18 Sep 2024
Similar, not cited
Close in wording, no citation either way
- 2024Taming False Positives in Out-of-Distribution Detection with Human Feedback
- 2023Class Relevance Learning For Out-of-distribution Detection
- 2023Distribution Shift Inversion for Out-of-Distribution Prediction
- 2025A Benchmark and Evaluation for Real-World Out-of-Distribution Detection Using Vision-Language Models
- 2020A Critical Evaluation of Open-World Machine Learning
- 2024Non-Linear Outlier Synthesis for Out-of-Distribution Detection
- 2025Retrieval-Augmented Prompt for OOD Detection
- 2024CRoFT: Robust Fine-Tuning with Concurrent Optimization for OOD Generalization and Open-Set OOD Detection