AtomThink: A Slow Thinking Framework for Multimodal Mathematical Reasoning
In this paper, we address the challenging task of multimodal mathematical reasoning by incorporating the ability of ``slow thinking" into multimodal large language models (MLLMs). Contrary to existing methods that rely on direct or fast thinking, our key idea is to construct long chains of thou...
Gespeichert in:
Hauptverfasser: | , , , , , , , , , , , , |
---|---|
Format: | Artikel |
Sprache: | eng |
Schlagworte: | |
Online-Zugang: | Volltext bestellen |
Tags: |
Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
|
Zusammenfassung: | In this paper, we address the challenging task of multimodal mathematical
reasoning by incorporating the ability of ``slow thinking" into multimodal
large language models (MLLMs). Contrary to existing methods that rely on direct
or fast thinking, our key idea is to construct long chains of thought (CoT)
consisting of atomic actions in a step-by-step manner, guiding MLLMs to perform
complex reasoning. To this end, we design a novel AtomThink framework composed
of three key modules: (i) a CoT annotation engine that automatically generates
high-quality CoT annotations to address the lack of high-quality visual
mathematical data; (ii) an atomic step fine-tuning strategy that jointly
optimizes an MLLM and a policy reward model (PRM) for step-wise reasoning; and
(iii) four different search strategies that can be applied with the PRM to
complete reasoning. Additionally, we propose AtomMATH, a large-scale multimodal
dataset of long CoTs, and an atomic capability evaluation metric for
mathematical tasks. Extensive experimental results show that the proposed
AtomThink significantly improves the performance of baseline MLLMs, achieving
approximately 50\% relative accuracy gains on MathVista and 120\% on MathVerse.
To support the advancement of multimodal slow-thinking models, we will make our
code and dataset publicly available on https://github.com/Quinn777/AtomThink. |
---|---|
DOI: | 10.48550/arxiv.2411.11930 |