CoSTA*: Cost-Sensitive Toolpath Agent for Multi-turn Image Editing

Advait Gupta, NandaKiran Velaga, Dang Nguyen, Tianyi Zhou


Abstract
Text-to-image models like Stable Diffusion and DALLE-3 still struggle with complex multi-turn image editing. We study how to break down such a task into a sequence of subtasks and address them by an agentic workflow (path) of AI tool use with minimum costs. Conventional search algorithms require expensive exploration to find tool paths. While large language models (LLMs) possess prior knowledge of subtask planning, their estimation of the quality and cost of tools is usually inaccurate to determine which to apply in each subtask. Can we combine the strengths of both LLMs and graph search to find cost-efficient tool paths? We propose a three-stage approach “CoSTA*” that leverages LLMs to create a subtask tree that prunes a graph of AI tools for the given task, and then conducts A* search on the small subgraph to find a tool path. To better balance the total cost and quality, CoSTA* combines both metrics of each tool on every subtask to guide the A* search. Each subtask’s output is evaluated by a vision-language model (VLM), where a failure will trigger an update of the tool’s cost and quality on that subtask. Hence, the A* search can recover from failures quickly to explore other paths. Moreover, CoSTA* can automatically switch between modalities across subtasks for a better cost-quality trade-off. We build a novel benchmark of challenging multi-turn image editing, on which CoSTA* outperforms state-of-the-art image-editing models or agents in both cost and quality, and performs versatile trade-offs upon user preference.
Anthology ID:
2026.findings-emnlp.1356
Volume:
Findings of the Association for Computational Linguistics: EMNLP 2026
Month:
October
Year:
2026
Address:
Budapest, Hungary
Editors:
Hung-Yi Lee, Sunipa Dev, Jessy Li, Desmond Elliott
Venue:
Findings
SIG:
SIGDAT
Publisher:
Association for Computational Linguistics
Note:
Pages:
28779–28814
Language:
URL:
https://aclanthology.org/2026.findings-emnlp.1356/
DOI:
Bibkey:
Cite (ACL):
Advait Gupta, NandaKiran Velaga, Dang Nguyen, and Tianyi Zhou. 2026. CoSTA*: Cost-Sensitive Toolpath Agent for Multi-turn Image Editing. In Findings of the Association for Computational Linguistics: EMNLP 2026, pages 28779–28814, Budapest, Hungary. Association for Computational Linguistics.
Cite (Informal):
CoSTA*: Cost-Sensitive Toolpath Agent for Multi-turn Image Editing (Gupta et al., Findings 2026)
Copy Citation:
PDF:
https://aclanthology.org/2026.findings-emnlp.1356.pdf