Math Word Problem Generation via Disentangled Memory Retrieval

Wei Qin,Xiaowei Wang,Zhenzhen Hu,Lei Wang,Yunshi Lan,Richang Hong
DOI: https://doi.org/10.1145/3639569
IF: 4.157
2024-01-26
ACM Transactions on Knowledge Discovery from Data
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.
computer science, information systems, software engineering
What problem does this paper attempt to address?