Enhancing Out-of-Distribution Detection in Natural Language Understanding via Implicit Layer Ensemble
Cho, Hyunsoo, Park, Choonghyun, Kang, Jaewook, Yoo, Kang Min, Kim, Taeuk, Lee, Sang-goo
–arXiv.org Artificial Intelligence
Out-of-distribution (OOD) detection aims to discern outliers from the intended data distribution, which is crucial to maintaining high reliability and a good user experience. Most recent studies in OOD detection utilize the information from a single representation that resides in the penultimate layer to determine whether the input is anomalous or not. Although such a method is straightforward, the potential of diverse information in the intermediate layers is overlooked. In this paper, we propose a novel framework based on contrastive learning that encourages intermediate features to learn layer-specialized representations and assembles them implicitly into a single representation to absorb rich information in the pre-trained language model. Extensive experiments in various intent classification and OOD datasets demonstrate that our approach is significantly more effective than other works.
arXiv.org Artificial Intelligence
Oct-20-2022
- Country:
- North America
- Cuba (0.04)
- United States > Louisiana
- Orleans Parish > New Orleans (0.04)
- Asia > South Korea
- North America
- Genre:
- Research Report (0.84)
- Industry:
- Media (0.46)
- Technology: