cs.CVSep 28, 2026

FlowTool: Controlling Tool Parameter in Image Retouching via Flow Matching

Authors: Thanh-Long V. Le, Steven Walton, Seunghyun Yoon, Branislav Kveton, Trung Bui, Eunho Yang, Viet Lai

Organizations: Adobe Research · KAIST

Abstract

Tool-based image editing (image retouching) is commonly formulated with autoregressive multimodal large language models (MLLMs) that sequentially generate reasoning, tool selections, and parameter values. In this work, we present a novel approach to tool-based image editing by framing the task as a flow matching problem. We introduce FlowTool, a framework that directly models the distribution of high-quality tool parameters conditioned on the input image and user instruction using conditional rectified flow. FlowTool combines a vision-language model backbone for multimodal understanding with a Diffusion Transformer parameter generator that transforms Gaussian noise into an editing plan. We train FlowTool with a two-stage supervised flow-matching curriculum, followed by reward-based post-training. Across MMArt-Bench, FlowTool-Eval, ArtEdit-Bench, and MIT-Adobe5K, FlowTool achieves significantly stronger reference-based performance than specialized MLLM editing agents and proprietary MLLMs, while remaining competitive with proprietary models under reference-free evaluation. Moreover, FlowTool significantly improves inference efficiency, reducing latency by at least 50×50\times while requiring nearly 2×2\times less memory than the compared baselines. These results demonstrate that tool-based image editing can be effectively modeled as conditional generation over structured continuous editing parameters, without autoregressive reasoning.

Explore similar work

CardsList
  1. h-Flow: Flexible Flow-based Image Editing via Doob's h-Transform

    Jul 12, 2026Zehui Guo, Zhen Wang, Junwei Shu +3Image EditingText-To-Image Diffusion Models

  2. IEA: Amateur-Friendly Conversational Image Editing Agent via Three Stages of Multitask Alignment

    Jun 6, 2026Zichen Zhu, Yuheng Sun, Mingxuan Zhu +11Image Editing

  3. MT-EditFlow: Reinforcement Learning for Multi-Turn Image Editing with Flow Matching

    Jun 1, 2026Jiahui Huang, Yasi Zhang, Tianyu Chen +6Image EditingMulti-Reference Image Generation