Enhancing Sentence Embedding with Generalized Pooling

By Qian Chen, Zhenhua Ling, Xiao-Dan Zhu. 2018

76 citations recorded by the source corpus.

Pooling is an essential component of a wide variety of sentence representation and embedding models. This paper explores generalized pooling methods to enhance sentence embedding. We propose vector-based multi-head attention that includes the widely used max pooling, mean pooling, and scalar self-attention as special cases. The model benefits from properly designed penalization terms to reduce redundancy in multi-head attention. We evaluate the proposed model on three different tasks: natural language inference (NLI), author profiling, and sentiment classification. The experiments show that the proposed model achieves significant improvement over strong sentence-encoding-based methods, resulting in state-of-the-art performances on four datasets. The proposed approach can be easily implemented for more problems than we discuss in this paper.

Subfields: Artificial Intelligence, Natural Language Processing

How researchers see this paper

The researchers whose trusted networks rank it highest, strongest advocate first. Every researcher gets a different order.

In Artificial Intelligence

  1. Jianfeng Gao
  2. Zhenhua Ling
  3. Mohit Bansal
  4. Guodong Long
  5. Xiao-Dan Zhu
  6. Erik Cambria
  7. Jing Jiang
  8. Qian Chen

In Natural Language Processing

  1. Zhenhua Ling
  2. Jianfeng Gao
  3. Qian Chen
  4. Erik Cambria
  5. Xiao-Dan Zhu
  6. Mohit Bansal
  7. Wen Wang
  8. William Yang Wang

Citations in the corpus