DéjàQ: Open-Ended Evolution of Diverse, Learnable and Verifiable Problems
Willem Röpke, Samuel Coward, Andrei Lupu, Thomas Foster, Tim Rocktäschel, Jakob Foerster
TL;DR
DéjàQ tackles the problem of static data limitations in reasoning-model training by evolving a diverse, verifiable, and learnable set of synthetic math problems in tandem with model optimization. It combines a MAP-Elites quality-diversity archive with RLVR-based post-training and introduces two LLM-guided mutation strategies to reshape problems, ensuring verifiability and ongoing learning signal. Empirical results show that learnability-driven evolution, especially when using the full set of mutations, yields superior in- and out-of-distribution performance and improved robustness to hard instances, while maintaining reasonable resource usage. The work highlights the potential of continually adapting training data to the model's current capabilities and points to broad applicability beyond mathematics, with plans to open-source code and datasets.
Abstract
Recent advances in reasoning models have yielded impressive results in mathematics and coding. However, most approaches rely on static datasets, which have been suggested to encourage memorisation and limit generalisation. We introduce DéjàQ, a framework that departs from this paradigm by jointly evolving a diverse set of synthetic mathematical problems alongside model training. This evolutionary process adapts to the model's ability throughout training, optimising problems for learnability. We propose two LLM-driven mutation strategies in which the model itself mutates the training data, either by altering contextual details or by directly modifying problem structure. We find that the model can generate novel and meaningful problems, and that these LLM-driven mutations improve RL training. We analyse key aspects of DéjàQ, including the validity of generated problems and computational overhead. Our results underscore the potential of dynamically evolving training data to enhance mathematical reasoning and indicate broader applicability, which we will support by open-sourcing our code.
