When Text-to-Image Helps Editing: The Effects of Conditioning During Denoising
Organizations: HSE University & Yandex · Yandex
Abstract
Unified models are trained for both instruction-based image editing and text-to-image (T2I) generation, but standard editing pipelines keep source-image conditioning throughout denoising. We ask whether editing can benefit from T2I, and study how the effects of conditioning vary across edits and denoising stages. In pure editing, source attention declines for some edits over the sampling trajectory. This observation led us to task switching, which lets the model draw on its T2I capabilities. Across three unified editors and four benchmarks, switching to the T2I task for bounded intervals improves edit quality, while mean perceptual preservation remains close to pure editing across all three models. Unified editors therefore benefit from using both conditioning modes they are trained for, and the timing of the switch sets the balance between quality and preservation.
Figures & tables
| Mode / control | Image | Text |
|---|---|---|
| w/ editing prompt | ||
| w/ caption |
| GEdit | ImgEdit | MagicBrush | GIE-Bench | ||||||
| Func. Corr. | Preservation | ||||||||
| Model | Sampling | SC | PQ | O | overall | CLIP-T | Overall | SSIM | GPT-4.1 |
| FLUX2-Klein 4B | std (3 seeds) | 0.05 | 0.03 | 0.04 | 0.04 | 0.001 | 0.35 | 0.001 | 0.05 |
| Pure editing | 7.72 | 7.46 | 7.11 | 3.74 | 0.317 | 73.40 | 0.805 | 7.65 | |
| Fixed | 8.08 | 7.90 | 7.67 | 3.99 | 0.322 | 76.46 | 0.781 | 7.57 | |
| Selective ∗ | 8.01 | 7.67 | 7.45 | 3.88 | 0.319 | 77.11 | 0.786 | 7.60 | |
Appendix figures & tables16 assets
Supplementary material from the paper’s appendix.
Appendix
| Functional correctness | Preservation | |||||||||||||||
| Schedule | Interval | Add Object | Attribute Change | Color Change | Layout Modif. | Object Replace | Remove Object | Scene / Bkgd Change | Size Change | Textual Edit | Overall | SSIM | CLIP | Unmasked CLIP | PSNR | MSE |
| Baseline | — | 91.67 | 88.33 | 90.76 | 44.17 | 93.33 | 68.33 | 95.83 | 20.83 | 67.50 | 73.40 | 0.805 | 0.97 | 0.913 | 20.88 | 1882.32 |
| 91.67 | 93.33 | 95.80 | 43.33 | 93.33 | 71.43 | 96.67 | 27.5 | 61.67 | 74.95 | 0.801 | 0.970 | 0.911 | 20.32 | 2117.10 | ||
| 89.52 | 88.00 | 92.86 | 45.45 | 94.29 | 66.34 | 93.48 | 23.58 | 67.65 | 73.13 | 0.81 | 0.97 | 0.914 | 20.76 | 1897.70 | ||
| 90.83 | 89.17 | 91.60 | 48.33 | 93.28 | 67.50 | 95.83 | 25.00 | 64.17 | 73.93 | 0.81 | 0.97 | 0.913 | 20.84 | 1878.95 | ||
| E ∗ | 90.83 | 93.33 | 94.12 | 41.67 | 95 | 70.59 | 96.67 | 30.83 | 70 | 75.88 | 0.799 | 0.970 | 0.910 | 20.25 | 2174.08 | |
| Schedule | Image | Text | Best SC | Best PQ | Best O |
|---|---|---|---|---|---|
| 7.91 | 7.65 | 7.33 | |||
| 8.50 | 7.69 | 7.80 | |||
| Caption in interval; image retained | 8.14 | 8.18 | 7.58 | ||
| Image off in interval; original instruction | 7.95 | 8.38 | 7.39 | ||
| Caption throughout; image off in interval | 7.58 | 8.36 | 7.53 | ||
| 8.07 | 8.36 | 7.69 |
| Schedule | Interval | background | adjust | style | extract | remove | add | replace | compose | action | overall |
|---|---|---|---|---|---|---|---|---|---|---|---|
| FLUX2-Klein 4B | |||||||||||
| Baseline | — | 4.21 | 3.85 | 4.80 | 1.47 | 2.99 | 4.40 | 4.18 | 3.26 | 4.70 | 3.74 |
| 4.11 | 3.63 | 4.80 | 3.74 | 4.33 | 4.49 | 4.16 | 3.68 | 4.73 | 4.18 | ||
| 4.08 | 3.83 | 4.74 | 1.38 | 2.98 | 4.36 | 4.14 | 3.19 | 4.91 | 3.69 | ||
| 4.15 | 3.81 | 4.80 | 1.55 | 3.01 | 4.40 | 4.20 | 3.26 | 4.80 | 3.73 | ||
| E ∗ | 4.08 | 3.73 | 4.75 | 3.64 | 4.28 | 4.42 | 4.19 | 3.55 | 4.73 | 4.15 | |
| GEdit | ImgEdit | MagicBrush | GIE-Bench | |||||
| Sampling | SC | PQ | O | overall | CLIP-T | FC | SSIM | U-CLIP |
| FLUX2-Klein 4B | ||||||||
| std (3 seeds) | 0.05 | 0.03 | 0.04 | 0.04 | 0.001 | 0.35 | 0.001 | 0.002 |
| Pure E | 7.72 | 7.46 | 7.11 | 3.74 | 0.317 | 73.40 | 0.805 | 0.913 |
| E E | 8.16 | 7.49 | 7.54 | 4.18 | 0.317 | 74.95 | 0.801 | 0.911 |
| E ∗ | 8.17 | 7.49 | 7.54 | 4.15 | 0.317 | 75.88 | 0.799 | 0.910 |
| Conditioning | Forwards / step | Required predictions |
|---|---|---|
| Pure E (baseline) | 2 | |
| Pure T2I | 2 | |
| Image guidance ( ) | 3 | |
| Multimodal guidance ( ) | 4 |
| GEdit | ImgEdit | MagicBrush | GIE-Bench | ||||||
| Fwd/ step | SC | PQ | O | overall | CLIP-T | FC | SSIM | U-CLIP | |
| Image guidance | |||||||||
| Interval | |||||||||
| 3 | 7.69 | 7.27 | 7.10 | 3.66 | 0.319 | 74.84 | 0.782 | 0.891 | |
| 3 | 7.74 | 7.36 | 7.15 | 3.71 | 0.319 | 74.98 | 0.790 | 0.898 | |
| 3 | 7.70 | 7.43 | 7.15 | 3.73 | 0.319 | 73.35 | 0.797 | 0.904 | |
| Sampling | Caption model | Image generation | VLM call | Total |
|---|---|---|---|---|
| Pure editing | — | 5.325 | — | 5.475 |
| Gemini-2.5-pro | 4.900 | 26.760 | 31.810 | |
| Qwen3.5-9B | 4.900 | 0.460 | 5.510 | |
| Qwen3.5-27B | 4.900 | 1.143 | 6.193 | |
| Qwen3.5-35B-A3B | 4.900 | 0.239 | 5.289 |
| GEdit | ImgEdit | MagicBrush | GIE-Bench | ||
| Caption model | O | Overall | CLIP-T | FC (%) | SSIM |
| std (3 seeds) | 0.04 | 0.04 | 0.001 | 0.35 | 0.001 |
| Pure editing (no caption) | 7.11 | 3.74 | 0.317 | 73.40 | 0.805 |
| Gemini-2.5-pro | 7.53 | 4.16 | 0.324 | 76.74 | 0.684 |
| Qwen3.5-27B | 7.51 | 4.10 | 0.324 | 77.02 | 0.684 |
| Qwen3.5-35B-A3B | 7.48 | 4.11 | 0.323 | 77.37 | 0.683 |
| Model | Routing | Preservation | Fidelity | PQ | O |
|---|---|---|---|---|---|
| FLUX2-Klein 4B | Pure editing | 8.902 | 8.395 | 7.421 | 7.112 |
| Attention rule ( switched) | 8.860 | 8.696 | 7.698 | 7.459 | |
| Random, preservation-matched ( ) | 8.860 | 8.546 | 7.578 | 7.295 | |
| Difference | |||||
| FLUX2-Klein 9B | Pure editing | 9.127 | 9.033 | 7.746 | 7.778 |
| Attention rule ( switched) | 9.085 | 9.201 | 7.869 | 7.949 |
| GPT preservation | SSIM | |||||
| Comparison | 95% CI | 95% CI | ||||
| FLUX2-Klein 4B: comparisons against pure editing | ||||||
| , | ||||||
| , | ||||||
| , | ||||||
| , | ||||||