CoT-Seg: Rethinking Segmentation with Chain-of-Thought Reasoning and Self-Correction

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Kao, Shiu-hong, Huang, Chak Ho, Liu, Huaiqian, Tai, Yu-Wing, Tang, Chi-Keung
Formato: Preprint
Publicado: 2026
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866909999920316416
author Kao, Shiu-hong
Huang, Chak Ho
Liu, Huaiqian
Tai, Yu-Wing
Tang, Chi-Keung
author_facet Kao, Shiu-hong
Huang, Chak Ho
Liu, Huaiqian
Tai, Yu-Wing
Tang, Chi-Keung
contents Existing works of reasoning segmentation often fall short in complex cases, particularly when addressing complicated queries and out-of-domain images. Inspired by the chain-of-thought reasoning, where harder problems require longer thinking steps/time, this paper aims to explore a system that can think step-by-step, look up information if needed, generate results, self-evaluate its own results, and refine the results, in the same way humans approach harder questions. We introduce CoT-Seg, a training-free framework that rethinks reasoning segmentation by combining chain-of-thought reasoning with self-correction. Instead of fine-tuning, CoT-Seg leverages the inherent reasoning ability of pre-trained MLLMs (GPT-4o) to decompose queries into meta-instructions, extract fine-grained semantics from images, and identify target objects even under implicit or complex prompts. Moreover, CoT-Seg incorporates a self-correction stage: the model evaluates its own segmentation against the original query and reasoning trace, identifies mismatches, and iteratively refines the mask. This tight integration of reasoning and correction significantly improves reliability and robustness, especially in ambiguous or error-prone cases. Furthermore, our CoT-Seg framework allows easy incorporation of retrieval-augmented reasoning, enabling the system to access external knowledge when the input lacks sufficient information. To showcase CoT-Seg's ability to handle very challenging cases ,we introduce a new dataset ReasonSeg-Hard. Our results highlight that combining chain-of-thought reasoning, self-correction, offers a powerful paradigm for vision-language integration driven segmentation.
format Preprint
id arxiv_https___arxiv_org_abs_2601_17420
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle CoT-Seg: Rethinking Segmentation with Chain-of-Thought Reasoning and Self-Correction
Kao, Shiu-hong
Huang, Chak Ho
Liu, Huaiqian
Tai, Yu-Wing
Tang, Chi-Keung
Computer Vision and Pattern Recognition
Existing works of reasoning segmentation often fall short in complex cases, particularly when addressing complicated queries and out-of-domain images. Inspired by the chain-of-thought reasoning, where harder problems require longer thinking steps/time, this paper aims to explore a system that can think step-by-step, look up information if needed, generate results, self-evaluate its own results, and refine the results, in the same way humans approach harder questions. We introduce CoT-Seg, a training-free framework that rethinks reasoning segmentation by combining chain-of-thought reasoning with self-correction. Instead of fine-tuning, CoT-Seg leverages the inherent reasoning ability of pre-trained MLLMs (GPT-4o) to decompose queries into meta-instructions, extract fine-grained semantics from images, and identify target objects even under implicit or complex prompts. Moreover, CoT-Seg incorporates a self-correction stage: the model evaluates its own segmentation against the original query and reasoning trace, identifies mismatches, and iteratively refines the mask. This tight integration of reasoning and correction significantly improves reliability and robustness, especially in ambiguous or error-prone cases. Furthermore, our CoT-Seg framework allows easy incorporation of retrieval-augmented reasoning, enabling the system to access external knowledge when the input lacks sufficient information. To showcase CoT-Seg's ability to handle very challenging cases ,we introduce a new dataset ReasonSeg-Hard. Our results highlight that combining chain-of-thought reasoning, self-correction, offers a powerful paradigm for vision-language integration driven segmentation.
title CoT-Seg: Rethinking Segmentation with Chain-of-Thought Reasoning and Self-Correction
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2601.17420