Visualization-of-Thought Elicits Spatial Reasoning in Large Language Models

• Published in: arXiv.org
• Main Authors: Wu, Wenshan, Mao, Shaoguang, Zhang, Yadong, Xia, Yan, Li, Dong, Cui, Lei, Furu Wei
• Format: Paper
• Language: English
• Published: Ithaca Cornell University Library, arXiv.org 04.04.2024
• Subjects: Cognition; Cognition & reasoning; Cognitive tasks; Language; Large language models; Natural language processing; Navigation; Reasoning; Tiling; Visualization
• ISSN: 2331-8422

Large language models (LLMs) have exhibited impressive performance in language comprehension and various reasoning tasks. However, their abilities in spatial reasoning, a crucial aspect of human cognition, remain relatively unexplored. Human possess a remarkable ability to create mental images of unseen objects and actions through a process known as \textbf{the Mind's Eye}, enabling the imagination of the unseen world. Inspired by this cognitive capacity, we propose Visualization-of-Thought (\textbf{VoT}) prompting. VoT aims to elicit spatial reasoning of LLMs by visualizing their reasoning traces, thereby guiding subsequent reasoning steps. We employed VoT for multi-hop spatial reasoning tasks, including natural language navigation, visual navigation, and visual tiling in 2D grid worlds. Experimental results demonstrated that VoT significantly enhances the spatial reasoning abilities of LLMs. Notably, VoT outperformed existing multimodal large language models (MLLMs) in these tasks. While VoT works surprisingly well on LLMs, the ability to generate \textit{mental images} to facilitate spatial reasoning resembles the mind's eye process, suggesting its potential viability in MLLMs.