Publication Type

Journal Article

Version

acceptedVersion

Publication Date

3-2024

Abstract

The task of math word problem (MWP) generation, which generates an MWP given an equation and relevant topic words, has increasingly attracted researchers’ attention. In this work, we introduce a simple memory retrieval module to search related training MWPs, which are used to augment the generation. To retrieve more relevant training data, we also propose a disentangled memory retrieval module based on the simple memory retrieval module. To this end, we first disentangle the training MWPs into logical description and scenario description and then record them in respective memory modules. Later, we use the given equation and topic words as queries to retrieve relevant logical descriptions and scenario descriptions from the corresponding memory modules, respectively. The retrieved results are then used to complement the process of the MWP generation. Extensive experiments and ablation studies verify the superior performance of our method and the effectiveness of each proposed module. The code is available at https://github.com/mwp-g/MWPG-DMR.

Keywords

Memory, math word problem, retrieval, text generation

Discipline

Artificial Intelligence and Robotics | Numerical Analysis and Scientific Computing

Publication

ACM Transactions on Knowledge Discovery from Data

Volume

18

Issue

5

Last Page

1

ISSN

1556-4681

Identifier

10.1145/3639569

Publisher

Association for Computing Machinery (ACM)

Copyright Owner and License

Authors

Additional URL

https://doi.org/10.1145/3639569

Download

Share

COinS