Modeling Discriminative Representations for Out-of-Domain Detection with Supervised Contrastive Learning
Abstract
Detecting Out-of-Domain (OOD) or unknown intents from user queries is essential in a task-oriented dialog system. A key challenge of OOD detection is to learn discriminative semantic features. Traditional cross-entropy loss only focuses on whether a sample is correctly classified, and does not explicitly distinguish the margins between categories. In this paper, we propose a supervised contrastive learning objective to minimize intra-class variance by pulling together in-domain intents belonging to the same class and maximize inter-class variance by pushing apart samples from different classes. Besides, we employ an adversarial augmentation mechanism to obtain pseudo diverse views of a sample in the latent space. Experiments on two public datasets prove the effectiveness of our method capturing discriminative representations for OOD detection.
Keywords
Cite
@article{arxiv.2105.14289,
title = {Modeling Discriminative Representations for Out-of-Domain Detection with Supervised Contrastive Learning},
author = {Zhiyuan Zeng and Keqing He and Yuanmeng Yan and Zijun Liu and Yanan Wu and Hong Xu and Huixing Jiang and Weiran Xu},
journal= {arXiv preprint arXiv:2105.14289},
year = {2021}
}
Comments
Accepted by ACL2021