Re-Align: Structured Reasoning-guided Alignment for In-Context Image Generation and Editing

     

Re-Align: Structured Reasoning-guided Alignment for In-Context Image Generation and Editing

Runze He^1,2,3, Yiji Cheng¹, Tiankai Hang¹, Zhimin Li¹, Yu Xu¹, Zijin Yin¹, Shiyi Zhang¹, Wenxun Dai¹, Penghui Du³, Ao Ma³, Chunyu Wang^1,†, Qinglin Lu¹, Jizhong Han^2,3, Jiao Dai^2,3,‡

¹Hunyuan, Tencent, ²IIE, CAS, ³UCAS

In-context image generation and editing (ICGE) enables users to specify visual concepts through interleaved image-text prompts, demanding precise understanding and faithful execution of user intent. Although recent unified multimodal models exhibit promising understanding capabilities, these strengths often fail to transfer effectively to image generation. We introduce Re-Align, a unified framework that bridges the gap between understanding and generation through structured reasoning-guided alignment. At its core lies the In-Context Chain-of-Thought (IC-CoT), a structured reasoning paradigm that decouples semantic guidance and reference association, providing clear textual target and mitigating confusion among reference images. Furthermore, Re-Align introduces an effective RL training scheme that leverages a surrogate reward to measure the alignment between structured reasoning text and the generated image, thereby improving the model’s overall performance on ICGE tasks. Extensive experiments verify that Re-Align outperforms competitive methods of comparable model scale and resources on both in-context image generation and editing tasks.

---

🔥 News

• [2026/1/9] We released the paper of ReAlign.

BibTeX

@misc{he2026realign,
      title={Re-Align: Structured Reasoning-guided Alignment for In-Context Image Generation and Editing}, 
      author={Runze He and Yiji Cheng and Tiankai Hang and Zhimin Li and Yu Xu and Zijin Yin and Shiyi Zhang and Wenxun Dai and Penghui Du and Ao Ma and Chunyu Wang and Qinglin Lu and Jizhong Han and Jiao Dai},
      year={2026},
      eprint={2601.05124},
      archivePrefix={arXiv},
      primaryClass={cs.CV},
      url={https://arxiv.org/abs/2601.05124}, 
}