cs.CVApr 15, 2026

ASTRA: Enhancing Multi-Subject Generation with Retrieval-Augmented Pose Guidance and Disentangled Position Embedding

Authors: Tianze Xia, Zijian Ning, Zonglin Zhao, Mingjia Wang

Organizations: Huazhong University of Science and Technology

Abstract

Subject-driven image generation has shown great success in creating personalized content, but its capabilities are largely confined to single subjects in common poses. Current approaches face a fundamental conflict when handling multiple subjects with complex, distinct actions: preserving individual identities while enforcing precise pose structures. This challenge often leads to identity fusion and pose distortion, as appearance and structure signals become entangled within the model's architecture. To resolve this conflict, we introduce ASTRA(Adaptive Synthesis through Targeted Retrieval Augmentation), a novel framework that architecturally disentangles subject appearance from pose structure within a unified Diffusion Transformer. ASTRA achieves this through a dual-pronged strategy. It first employs a Retrieval-Augmented Pose (RAG-Pose) pipeline to provide a clean, explicit structural prior from a curated database. Then, its core generative model learns to process these dual visual conditions using our Enhanced Universal Rotary Position Embedding (EURoPE), an asymmetric encoding mechanism that decouples identity tokens from spatial locations while binding pose tokens to the canvas. Concurrently, a Disentangled Semantic Modulation (DSM) adapter offloads the identity preservation task into the text conditioning stream. Extensive experiments demonstrate that our integrated approach achieves superior disentanglement. On our designed COCO-based complex pose benchmark, ASTRA achieves a new state-of-the-art in pose adherence, while maintaining high identity fidelity and text alignment in DreamBench.

Figures & tables

Explore similar work

CardsList
  1. TrioPose: Native Triple-Stream Diffusion Transformers for Pose-Guided Text-to-Image Generation

    Jun 5, 2026Dian Gu, Zhengyi YangDiffusion TransformerT2I Generation

  2. Hierarchical Concept-to-Appearance Guidance for Multi-Subject Image Generation

    Feb 3, 2026Yijia Xu, Zihao Wang, Haokun Gui +1Subject-Driven Image GenerationImage Generation

  3. Pose-ICL: 3D-Aware In-Context Learning for Pose-Controllable Subject Customization

    Jun 9, 2026Xuan Han, Yihao Zhao, Mingyu YouSubject-Driven Image GenerationIn-Context Learning