Learned in Translation: Contextualized Word Vectors

Code

• github.com/salesforce/cove
  OfficialIn paperpytorch★ 0
• github.com/menajosep/AleatoricSent
  tf★ 2
• github.com/richinkabra/CoVe-BCN
  pytorch★ 0
• github.com/adi2103/AML-CoVe
  tf★ 0
• github.com/cove-adml/adml-anon
  pytorch★ 0

Abstract

Computer vision has benefited from initializing multiple deep layers with weights pretrained on large supervised training sets like ImageNet. Natural language processing (NLP) typically sees initialization of only the lowest layer of deep models with pretrained word vectors. In this paper, we use a deep LSTM encoder from an attentional sequence-to-sequence model trained for machine translation (MT) to contextualize word vectors. We show that adding these context vectors (CoVe) improves performance over using only unsupervised word and character vectors on a wide variety of common NLP tasks: sentiment analysis (SST, IMDb), question classification (TREC), entailment (SNLI), and question answering (SQuAD). For fine-grained sentiment analysis and entailment, CoVe improves performance of our baseline models to the state of the art.

Tasks

Benchmark Results

Reproductions