Efficient Uncertainty Quantification for Multilabel Text Classification

Yu, Jialin; Cristea, Alexandra I.; Harit, Anoushka; Sun, Zhongtian; Aduragba, Olanrewaju Tahir; Shi, Lei; Al Moubayed, Noura

Abstract

Despite rapid advances of modern artificial intelligence (AI), there is a growing concern regarding its capacity to be explainable, transparent, and accountable. One crucial step towards such AI systems involves reliable and efficient uncertainty quantification methods. Existing approaches to uncertainty quantification in natural language processing (NLP) take a Bayesian Deep Learning approach. However, the latter is known to not be computationally efficient in testing time, thus hindering its applicability in real-life scenarios. This paper proposes a new focus on the efficiency of uncertainty quantification methods, evaluating them on four multi-label text classification tasks. Our novel methods of representing epistemic and aleatoric uncertainties enable efficient uncertainty quantification (around 13 to 45 times faster than existing approaches, depending on architecture) with posterior analysis in the (approximated) latent- and data space. We conduct extensive experiments and studies on diverse neural network architectures (LSTM, CNN and Transformer) to analyse their power. Our results prove the benefits of explicitly modelling uncertainty in neural networks.

Citation

Yu, J., Cristea, A. I., Harit, A., Sun, Z., Aduragba, O. T., Shi, L., & Al Moubayed, N. (2022, July). Efficient Uncertainty Quantification for Multilabel Text Classification. Presented at 2022 International Joint Conference on Neural Networks (IJCNN), Padova, Italy

• Presentation Conference Type: Conference Paper (published)
• Conference Name: 2022 International Joint Conference on Neural Networks (IJCNN)
• Start Date: Jul 18, 2022
• End Date: Jul 23, 2022
• Acceptance Date: Apr 26, 2022
• Online Publication Date: Sep 30, 2022
• Publication Date: 2022
• Deposit Date: Aug 31, 2022
• Publicly Available Date: Sep 1, 2022
• Series ISSN: 2161-4393,2161-4407
• DOI: https://doi.org/10.1109/ijcnn55064.2022.9892871
• Public URL: https://durham-repository.worktribe.com/output/1136303