Command Palette
Search for a command to run...
Supervised Learning of Universal Sentence Representations from Natural Language Inference Data
Alexis Conneau; Douwe Kiela; Holger Schwenk; Loic Barrault; Antoine Bordes

Abstract
Many modern NLP systems rely on word embeddings, previously trained in an unsupervised manner on large corpora, as base features. Efforts to obtain embeddings for larger chunks of text, such as sentences, have however not been so successful. Several attempts at learning unsupervised representations of sentences have not reached satisfactory enough performance to be widely adopted. In this paper, we show how universal sentence representations trained using the supervised data of the Stanford Natural Language Inference datasets can consistently outperform unsupervised methods like SkipThought vectors on a wide range of transfer tasks. Much like how computer vision uses ImageNet to obtain features, which can then be transferred to other tasks, our work tends to indicate the suitability of natural language inference for transfer learning to other NLP tasks. Our encoder is publicly available.
Code Repositories
Benchmarks
| Benchmark | Methodology | Metrics |
|---|---|---|
| cross-lingual-natural-language-inference-on | X-BiLSTM | Accuracy: 67.7% |
| cross-lingual-natural-language-inference-on | X-CBOW | Accuracy: 60.3% |
| cross-lingual-natural-language-inference-on-1 | X-CBOW | Accuracy: 60.7% |
| cross-lingual-natural-language-inference-on-1 | X-BiLSTM | Accuracy: 68.7% |
| cross-lingual-natural-language-inference-on-3 | X-BiLSTM | Accuracy: 67.7% |
| cross-lingual-natural-language-inference-on-3 | X-CBOW | Accuracy: 61.0% |
| natural-language-inference-on-snli | 4096D BiLSTM with max-pooling | % Test Accuracy: 84.5 % Train Accuracy: 85.6 Parameters: 40m |
| semantic-textual-similarity-on-mrpc | InferSent | Accuracy: 76.2% F1: 83.1% |
| semantic-textual-similarity-on-senteval | InferSent | MRPC: 76.2/83.1 SICK-E: 86.3 SICK-R: 0.884 STS: 75.8/75.5 |
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.