English

Learning Slice-Aware Representations with Mixture of Attentions

Machine Learning 2021-06-07 v1 Computation and Language

Abstract

Real-world machine learning systems are achieving remarkable performance in terms of coarse-grained metrics like overall accuracy and F-1 score. However, model improvement and development often require fine-grained modeling on individual data subsets or slices, for instance, the data slices where the models have unsatisfactory results. In practice, it gives tangible values for developing such models that can pay extra attention to critical or interested slices while retaining the original overall performance. This work extends the recent slice-based learning (SBL)~\cite{chen2019slice} with a mixture of attentions (MoA) to learn slice-aware dual attentive representations. We empirically show that the MoA approach outperforms the baseline method as well as the original SBL approach on monitored slices with two natural language understanding (NLU) tasks.

Keywords

Cite

@article{arxiv.2106.02363,
  title  = {Learning Slice-Aware Representations with Mixture of Attentions},
  author = {Cheng Wang and Sungjin Lee and Sunghyun Park and Han Li and Young-Bum Kim and Ruhi Sarikaya},
  journal= {arXiv preprint arXiv:2106.02363},
  year   = {2021}
}

Comments

Findings of the ACL: ACL-IJCNLP 2021

R2 v1 2026-06-24T02:49:56.296Z