Analysis-by-Proxy: Localization Signals in VLMs Operating as Condition Encoders
Summary
The "Analysis-by-Proxy" framework investigates why Vision-Language Models (VLMs) used as single-pass condition encoders in diffusion-based image editing pipelines often fail to localize edits accurately, despite standalone VLMs demonstrating strong localization capabilities. The study, focusing on Qwen-Image-Edit with its Qwen2.5-VL-7B backbone, found a 31.5% performance gap: standalone VLMs achieved 89.0% localization accuracy, while the full pipeline managed only 57.5%. This discrepancy arises because crucial spatial signals are not reliably propagated to the final layers typically used for conditioning. Instead, these signals reside in intermediate VLM representations, at locations that dynamically shift based on the input prompt. The framework uses a lightweight Q-Former proxy to uncover these hidden signals, demonstrating that recovering and integrating them significantly improves edit localization.
Key takeaway
For Machine Learning Engineers developing diffusion-based image editing pipelines, if you are struggling with localization accuracy in complex, multi-entity scenes, you should re-evaluate your VLM feature extraction strategy. Instead of relying on final-layer representations, dynamically identify and extract spatial signals from intermediate VLM layers. This targeted approach, potentially using a proxy model to generate explicit bounding box cues, can significantly improve edit precision by providing more accurate spatial conditioning to the Diffusion Transformer.
Key insights
VLMs as condition encoders lose localization accuracy because spatial signals are hidden in intermediate layers, not exposed by standard extraction.
Principles
- VLM spatial knowledge is optimized for autoregressive generation.
- Final VLM layers over-abstract crucial spatial cues.
- Localization signals peak in input-dependent intermediate layers.
Method
Analysis-by-Proxy trains a lightweight Q-Former proxy on VLM intermediate representations for an auxiliary localization task, using GIoU and L1 losses to predict bounding boxes.
In practice
- Use intermediate VLM layers for spatial conditioning.
- Dynamically select VLM layers based on input context.
- Overlay predicted bounding boxes for explicit conditioning.
Topics
- Vision-Language Models
- Image Editing
- Diffusion Models
- Localization
- Mechanistic Interpretability
- Condition Encoders
Best for: Research Scientist, AI Scientist, Machine Learning Engineer, Computer Vision Engineer
Related on AIssential
Editorial summary, takeaway, and curation by AIssential. Original article published by cs.CV updates on arXiv.org.