Instruction-based Image Editing with Planning, Reasoning, and Generation
Organizations: HKUST
Abstract
Editing images via instruction provides a natural way to generate interactive content, but it is a big challenge due to the higher requirement of scene understanding and generation. Prior work utilizes a chain of large language models, object segmentation models, and editing models for this task. However, the understanding models provide only a single modality ability, restricting the editing quality. We aim to bridge understanding and generation via a new multi-modality model that provides the intelligent abilities to instruction-based image editing models for more complex cases. To achieve this goal, we individually separate the instruction editing task with the multi-modality chain of thought prompts, i.e., Chain-of-Thought (CoT) planning, editing region reasoning, and editing. For Chain-of-Thought planning, the large language model could reason the appropriate sub-prompts considering the instruction provided and the ability of the editing network. For editing region reasoning, we train an instruction-based editing region generation network with a multi-modal large language model. Finally, a hint-guided instruction-based editing network is proposed for editing image generations based on the sizeable text-to-image diffusion model to accept the hints for generation. Extensive experiments demonstrate that our method has competitive editing abilities on complex real-world images.
Figures & tables
| Ablation | Methods | CLIP-I | CLIP-T |
|---|---|---|---|
| Hints method | Pretrained LISA | 0.9081 | 0.2766 |
| Ground Truth | 0.9277 | 0.2838 | |
| Ours | 0.9219 | 0.2835 | |
| The ratio of | 0 | 0.9314 | 0.2771 |
| 0.25 | 0.9290 | 0.2796 | |
| augmented data | 0.5 | 0.9219 | 0.2835 |