arXiv Computer Vision
5d ago

Attention Maps Know When to Think: Training-Free Difficulty-Aware Chain-of-Thought for Reasoning Segmentation

The paper introduces Difficulty-Aware Chain-of-Thought (DA‑CoT), a training‑free framework that selectively applies chain‑of‑thought prompting in reasoning segmentation tasks based on the entropy of attention maps. Low entropy signals confident, localized grounding, while high entropy indicates uncertainty and the potential benefit of additional reasoning. DA‑CoT also incorporates a Bidirectional Rethinking mechanism to automatically reverse reasoning decisions when initial outputs fail, thereby improving robustness and inference efficiency while achieving state‑of‑the‑art results on Reasoning Segmentation benchmarks.

By Chanseul Cho, Jaesung Jun, Dongheon Lee, Youngmin Ro