ReFlex: Text-Guided Editing of Real Images in Rectified Flow via Mid-Step Feature Extraction and Attention Adaptation
Jimyeong Kim, Jungwon Park, Yeji Song, Nojun Kwak, Wonjong Rhee
Abstract
Rectified Flow text-to-image models surpass diffusion models in image quality and text alignment, but adapting Re-Flow for real-image editing remains challenging. We propose a new real-image editing method for ReFlow by analyzing the intermediate representations of multimodal transformer blocks and identifying three key features. To extract these features from real images with sufficient structural preservation, we leverage mid-step latent, which is inverted only up to the mid-step. We then adapt attention during injection to improve editability and enhance alignment to the target text. Our method is training-free, requires no user-provided mask, and can be applied even without a source prompt. Extensive experiments on two benchmarks with nine baselines demonstrate its superior performance over prior methods, further validated by human evaluations confirming a strong user preference for our approach.
Ask about this paper
Your agent reads all of it.
Lune indexed this paper to the last equation, along with the top-tier papers that cite it. Ask a question and the answer quotes them.
Your agent calls
Luneget_paper_fulltext
Free to start. No credit card required.
Terminal
Install the CLIlune papers fulltext c513c443-0210-4738-bdcb-cd2cf02acc0bCited by top-tier papers4
- Diagnosing and Correcting Concept Omission in Multimodal Diffusion TransformersKanghyun Baek, Jaihyun Lew, Chaehun Shin, Jungbeom Lee et al.ICML 2026 · 1 citation
- Language-Free Generative Editing from One Visual ExampleOmar Elezabi, Eduard Zamfir, Zongwei Wu, Radu TimofteCVPR 2026 · 1 citation
- Breaking the Lock-in: Diversifying Text-to-Image Generation via Representation ModulationDahee Kwon, Haeun Lee, Jaesik ChoiICML 2026
- Deconstructing Guidance: A Semantic Hierarchy for Precise Diffusion Model EditingWootaek Jeong, Junghyo Sohn, Jee Seok Yoon, Heung-Il SukICLR 2026
Builds on30
- Learning Transferable Visual Models From Natural Language SupervisionAlec Radford, Jong Wook Kim, Chris Hallacy, Aditya Ramesh et al.ICML 2021 · 47,906 citations
- High-Resolution Image Synthesis with Latent Diffusion ModelsRobin Rombach, Andreas Blattmann, Dominik Lorenz, Patrick Esser et al.CVPR 2022 · 13,123 citations
- Directly Denoising Diffusion ModelsDan Zhang, Jingjing Wang, Feng LuoICML 2024 · 11,724 citations
- Photorealistic Text-to-Image Diffusion Models with Deep Language UnderstandingChitwan Saharia, William Chan, Saurabh Saxena, Lala Li et al.NeurIPS 2022 · 8,965 citations
- Emerging Properties in Self-Supervised Vision TransformersMathilde Caron, Hugo Touvron, Ishan Misra, Hervé Jégou et al.ICCV 2021 · 8,921 citations
Related papers
- DirectEdit: Step-Level Accurate Inversion for Flow-Based Image EditingDesong Yang, Mang YeICML 2026
- W-EDIT: A Wavelet-Based Frequency-Aware Framework for Text-Driven Image EditingJiahui Sun, Weining Wang, Mingzhen Sun, Peiyao Wang et al.ICLR 2026
- Flowedit: Inversion-Free Text-Based Editing Using Pre-Trained Flow ModelsVladimir Kulikov, Matan Kleiner, Inbar Huberman-Spiegelglas, Tomer MichaeliICCV 2025 · 30 citations
- Training-Free Text-Guided Image Editing with Visual Autoregressive ModelYufei Wang, Lanqing Guo, Zhihao Li, Jiaxing Huang et al.ICCV 2025
- DNAEdit: Direct Noise Alignment for Text-Guided Rectified Flow EditingChenxi Xie, Minghan Li, Shuai Li, Yuhui Wu et al.NeurIPS 2025 · 27 citations
