StructDiff: A Structure-Preserving and Spatially Controllable Diffusion Model for Single-Image Generation

Yinxi He, Kang Liao, Chunyu Lin, Tianyi Wei, Yao Zhao
4/14/2026
cs.CV

No official code found. Best implementation: Butter-Crab/StructDiff (5 stars, last commit 2026-05-11).

Abstract

This paper introduces StructDiff, a generative framework based on a single-scale diffusion model for single-image generation. Single-image generation aims to synthesize diverse samples with similar visual content to the source image by capturing its internal statistics, without relying on external data. However, existing methods often struggle to preserve the structural layout, especially for images with large rigid objects or strict spatial constraints. Moreover, most approaches lack spatial controllability, making it difficult to guide the structure or placement of generated content. To address these challenges, StructDiff introduces an \textit{adaptive receptive field} module to maintain both global and local distributions. Building on this foundation, StructDiff incorporates 3D positional encoding (PE) as a spatial prior, allowing flexible control over positions, scale, and local details of generated objects. To our knowledge, this spatial control capability represents the first exploration of PE-based manipulation in single-image generation. Furthermore, we propose a novel evaluation criterion for single-image generation based on large language models (LLMs). This criterion specifically addresses the limitations of existing objective metrics and the high labor costs associated with user studies. StructDiff also demonstrates broad applicability across downstream tasks, such as text-guided image generation, image editing, outpainting, and paint-to-image synthesis. Extensive experiments demonstrate that StructDiff outperforms existing methods in structural consistency, visual quality, and spatial controllability. The project page is available at https://butter-crab.github.io/StructDiff/.

View on arXivView PDF

Code Implementations(1)

Official pytorch implementation of the paper: " StructDiff: A Structure-Preserving and Spatially Controllable Diffusion Model for Single-Image Generation"

50Apr 14, 20265 months ago

Cite this paper

@article{he2026structdiff,
  title  = {StructDiff: A Structure-Preserving and Spatially Controllable Diffusion Model for Single-Image Generation},
  author = {Yinxi He and Kang Liao and Chunyu Lin and Tianyi Wei and Yao Zhao},
  year   = {2026},
  eprint = {2604.12575},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2604.12575v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2604.12575.svg)](https://codeofpaper.com/paper/2604.12575)

Discussion