TokenAR: Multiple Subject Generation via Autoregressive Token-level enhancement

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Sun, Haiyue, He, Qingdong, Peng, Jinlong, Tang, Peng, Zhang, Jiangning, Zhu, Junwei, Hu, Xiaobin, Yan, Shuicheng
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866915561393356800
author Sun, Haiyue
He, Qingdong
Peng, Jinlong
Tang, Peng
Zhang, Jiangning
Zhu, Junwei
Hu, Xiaobin
Yan, Shuicheng
author_facet Sun, Haiyue
He, Qingdong
Peng, Jinlong
Tang, Peng
Zhang, Jiangning
Zhu, Junwei
Hu, Xiaobin
Yan, Shuicheng
contents Autoregressive Model (AR) has shown remarkable success in conditional image generation. However, these approaches for multiple reference generation struggle with decoupling different reference identities. In this work, we propose the TokenAR framework, specifically focused on a simple but effective token-level enhancement mechanism to address reference identity confusion problem. Such token-level enhancement consists of three parts, 1). Token Index Embedding clusters the tokens index for better representing the same reference images; 2). Instruct Token Injection plays as a role of extra visual feature container to inject detailed and complementary priors for reference tokens; 3). The identity-token disentanglement strategy (ITD) explicitly guides the token representations toward independently representing the features of each identity.This token-enhancement framework significantly augments the capabilities of existing AR based methods in conditional image generation, enabling good identity consistency while preserving high quality background reconstruction. Driven by the goal of high-quality and high-diversity in multi-subject generation, we introduce the InstructAR Dataset, the first open-source, large-scale, multi-reference input, open domain image generation dataset that includes 28K training pairs, each example has two reference subjects, a relative prompt and a background with mask annotation, curated for multiple reference image generation training and evaluating. Comprehensive experiments validate that our approach surpasses current state-of-the-art models in multiple reference image generation task. The implementation code and datasets will be made publicly. Codes are available, see https://github.com/lyrig/TokenAR
format Preprint
id arxiv_https___arxiv_org_abs_2510_16332
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle TokenAR: Multiple Subject Generation via Autoregressive Token-level enhancement
Sun, Haiyue
He, Qingdong
Peng, Jinlong
Tang, Peng
Zhang, Jiangning
Zhu, Junwei
Hu, Xiaobin
Yan, Shuicheng
Computer Vision and Pattern Recognition
Autoregressive Model (AR) has shown remarkable success in conditional image generation. However, these approaches for multiple reference generation struggle with decoupling different reference identities. In this work, we propose the TokenAR framework, specifically focused on a simple but effective token-level enhancement mechanism to address reference identity confusion problem. Such token-level enhancement consists of three parts, 1). Token Index Embedding clusters the tokens index for better representing the same reference images; 2). Instruct Token Injection plays as a role of extra visual feature container to inject detailed and complementary priors for reference tokens; 3). The identity-token disentanglement strategy (ITD) explicitly guides the token representations toward independently representing the features of each identity.This token-enhancement framework significantly augments the capabilities of existing AR based methods in conditional image generation, enabling good identity consistency while preserving high quality background reconstruction. Driven by the goal of high-quality and high-diversity in multi-subject generation, we introduce the InstructAR Dataset, the first open-source, large-scale, multi-reference input, open domain image generation dataset that includes 28K training pairs, each example has two reference subjects, a relative prompt and a background with mask annotation, curated for multiple reference image generation training and evaluating. Comprehensive experiments validate that our approach surpasses current state-of-the-art models in multiple reference image generation task. The implementation code and datasets will be made publicly. Codes are available, see https://github.com/lyrig/TokenAR
title TokenAR: Multiple Subject Generation via Autoregressive Token-level enhancement
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2510.16332