Do NLP Models Know Numbers? Probing Numeracy in Embeddings

By Eric Wallace, Yizhong Wang, Sujian Li, Sameer Singh, Matt Gardner. 2019

311 citations recorded by the source corpus.

The ability to understand and work with numbers (numeracy) is critical for many complex reasoning tasks. Currently, most NLP models treat numbers in text in the same way as other tokens—they embed them as distributed vectors. Is this enough to capture numeracy? We begin by investigating the numerical reasoning capabilities of a state-of-the-art question answering model on the DROP dataset. We find this model excels on questions that require numerical reasoning, i.e., it already captures numeracy. To understand how this capability emerges, we probe token embedding methods (e.g., BERT, GloVe) on synthetic list maximum, number decoding, and addition tasks. A surprising degree of numeracy is naturally present in standard embeddings. For example, GloVe and word2vec accurately encode magnitude for numbers up to 1,000. Furthermore, character-level embeddings are even more precise—ELMo captures numeracy the best for all pre-trained methods—but BERT, which uses sub-word units, is less exact.

Subfields: Artificial Intelligence, Large Language Models, Evaluation & Benchmarks

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. Mohit Bansal
  2. Graham Neubig
  3. Luke S. Zettlemoyer
  4. Yoav Goldberg
  5. Hannaneh Hajishirzi
  6. Percy Liang
  7. Yejin Choi
  8. Yonatan Belinkov

In Large Language Models

  1. Yoav Goldberg
  2. Hinrich Schütze
  3. Zhiyuan Liu
  4. Jonathan Berant
  5. Anna Rogers
  6. Graham Neubig
  7. Olga Kovaleva
  8. Anna Rumshisky

In Evaluation & Benchmarks

  1. Jonathan Berant
  2. Matt Gardner
  3. Xiang Ren
  4. Anna Rogers
  5. Alon Talmor
  6. Pengfei Liu
  7. Sameer Singh
  8. Chitta Baral

Citations in the corpus