Seeking Patterns, Not just Memorizing Procedures: Contrastive Learning for Solving Math Word Problems

By Zhongli Li, Wenxuan Zhang, Chao Yan, Qingyu Zhou, Chao Li, Hongzhi Liu, Yunbo Cao. 2021

62 citations recorded by the source corpus.

Math Word Problem (MWP) solving needs to discover the quantitative relationships over natural language narratives. Recent work shows that existing models memorize procedures from context and rely on shallow heuristics to solve MWPs. In this paper, we look at this issue and argue that the cause is a lack of overall understanding of MWP patterns. We first investigate how a neural network understands patterns only from semantics, and observe that, if the prototype equations are the same, most problems get closer representations and those representations apart from them or close to other prototypes tend to produce wrong solutions. Inspired by it, we propose a contrastive learning approach, where the neural network perceives the divergence of patterns. We collect contrastive examples by converting the prototype equation into a tree and seeking similar tree structures. The solving model is trained with an auxiliary objective on the collected examples, resulting in the representations of problems with similar prototypes being pulled closer. We conduct experiments on the Chinese dataset Math23k and the English dataset MathQA. Our method greatly improves the performance in monolingual and multilingual settings.

Subfields: Artificial Intelligence, Computer Vision and Pattern Recognition, Natural Language Processing, Representation & Self-Supervised Learning

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. Wei Dang Lu
  2. Jierui Li
  3. Zhanming Jie
  4. Jing Jiang
  5. Lei Wang
  6. Lingxiao Jiang
  7. Minghuan Tan
  8. Shuohang Wang

In Computer Vision and Pattern Recognition

  1. Wei Dang Lu
  2. Jing Jiang
  3. Lei Wang
  4. Lingxiao Jiang
  5. Minghuan Tan
  6. Zhanming Jie
  7. Jierui Li
  8. Shuohang Wang

In Natural Language Processing

  1. Wei Dang Lu
  2. Jierui Li
  3. Zhanming Jie
  4. Jing Jiang
  5. Lei Wang
  6. Lingxiao Jiang
  7. Minghuan Tan
  8. Shuohang Wang

In Representation & Self-Supervised Learning

  1. Wei Dang Lu
  2. Jing Jiang
  3. Mehryar Mohri
  4. Qiang Yang
  5. Sinno Jialin Pan
  6. Bryan Perozzi

Citations in the corpus