Insert Anything: Image Insertion via In-Context Editing in DiT
Wensong Song, Hong Jiang, Zongxing Yang, Ruijie Quan, Yi Yang
TL;DR
Insert Anything advances reference-based image editing by unifying mask- and text-guided insertion within a single DiT-based framework. The AnyInsertion dataset enables broad task coverage (person, object, garment) and supports two prompting modalities through polyptych in-context editing (diptych for masks, triptych for text). The approach achieves state-of-the-art results across multiple benchmarks (AnyInsertion, DreamBooth, VTON-HD) and offers practical benefits for creative content, virtual try-on, and scene composition. Ablation studies confirm the value of in-context editing, semantic guidance, and dataset scale for preserving high-frequency details and semantic fidelity.
Abstract
This work presents Insert Anything, a unified framework for reference-based image insertion that seamlessly integrates objects from reference images into target scenes under flexible, user-specified control guidance. Instead of training separate models for individual tasks, our approach is trained once on our new AnyInsertion dataset--comprising 120K prompt-image pairs covering diverse tasks such as person, object, and garment insertion--and effortlessly generalizes to a wide range of insertion scenarios. Such a challenging setting requires capturing both identity features and fine-grained details, while allowing versatile local adaptations in style, color, and texture. To this end, we propose to leverage the multimodal attention of the Diffusion Transformer (DiT) to support both mask- and text-guided editing. Furthermore, we introduce an in-context editing mechanism that treats the reference image as contextual information, employing two prompting strategies to harmonize the inserted elements with the target scene while faithfully preserving their distinctive features. Extensive experiments on AnyInsertion, DreamBooth, and VTON-HD benchmarks demonstrate that our method consistently outperforms existing alternatives, underscoring its great potential in real-world applications such as creative content generation, virtual try-on, and scene composition.
