Variational Hybrid-Attention Framework for Multi-Label Few-Shot Aspect Category Detection

Authors

  • Cheng Peng The State Key Laboratory of Blockchain and Data Security, Zhejiang University
  • Ke Chen The State Key Laboratory of Blockchain and Data Security, Zhejiang University
  • Lidan Shou The State Key Laboratory of Blockchain and Data Security, Zhejiang University
  • Gang Chen The State Key Laboratory of Blockchain and Data Security, Zhejiang University

DOI:

https://doi.org/10.1609/aaai.v38i13.29375

Keywords:

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

Abstract

Multi-label few-shot aspect category detection (FS-ACD) is a challenging sentiment analysis task, which aims to learn a multi-label learning paradigm with limited training data. The difficulty of this task is how to use limited data to generalize effective discriminative representations for different categories. Nowadays, all advanced FS-ACD works utilize the prototypical network to learn label prototypes to represent different aspects. However, such point-based estimation methods are inherently noise-susceptible and bias-vulnerable. To this end, this paper proposes a novel Variational Hybrid-Attention Framework (VHAF) for the FS-ACD task. Specifically, to alleviate the data noise, we adopt a hybrid-attention mechanism to generate more discriminative aspect-specific embeddings. Then, based on these embeddings, we introduce the variational distribution inference to obtain the aspect-specific distribution as a more robust aspect representation, which can eliminate the scarce data bias for better inference. Moreover, we further leverage an adaptive threshold estimation to help VHAF better identify multiple relevant aspects. Extensive experiments on three datasets demonstrate the effectiveness of our VHAF over other state-of-the-art methods. Code is available at https://github.com/chengzju/VHAF.

Published

2024-03-24

How to Cite

Peng, C., Chen, K., Shou, L., & Chen, G. (2024). Variational Hybrid-Attention Framework for Multi-Label Few-Shot Aspect Category Detection. Proceedings of the AAAI Conference on Artificial Intelligence, 38(13), 14590-14598. https://doi.org/10.1609/aaai.v38i13.29375

Issue

Section

AAAI Technical Track on Machine Learning IV