2 papers
cs.LG2026
Learning When to Abort in Transformer Planning: A Mechanistic Study
Shi Feng, Qimin Zhong, Siwei Wang +4
Planning does not by itself imply the ability to recognise errors, but recognising errors can improve planning capacity. In this paper, we study this phenomenon in Transformer plan…
cs.AI2025
Understanding and Enhancing the Planning Capability of Language Models via Multi-Token Prediction
Qimin Zhong, Hao Liao, Siwei Wang +4
Large Language Models (LLMs) have achieved impressive performance across diverse tasks but continue to struggle with learning transitive relations, a cornerstone for complex planni…