AHA: Human-Assisted Out-of-Distribution Generalization and Detection

Modern machine learning models deployed often encounter distribution shifts in real-world applications, manifesting as covariate or semantic out-of-distribution (OOD) shifts. These shifts give rise to challenges in OOD generalization and OOD detection. This paper introduces a novel, integrated approach AHA (Adaptive Human-Assisted OOD learning) to simultaneously address both OOD generalization and detection through a human-assisted framework by labeling data in the wild. Our approach strategically labels examples within a novel maximum disambiguation region, where the number of semantic and covariate OOD data roughly equalizes. By labeling within this region, we can maximally disambiguate the two types of OOD data, thereby maximizing the utility of the fixed labeling budget. Our algorithm first utilizes a noisy binary search algorithm that identifies the maximal disambiguation region with high probability. The algorithm then continues with annotating inside the identified labeling region, reaping the full benefit of human feedback. Extensive experiments validate the efficacy of our framework. We observed that with only a few hundred human annotations, our method significantly outperforms existing state-of-the-art methods that do not involve human assistance, in both OOD generalization and OOD detection. Code is publicly available at \url{https://github.com/HaoyueBaiZJU/aha}.

Paper

References (100)

Scroll for more · 38 remaining

Similar papers

Peer review

Reviewer CyPp7/10 · confidence 4/52024-06-13

Summary

The paper presents a novel approach to the problem of OOD generalization and detection. The authors introduce the AHA (Adaptive Human-Assisted OOD learning) framework, which aims to enhance both out-of-distribution (OOD) generalization and detection by strategically leveraging human-assisted labeling within a maximum disambiguation region. The paper reports significant improvements over state-of-the-art methods with only a few hundred human annotations, demonstrating the efficacy of the proposed framework.

Strengths

- The AHA framework is a creative solution that addresses the challenges of OOD generalization and detection, which are critical for real-world applications of machine learning models. - The authors provide extensive experimental results that validate the effectiveness of their approach, showing robust performance across various datasets. - The incorporation of human feedback in a strategic manner is a strength, as it capitalizes on the limited labeling budget to maximize model performance. - The paper's contributions are articulated, with the novel labeling strategy and the integration of human assistance being the highlights. - The transformation of the problem into a noisy binary search is an intelligent methodological choice that allows for the efficient identification of the maximum ambiguity threshold.

Weaknesses

- While the paper demonstrates strong results, it is not clear how the AHA framework scales with larger and more complex datasets. - The reliance on human annotations could be a limitation in scenarios where such resources are not readily available or are cost-prohibitive. - The paper could benefit from a discussion on how the findings generalize beyond the tested datasets and scenarios. - The computational complexity of the AHA algorithm and its runtime performance on large datasets are not discussed. - The paper could address potential biases introduced by human labeling, especially in the context of OOD detection.

Questions

- How does the AHA framework perform as the size and complexity of the dataset increase? - What are the specific steps taken to mitigate potential biases in human labeling? - Can the authors provide more details on the computational efficiency of the AHA algorithm, especially for large-scale applications? - How does the framework handle a class imbalance in the context of OOD detection? - Are there any specific domains or applications where the AHA framework is expected to be more or less effective, and why? - Could the proposed method benefit the OOD detection with unreliable sources [R1] and inspire unsupervised OOD detection [R2]? ---- [R1] Out-of-distribution detection learning with unreliable out-of-distribution sources. NeurIPS 2023. [R2] Out-of-distribution detection with an adaptive likelihood ratio on informative hierarchical vae. NeurIPS 2022.

Rating

7

Confidence

4

Soundness

3

Presentation

3

Contribution

3

Limitations

NA.

Reviewer Rqpf5/10 · confidence 4/52024-06-24

Summary

This paper proposes to address both out-of-distribution detection and generalization within one joint framework under human-assistance. The proposed method first utilizes a noisy binary search algorithm to identify the most informative samples to be labeled. Then, it continues to annotate these samples with human feedback. The authors conduct experiments on CIFAR and PACS to evaluate the proposed method.

Strengths

- The proposed method can handle OOD detection and generalization at the same time, which is impactful to both these two individual research areas. - Covariat-shifts and semantic shifts are both inevitable in real-world applications. Thus the proposed method is straightforward and well-motivated. - Most parts of this paper are well presented with good visualization.

Weaknesses

- AHA may only work under a rather strict assumption. Compared to outlier exposure, AHA needs to access the real test data distribution $S_{wild}$ to selectively label some samples. While outlier exposure does not need such an assumption. Many previous OOD detection methods can only access training data distribution and an auxiliary OOD data distribution (noted that such auxiliary OOD has no overlapped samples with test-time OOD data in common settings). Thus I think AHA may only work under a more strict assumption (i.e., the test data distribution $S_{wild}$ is accessible) than previous outlier exposure. - I generally believe enhancing OOD detection and generalization with human feedback is laborious. I am aware the proposed method can get good performance with hundreds or thousands of labeled samples in many cases. However, such human feedback still seems laborious to me. For example, in CIFAR experiments, the images are only 32*32. Thus I think it would take a lot of time to label such samples. Not to mention the samples can be noised or corrupted (CIFAR-10-C). - The experiments are not adequate. It is widely acknowledged that OOD detection and generalization are more difficult on large-scale high-resolution datasets. For example, ImageNet benchmarks. The authors do not conduct evaluations on such datasets. - In some cases, there may be no clear boundary between covariate-shift data and semantic shift data. As mentioned in recent work[1], these two types of OOD samples can be very hard to separate (even for humans). Could the authors comment on this phenomenon? Given the above points, I tend to reject this paper because the overall quality do not meet the expectations of NIPS. However, I may adjust my score if there is a strong argument. [1] Yang, William, Byron Zhang, and Olga Russakovsky. "ImageNet-OOD: Deciphering Modern Out-of-Distribution Detection Algorithms."

Questions

Please refer to weaknesses and address my concerns.

Rating

5

Confidence

4

Soundness

3

Presentation

3

Contribution

2

Limitations

N/A

Authorsrebuttal2024-08-09

Response to Reviewer Rqpf (Followup)

Thank you for taking the time to read our response. We address your comments below. > *1. I still have further concerns about the basic settings of this paper.* We would like to clarify that the wild data setting, which includes a mixture of ID, covariate OOD, and semantic OOD data, is commonly observed in practice. The overall wild mixture data distribution $P_\text{wild}$ differs from the test environment data distribution. Additionally, the mixing ratios of $π_s$ and $π_c$ are unknown in our formulation, making this setting well-suited to real-world scenarios with varied distributions of wild data. We have also empirically tested different mixing ratios, as detailed in **Appendix H**, where we demonstrate the robust and strong performance of our AHA framework. Moreover, as suggested, we have summarized the OOD accuracy results on covariate test data when the model is exposed to one type of corruption but tested on another. Specifically, we exposed the model to wild data containing Gaussian noise corruption and then tested it on nine other types of corruption: impulse noise, spatter, shot noise, saturate, speckle noise, frosted glass blur, motion blur, frost, and zoom blur. The results indicate AHA displays strong performance, even when the test-time covariate-shifted distribution remains unknown. | Algorithm | Impulse noise | Spatter | Shot noise | Saturate | Speckle noise | Frosted glass blur | Motion blur | Frost | Zoom blur | Average | | --------- | :-------------: | :----------------: | :----------: | :----------: | :-------------: | :---------: | :------------------: | :---------------: | :----------: | :----------: | | ERM | $85.15$ | $92.88$ | $84.62$ | $92.49$ | $84.46$ | $52.29$ | $89.10$ | $89.67$ | $85.55$ | $84.02$ | | AHA (ours)| **87.25** | **93.03** | **90.90** | **92.72** | **90.77** | **63.85** | **89.35** | **91.52** | **86.58** | **87.33** | > *2. Regarding the argument for W2.* Thank you for your constructive comments. We will incorporate discussions about the cost of additional labeling in the paper as suggested.

Reviewer Rqpf2024-08-12

Thanks for your response and hard work during the rebuttal. I am pleased to acknowledge that this paper has no significant flaws and the additional experiments are well-appreciated. However, I agree with other reviewers that the need for human assistance is both an advantage and an inevitable weakness of AHA. The uncommon settings should also be carefully explained and compared with classic domain adaption or OOD generalization. It seems to be a rather strict assumption than those used in domain adaption or OOD generalization (accessing samples drawn from test-time covariant-shift distribution and labeling some of them). Removing such an assumption can greatly strengthen the proposed method (e.g., involving various types of corruption). Theoretical support of why unlabeled data can help both OOD detection and generalization in AHA can be also taken into consideration in future work. With a better understanding of this paper, I have adjusted my score accordingly.

Authorsrebuttal2024-08-12

Thank you for your insightful feedback and constructive comments, which have been invaluable in enhancing our manuscript. We will incorporate the additional results (including various types of corruptions) and discussions in the paper as suggested. Regarding the need for human assistance, there are many applications where human annotation is particularly useful, such as medical diagnostics. In these contexts, human assistance provides crucial expertise and contextual information, and should be considered a significant advantage rather than a limitation.

Reviewer Eoxk5/10 · confidence 4/52024-07-13

Summary

This paper introduces a novel, integrated approach AHA (Adaptive Human-Assisted OOD learning) to simultaneously address both OOD generalization and detection through a human-assisted framework by labeling data in the wild. Extensive experiments validate the efficacy of AHA.

Strengths

1. this paper is well written and easy to follow 2. good visualization and extensive experiments 3. Maximum Disambiguation Region and reduction to noisy binary search are new to me

Weaknesses

1. there exsits a strong assumption that the weighted densities of semantic and covariance ood should equalize 2. what is difference between active learning and the proposed Human-Assisted Learning 3. what is the time used for noisy binary search 4. whether the lamda searched on the one dataset can be transfer to another dataset

Questions

see weakness

Rating

5

Confidence

4

Soundness

3

Presentation

3

Contribution

3

Limitations

see weakness

Reviewer Rqpf2024-08-08

I thank the authors for their detailed explanation in the rebuttal. My concern about performance on high-resolution large-scale benchmarks has been addressed (w3). The argument for w2 is also OK for me. I recommend adding these to the paper to enhance the transparency about the cost of additional labeling. However, I still have further concerns about the basic settings of this paper. AHA assumes that there is an overlap between test-time covariant-shifted data distribution and training-time unlabeled data. For instance, it assumes that during training, the model can access samples corrupted by Gaussian noise. It is wired to me and may be a rather strict condition compared to that used in standard OOD generalization methods. What if we keep the test-time covariant-shifted distribution unknown? For example, only expose the model to one certain type of corruption, but test it on another type of corruption.

Area Chair AV1R2024-08-08

The discussion has begun

Dear Reviewers, Thank you for taking the time to review the paper. The discussion has begun, and active participation is highly appreciated and recommended. Thanks for your continued efforts and contributions to NeurIPS 2024. Best regards, Your Area Chair

Reviewer CyPp2024-08-08

Thanks for your feedback. It addresses my concerns, especially experiments on large-scale datasets. I have decided to increase the rating on soundness and my confidence. The need for human assistance is both the motivation of this work and its inevitable weakness, depending on the task requirements and the specific scenario. Overall, I am positive about this work (weak acceptance). However, I also would like to hear other reviewers' opinions and discuss this, and make further judgments.

Reviewer Eoxk2024-08-09

Official Comment by Reviewer Eoxk

Thank the authors for answering my questions. I would like to keep my rating "Borderline accept".

Authorsrebuttal2024-08-13

Response to Reviewer CyPp (Followup)

Thank you again for your feedback and comments. We address your comments below. > *The need for human assistance is both the motivation of this work and its inevitable weakness.* We would like to clarify that there are many scenarios where human annotation is particularly valuable, for example, in medical diagnostics. In these contexts, human assistance provides crucial expertise and contextual information, which should be considered a significant advantage rather than a weakness. We would greatly appreciate an increased overall rating if we have successfully answered your questions. Otherwise, we are happy to provide additional discussions to address any further concerns.

Program Chairsdecision2024-09-25

Decision

Accept (poster)

© 2026 NYSGPT2525 LLC