mlsanityData to 7 Oct 2026

Out-of-Distribution Learning with Human Feedback

arXiv 2408.07772 ↗First posted 14 Aug 2024cs.LG stat.MLcited by 6

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

  1. 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,456
  2. 2017mixup: Beyond Empirical Risk Minimizationcited by 5,155
  3. 2015Dropout as a Bayesian Approximation: Representing Model Uncertainty in Deep Learningcited by 4,416
  4. 2016Wide Residual Networkscited by 3,057
  5. 2015Domain-Adversarial Training of Neural Networkscited by 3,023
  6. 2016A Baseline for Detecting Misclassified and Out-of-Distribution Examples in Neural Networkscited by 2,362
  7. 2021Learning to Prompt for Vision-Language Modelscited by 2,269
  8. 2013Describing Textures in the Wildcited by 2,110
  9. 2019Invariant Risk Minimizationcited by 1,832
  10. 2018A Simple Unified Framework for Detecting Out-of-Distribution Samples and Adversarial Attackscited by 1,472
  11. 2022Conditional Prompt Learning for Vision-Language Modelscited by 1,427
  12. 2020WILDS: A Benchmark of in-the-Wild Distribution Shiftscited by 1,366
38 more
  1. 2015LSUN: Construction of a Large-scale Image Dataset using Deep Learning with Humans in the Loopcited by 1,350
  2. 2017Enhancing The Reliability of Out-of-distribution Image Detection in Neural Networkscited by 1,325
  3. 2020Energy-based Out-of-distribution Detectioncited by 1,296
  4. 2019Distributionally Robust Neural Networks for Group Shifts: On the Importance of Regularization for Worst-Case Generalizationcited by 1,232
  5. 2016Deep CORAL: Correlation Alignment for Deep Domain Adaptationcited by 1,194
  6. 2020In Search of Lost Domain Generalizationcited by 1,042
  7. 2017Deeper, Broader and Artier Domain Generalizationcited by 983
  8. 2018Deep Anomaly Detection with Outlier Exposurecited by 939
  9. 2021Generalized Out-of-Distribution Detection: A Surveycited by 796
  10. 2017Learning to Generalize: Meta-Learning for Domain Generalizationcited by 762
  11. 2021Domain Generalization: A Surveycited by 760
  12. 2020Out-of-Distribution Generalization via Risk Extrapolation (REx)cited by 737
  13. 2019Deep Batch Active Learning by Diverse, Uncertain Gradient Lower Boundscited by 602
  14. 2022Out-of-Distribution Detection with Deep Nearest Neighborscited by 596
  15. 2021Domain Generalization with MixStylecited by 563
  16. 2019A Simple Baseline for Bayesian Uncertainty in Deep Learningcited by 555
  17. 2020A Survey of Deep Active Learningcited by 547
  18. 2022Continual Test-Time Domain Adaptationcited by 520
  19. 2021ReAct: Out-of-distribution Detection With Rectified Activationscited by 486
  20. 2020Self-Challenging Improves Cross-Domain Generalizationcited by 426
  21. 2022ViM: Out-Of-Distribution with Virtual-logit Matchingcited by 404
  22. 2021On the Importance of Gradients for Detecting Distributional Shifts in the Wildcited by 338
  23. 2018Why ReLU networks yield high-confidence predictions far away from the training data and how to mitigate the problemcited by 323
  24. 2022VOS: Learning What You Don't Know by Virtual Outlier Synthesiscited by 300
  25. 2022OpenOOD: Benchmarking Generalized Out-of-Distribution Detectioncited by 267
  26. 2021Gradient Matching for Domain Generalizationcited by 258
  27. 2020Being Bayesian, Even Just a Bit, Fixes Overconfidence in ReLU Networkscited by 227
  28. 2022Extremely Simple Activation Shaping for Out-of-Distribution Detectioncited by 212
  29. 2021SelfReg: Self-supervised Contrastive Regularization for Domain Generalizationcited by 189
  30. 2020Invariant Risk Minimization Gamescited by 188
  31. 2021Fishr: Invariant Gradient Variances for Out-of-Distribution Generalizationcited by 186
  32. 2020Adaptive Risk Minimization: Learning to Adapt to Domain Shiftcited by 183
  33. 2017Domain Generalization by Marginal Transfer Learningcited by 180
  34. 2023OpenOOD v1.5: Enhanced Benchmark for Out-of-Distribution Detectioncited by 178
  35. 2018Adversarial Active Learning for Deep Networks: a Margin Based Approachcited by 155
  36. 2020Understanding Self-Training for Gradual Domain Adaptationcited by 149
  37. 2023Non-Parametric Outlier Synthesiscited by 129
  38. 2022How to Exploit Hyperspherical Embeddings for Out-of-Distribution Detection?cited by 128

50 most cited of 106 All 106 (JSON)

Built on by

Cited by year (2026 to 7 Oct)

Citing papers per year: 2024: 3; 2025: 0; 2026: 3.

6 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2024Taming False Positives in Out-of-Distribution Detection with Human Feedback
  2. 2023Class Relevance Learning For Out-of-distribution Detection
  3. 2023Distribution Shift Inversion for Out-of-Distribution Prediction
  4. 2025A Benchmark and Evaluation for Real-World Out-of-Distribution Detection Using Vision-Language Models
  5. 2020A Critical Evaluation of Open-World Machine Learning
  6. 2024Non-Linear Outlier Synthesis for Out-of-Distribution Detection
  7. 2025Retrieval-Augmented Prompt for OOD Detection
  8. 2024CRoFT: Robust Fine-Tuning with Concurrent Optimization for OOD Generalization and Open-Set OOD Detection