PixARG: Pixel-Asynchronous Retrieval-Augmented Diffusion for Fast and Realistic Image Generation
cs.CV
Submitted: 2026-10-10
Updated: 2026-10-10
Comments: Accepted to ACM Multimedia 2026 (MM '26) as a poster presentation
Code: https://github.com/kakaobrain/coyo-dataset
License: http://creativecommons.org/licenses/by/4.0/
The gist: While diffusion models achieve impressive image generation capabilities, they suffer from high computational costs, implausible artifacts, and limited adaptability to dynamic or private data due to
Abstract
While diffusion models achieve impressive image generation capabilities, they suffer from high computational costs, implausible artifacts, and limited adaptability to dynamic or private data due to static, centralized training. To address these, we propose PixARG, a novel retrieval-augmented generative framework that integrates dynamic real-world image retrieval into the diffusion pipeline to simultaneously enhance efficiency and visual realism. Specifically, PixARG constructs a pixel-level affinity map between the user's text prompt and the retrieved image to guide a heterogeneous noise initialization. By modulating noise intensity across spatial regions, this mechanism preserves semantically relevant structural priors while perturbing less relevant areas, thereby significantly accelerating convergence. To accommodate this spatially non-uniform noise and ensure synchronized generation, we introduce a pixel-wise asynchronous denoising scheduler and a spatiotemporal denoiser conditioned on the affinity map. Furthermore, we derive an affinity-adaptive training objective that elegantly balances deterministic restoration with stochastic regeneration across pixels. Experiments on LAION and COYO datasets across offline, online, and user-specific settings show PixARG enhances realism and semantic alignment while drastically reducing diffusion steps.
Sources
- Imagen 3
- PixArt-$\alpha$: Fast Training of Diffusion Transformer for Photorealistic Text-to-Image Synthesis
- Re-Imagen: Retrieval-Augmented Text-to-Image Generator
- RedCaps: web-curated image-text data created by the people, for the people
- Interpreting CLIP's Image Representation via Text-Based Decomposition
- Retrieval-Augmented Generation for Large Language Models: A Survey
- DeepRAG: Thinking to Retrieve Step by Step for Large Language Models
- ILLUME+: Illuminating Unified MLLM with Dual Visual Tokenization and Diffusion Refinement
- Less is More: Making Smaller Language Models Competent Subgraph Retrievers for Multi-hop KGQA
- TOBUGraph: Knowledge Graph-Based Retrieval for Enhanced LLM Performance Beyond RAG
- SVDQuant: Absorbing Outliers by Low-Rank Components for 4-Bit Diffusion Models
- From Reusing to Forecasting: Accelerating Diffusion Models with TaylorSeers
- Pseudo Numerical Methods for Diffusion Models on Manifolds
- Latent Consistency Models: Synthesizing High-Resolution Images with Few-Step Inference
- SDEdit: Guided Image Synthesis and Editing with Stochastic Differential Equations
- DINOv2: Learning Robust Visual Features without Supervision
- SDXL: Improving Latent Diffusion Models for High-Resolution Image Synthesis
- Improving User Privacy in Personalized Generation: Client-Side Retrieval-Augmented Modification of Server-Side Generated Speculations
- KNN-Diffusion: Image Generation via Large-Scale Retrieval
- Denoising Diffusion Implicit Models
Related papers
- Loss Knows Best: Detecting Annotation Errors in Videos via Loss Trajectories
- AnchorWeave: World-Consistent Video Generation with Retrieved Local Spatial Memories
- Benchmarking the Robustness of Foundation Models for Mammography under Domain Shift
- MambaX-Net: Dual-Input Mamba-Enhanced Cross-Attention Network for Longitudinal MRI Segmentation
- TeleOCR: Navigating Document Parsing Across Digital and Camera-Captured Documents
- A Survey on Efficient Vision-Language-Action Models