用32个分层掩码令牌实现图像分割,无需原图即可还原掩码。
HiMTok: Learning Hierarchical Mask Tokens for Image Segmentation with Large Multimodal Model
- 设计分层掩码令牌,用32个令牌从粗到细表示分割掩码。
- 在多个分割任务上达到顶尖性能,且提升视觉定位能力。
- 支持框与掩码间双向转换,适合多任务训练场景。
大型多模态模型(LMMs)在图像分割领域表现突出。现有方法多采用边界点或特殊分割令牌,依赖原始图像解码,存在表征不足和架构复杂的问题。本文提出分层掩码令牌生成器(HiMTok),以最多32个令牌表示分割掩码,并在解码时无需原始图像输入。该方法支持紧凑、粗到细的掩码表示,契合大语言模型的下一个词预测范式,直接赋予模型分割能力。我们设计三阶段训练策略,包含分层掩码损失,实现渐进式学习;并引入双向信息流,支持边界框与掩码令牌间的相互转换,充分挖掘多任务训练潜力。大量实验表明,该方法在多种分割任务中达到先进水平,同时增强视觉定位能力,保持整体视觉理解性能。
原文摘要 · Abstract (English)
The remarkable performance of large multimodal models (LMMs) has attracted significant interest from the image segmentation community. To align with the next-token-prediction paradigm, current LMM-driven segmentation methods either use object boundary points to represent masks or introduce special segmentation tokens, whose hidden states are decoded by a segmentation model requiring the original image as input. However, these approaches often suffer from inadequate mask representation and complex architectures, limiting the potential of LMMs. In this work, we propose the Hierarchical Mask Tokenizer (HiMTok), which represents segmentation masks with up to 32 tokens and eliminates the need for the original image during mask de-tokenization. HiMTok allows for compact and coarse-to-fine mask representations, aligning well with the LLM next-token-prediction paradigm and facilitating the direct acquisition of segmentation capabilities. We develop a 3-stage training recipe for progressive learning of segmentation and visual capabilities, featuring a hierarchical mask loss for effective coarse-to-fine learning. Additionally, we enable bidirectional information flow, allowing conversion between bounding boxes and mask tokens to fully leverage multi-task training potential. Extensive experiments demonstrate that our method achieves state-of-the-art performance across various segmentation tasks,while also enhancing visual grounding and maintaining overall visual understanding.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。