Math Word Problem Generation via Disentangled Memory Retrieval

doi:10.1145/3639569

Math Word Problem Generation via Disentangled Memory Retrieval

2024-03-26

1

OA

AI

DOI：10.1145/3639569

Original request for help

Save

Abstract

En 中文

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

retrieval

math word problem

text generation

Math Word Problem Generation via Disentangled Memory Retrieval

Abstract

Journal

Organization