🤖 AI Summary
Large language models often struggle with code generation due to limited reasoning and planning capabilities, particularly when integrating fragmented problem constraints. This work proposes a narrative restructuring framework that, for the first time, incorporates human-like information organization into code generation. By guiding models through structured natural language narratives—including task overviews, explicit constraints, and example test cases—the approach enhances contextual understanding, improves algorithm selection accuracy, reduces implementation errors, and encourages modular code synthesis. Combining narrative restructuring, structured prompting, and reasoning guidance, the method achieves an average 18.7% improvement in zero-shot pass@10 across eleven mainstream models and demonstrates consistent effectiveness on multiple benchmarks, including HumanEval, LiveCodeBench, and CodeForces.
📝 Abstract
Effective code generation requires both model capability and a problem representation that carefully structures how models reason and plan. Existing approaches augment reasoning steps or inject specific structure into how models think, but leave scattered problem conditions unchanged. Inspired by the way humans organize fragmented information into coherent explanations, we propose StoryCoder, a narrative reformulation framework that transforms code generation questions into coherent natural language narratives, providing richer contextual structure than simple rephrasings. Each narrative consists of three components: a task overview, constraints, and example test cases, guided by the selected algorithm and genre. Experiments across 11 models on HumanEval, LiveCodeBench, and CodeForces demonstrate consistent improvements, with an average gain of 18.7% in zero-shot pass@10. Beyond accuracy, our analyses reveal that narrative reformulation guides models toward correct algorithmic strategies, reduces implementation errors, and induces a more modular code structure. The analyses further show that these benefits depend on narrative coherence and genre alignment, suggesting that structured problem representation is important for code generation regardless of model scale or architecture. Our code is available at https://github.com/gu-ni/StoryCoder.