Command Palette
Search for a command to run...
Bryan McCann; James Bradbury; Caiming Xiong; Richard Socher
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.
Code Repositories
Benchmarks
| Benchmark | Methodology | Metrics |
|---|---|---|
| natural-language-inference-on-snli | Biattentive Classification Network + CoVe + Char | % Test Accuracy: 88.1 % Train Accuracy: 88.5 Parameters: 22m |
| question-answering-on-squad11 | DCN + Char + CoVe | EM: 71.3 F1: 79.9 Hardware Burden: Operations per network pass: |
| question-answering-on-squad11-dev | DCN (Char + CoVe) | EM: 71.3 F1: 79.9 |
| sentiment-analysis-on-imdb | BCN+Char+CoVe | Accuracy: 91.8 |
| sentiment-analysis-on-sst-2-binary | BCN+Char+CoVe | Accuracy: 90.3 |
| sentiment-analysis-on-sst-5-fine-grained | BCN+Char+CoVe | Accuracy: 53.7 |
| text-classification-on-trec-6 | CoVe | Error: 4.2 |
Build AI with AI
From idea to launch — accelerate your AI development with free AI co-coding, out-of-the-box environment and best price of GPUs.