Learning by Fixing: Solving Math Word Problems with Weak Supervision

Authors

  • Yining Hong University of California, Los Angeles
  • Qing Li University of California, Los Angeles
  • Daniel Ciao University of California, Los Angeles
  • Siyuan Huang University of California, Los Angeles
  • Song-Chun Zhu University of California, Los Angeles

DOI:

https://doi.org/10.1609/aaai.v35i6.16629

Keywords:

Neuro-Symbolic AI (NSAI), Education, Question Answering, Applications

Abstract

Previous neural solvers of math word problems (MWPs) are learned with full supervision and fail to generate diverse solutions. In this paper, we address this issue by introducing a weakly-supervised paradigm for learning MWPs. Our method only requires the annotations of the final answers and can generate various solutions for a single problem. To boost weakly-supervised learning, we propose a novel learning-by-fixing (LBF) framework, which corrects the misperceptions of the neural network via symbolic reasoning. Specifically, for an incorrect solution tree generated by the neural network, the fixing mechanism propagates the error from the root node to the leaf nodes and infers the most probable fix that can be executed to get the desired answer. To generate more diverse solutions, tree regularization is applied to guide the efficient shrinkage and exploration of the solution space, and a memory buffer is designed to track and save the discovered various fixes for each problem. Experimental results on the Math23K dataset show the proposed LBF framework significantly outperforms reinforcement learning baselines in weakly-supervised learning. Furthermore, it achieves comparable top-1 and much better top-3/5 answer accuracies than fully-supervised methods, demonstrating its strength in producing diverse solutions.

Downloads

Published

2021-05-18

How to Cite

Hong, Y., Li, Q., Ciao, D., Huang, S., & Zhu, S.-C. (2021). Learning by Fixing: Solving Math Word Problems with Weak Supervision. Proceedings of the AAAI Conference on Artificial Intelligence, 35(6), 4959-4967. https://doi.org/10.1609/aaai.v35i6.16629

Issue

Section

AAAI Technical Track Focus Area on Neuro-Symbolic AI