inclusionAI/ARGenSeg-8B

ARGenSeg: Image Segmentation with Autoregressive Image Generation Model

Xiaolong Wang¹  ·  Lixiang Ru¹  ·  Ziyuan Huang¹  ·  Kaixiang Ji¹
Dandan Zheng¹  ·  Jingdong Chen¹  ·  Jun Zhou¹

¹Ant Group  ·  NeurIPS 2025

🏠 About

We propose a novel AutoRegressive Generation-based paradigm for image Segmentation (ARGenSeg), achieving multimodal understanding and pixel-level perception within a unified framework. Prior works integrating image segmentation into multimodal large language models (MLLMs) typically employ either boundary points representation or dedicated segmentation heads. These methods rely on discrete representations or semantic prompts fed into task-specific decoders, which limits the ability of the MLLM to capture fine-grained visual details. To address these challenges, we introduce a segmentation framework for MLLM based on image generation, which naturally produces dense masks for target objects. We leverage MLLM to output visual tokens and detokenize them into images using an universal VQ-VAE, making the segmentation fully dependent on the pixel-level understanding of the MLLM. To reduce inference latency, we employ a next-scale-prediction strategy to generate required visual tokens in parallel. Extensive experiments demonstrate that our method surpasses prior state-of-the-art approaches on multiple segmentation datasets with a remarkable boost in inference speed, while maintaining strong understanding capabilities.

🔗 Citation

If you find this work useful, please cite:

@article{wang2025argenseg,
  title={ARGenSeg: Image Segmentation with Autoregressive Image Generation Model},
  author={Wang, Xiaolong and Ru, Lixiang and Huang, Ziyuan and Ji, Kaixiang and Zheng, Dandan and Chen, Jingdong and Zhou, Jun},
  journal={arXiv preprint arXiv:2510.20803},
  year={2025}
}

---

👏 Acknowledgements

We sincerely thank the contributors of InternVL, VAR, and PSALM for their foundational work and open-source spirit.