Exploring Channel-Aware Typical Features for Out-of-Distribution Detection

Authors

  • Rundong He Shandong University
  • Yue Yuan Shandong University
  • Zhongyi Han Mohamed bin Zayed University of Artificial Intelligence
  • Fan Wang Shandong University
  • Wan Su Shandong University
  • Yilong Yin Shandong University
  • Tongliang Liu The University of Sydney Mohamed bin Zayed University of Artificial Intelligence
  • Yongshun Gong Shandong University

DOI:

https://doi.org/10.1609/aaai.v38i11.29132

Keywords:

ML: Transfer, Domain Adaptation, Multi-Task Learning, DMKM: Anomaly/Outlier Detection, ML: Multi-class/Multi-label Learning & Extreme Classification

Abstract

Detecting out-of-distribution (OOD) data is essential to ensure the reliability of machine learning models when deployed in real-world scenarios. Different from most previous test-time OOD detection methods that focus on designing OOD scores, we delve into the challenges in OOD detection from the perspective of typicality and regard the feature’s high-probability region as the feature’s typical set. However, the existing typical-feature-based OOD detection method implies an assumption: the proportion of typical feature sets for each channel is fixed. According to our experimental analysis, each channel contributes differently to OOD detection. Adopting a fixed proportion for all channels results in several channels losing too many typical features or incorporating too many abnormal features, resulting in low performance. Therefore, exploring the channel-aware typical features is crucial to better-separating ID and OOD data. Driven by this insight, we propose expLoring channel-Aware tyPical featureS (LAPS). Firstly, LAPS obtains the channel-aware typical set by calibrating the channel-level typical set with the global typical set from the mean and standard deviation. Then, LAPS rectifies the features into channel-aware typical sets to obtain channel-aware typical features. Finally, LAPS leverages the channel-aware typical features to calculate the energy score for OOD detection. Theoretical and visual analyses verify that LAPS achieves a better bias-variance trade-off. Experiments verify the effectiveness and generalization of LAPS under different architectures and OOD scores.

Published

2024-03-24

How to Cite

He, R., Yuan, Y., Han, Z., Wang, F., Su, W., Yin, Y., Liu, T., & Gong, Y. (2024). Exploring Channel-Aware Typical Features for Out-of-Distribution Detection. Proceedings of the AAAI Conference on Artificial Intelligence, 38(11), 12402-12410. https://doi.org/10.1609/aaai.v38i11.29132

Issue

Section

AAAI Technical Track on Machine Learning II