arXiv:2412.14145cs.CV2024-12

用多尺度特征分层编码,让视觉令牌兼具语义与结构信息。

Incorporating Feature Pyramid Tokenization and Open Vocabulary Semantic Segmentation

  • 构建分层特征令牌化机制,联合重建与分割任务训练
  • 在开放词汇分割上超越基线模型,实现高精度像素级理解
  • 轻量级设计适合融合大模型,可独立使用于多种任务

视觉理解通常从图像、块和像素三个粒度层次展开。视觉令牌化通过自监督重建学习,在块级别以码本压缩视觉数据,信息损失较小,但令牌缺乏语义意义。开放词汇语义分割得益于具备强大零样本图像能力的视觉-语言模型(VLM),但如何将图像级理解迁移至像素级仍是关键挑战。本文将分割视为像素级令牌化,研究统一感知与语义的令牌压缩方法,促进开放词汇语义分割。参考预训练VLM中低层特征逐步组合成高层语义的认知过程,提出特征金字塔令牌化(PAT),利用可学习码本对多分辨率特征进行聚类与表示,并通过联合学习像素重建与语义分割进行解码。设计松耦合的像素与语义学习分支:像素分支模拟码本令牌的自底向上组合与自顶向下可视化,语义分支则聚合层级码本作为辅助分割引导。实验表明,PAT增强了VLM特征金字塔的语义直观性,提升了基线分割模型性能,在开放词汇语义分割基准上达到有竞争力的结果。模型参数高效,适合VLM集成,且独立令牌化灵活。希望为提升分割能力及语义视觉令牌利用提供新思路。

原文摘要 · Abstract (English)

The visual understanding are often approached from 3 granular levels: image, patch and pixel. Visual Tokenization, trained by self-supervised reconstructive learning, compresses visual data by codebook in patch-level with marginal information loss, but the visual tokens does not have semantic meaning. Open Vocabulary semantic segmentation benefits from the evolving Vision-Language models (VLMs) with strong image zero-shot capability, but transferring image-level to pixel-level understanding remains an imminent challenge. In this paper, we treat segmentation as tokenizing pixels and study a united perceptual and semantic token compression for all granular understanding and consequently facilitate open vocabulary semantic segmentation. Referring to the cognitive process of pretrained VLM where the low-level features are progressively composed to high-level semantics, we propose Feature Pyramid Tokenization (PAT) to cluster and represent multi-resolution feature by learnable codebooks and then decode them by joint learning pixel reconstruction and semantic segmentation. We design loosely coupled pixel and semantic learning branches. The pixel branch simulates bottom-up composition and top-down visualization of codebook tokens, while the semantic branch collectively fuse hierarchical codebooks as auxiliary segmentation guidance. Our experiments show that PAT enhances the semantic intuition of VLM feature pyramid, improves performance over the baseline segmentation model and achieves competitive performance on open vocabulary semantic segmentation benchmark. Our model is parameter-efficient for VLM integration and flexible for the independent tokenization. We hope to give inspiration not only on improving segmentation but also on semantic visual token utilization.

语义分割特征编码视觉语言模型多尺度建模

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。