Generalization in LLM Problem Solving: The Case of the Shortest Path
Abstract
Whether language models can systematically generalize remains actively debated. Yet empirical performance is jointly shaped by multiple factors such as training data, training paradigms, and inference-time strategies, making failures difficult to interpret. We introduce a controlled synthetic environment based on shortest-path planning, a canonical composable sequential optimization problem. The setup enables clean separation of these factors and supports two orthogonal axes of generalization: ***spatial transfer*** to unseen maps and ***length scaling*** to longer-horizon problems. We find that models exhibit strong spatial transfer but consistently fail under length scaling due to recursive instability. We further analyze how distinct stages of the learning pipeline influence systematic problem-solving: for example, data coverage sets capability limits; reinforcement learning improves training stability but does not expand those limits; and inference-time scaling enhances performance but cannot rescue length-scaling failures.
Cite
Text
Tong et al. "Generalization in LLM Problem Solving: The Case of the Shortest Path." International Conference on Learning Representations, 2026.Markdown
[Tong et al. "Generalization in LLM Problem Solving: The Case of the Shortest Path." International Conference on Learning Representations, 2026.](https://mlanthology.org/iclr/2026/tong2026iclr-generalization/)BibTeX
@inproceedings{tong2026iclr-generalization,
title = {{Generalization in LLM Problem Solving: The Case of the Shortest Path}},
author = {Tong, Yao and Ye, Jiayuan and Borovykh, Anastasia and Shokri, Reza},
booktitle = {International Conference on Learning Representations},
year = {2026},
url = {https://mlanthology.org/iclr/2026/tong2026iclr-generalization/}
}