Code-driven Number Sequence Calculation: Enhancing the inductive Reasoning Abilities of Large Language Models
Kedi Chen, Zhikai Lei, Xu Guo, Xuecheng Wu, Siyuan Zeng, Jianghao Yin, Yinqi Zhang, Qin Chen, Jie Zhou, Liang He, Qipeng Guo, Kai Chen, Wei Zhang
TL;DR
This work tackles inductive reasoning in large language models by introducing CodeSeq, a synthetic data pipeline that converts number sequences into algorithmic GTG problems solved via code. It combines case-based reflection in supervised fine-tuning with a solvability-aware reinforcement learning reward (CSSR) to teach LLMs to autonomously generate and verify cases. Experiments on open-source backbones show CodeSeq markedly improves GTG and several code-reasoning benchmarks while largely preserving OOD performance, even enabling small models to approach the performance of much larger models on inductive tasks. The approach demonstrates scalable, automated induction data construction and a principled framework for learning from both successes and failures, with potential extensions to broader inductive-reasoning and multimodal tasks.
Abstract
Large language models (LLMs) make remarkable progress in reasoning tasks. Among different reasoning modes, inductive reasoning, due to its better alignment with human learning, attracts increasing interest. However, research on inductive reasoning faces certain challenges. First, existing inductive data mostly focuses on superficial regularities while lacking more complex internal patterns. Second, current works merely prompt LLMs or finetune on simple prompt-response pairs, but do not provide precise thinking processes nor implement difficulty control. Unlike previous work, we address these challenges by introducing \textit{CodeSeq}, a synthetic post-training dataset built from number sequences. We package number sequences into algorithmic problems to discover their general terms, defining a general term generation (GTG) task correspondingly. Our pipeline generates supervised finetuning data by reflecting on failed test cases and incorporating iterative corrections, thereby teaching LLMs to learn autonomous case generation and self-checking. Additionally, it leverages reinforcement learning with a novel Case-Synergy Solvability Scaling Reward based on both solvability, estimated from the problem pass rate, and the success rate of self-directed case generation, enabling models to learn more effectively from both successes and failures. Experimental results show that the models trained with \textit{CodeSeq} improve on various reasoning tasks and can preserve the models' OOD performance.
