Concept-Centric Token Interpretation for Vector-Quantized Generative Models

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Yang, Tianze, Shi, Yucheng, Du, Mengnan, Wu, Xuansheng, Tan, Qiaoyu, Sun, Jin, Liu, Ninghao
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866913002883645440
author Yang, Tianze
Shi, Yucheng
Du, Mengnan
Wu, Xuansheng
Tan, Qiaoyu
Sun, Jin
Liu, Ninghao
author_facet Yang, Tianze
Shi, Yucheng
Du, Mengnan
Wu, Xuansheng
Tan, Qiaoyu
Sun, Jin
Liu, Ninghao
contents Vector-Quantized Generative Models (VQGMs) have emerged as powerful tools for image generation. However, the key component of VQGMs -- the codebook of discrete tokens -- is still not well understood, e.g., which tokens are critical to generate an image of a certain concept? This paper introduces Concept-Oriented Token Explanation (CORTEX), a novel approach for interpreting VQGMs by identifying concept-specific token combinations. Our framework employs two methods: (1) a sample-level explanation method that analyzes token importance scores in individual images, and (2) a codebook-level explanation method that explores the entire codebook to find globally relevant tokens. Experimental results demonstrate CORTEX's efficacy in providing clear explanations of token usage in the generative process, outperforming baselines across multiple pretrained VQGMs. Besides enhancing VQGMs transparency, CORTEX is useful in applications such as targeted image editing and shortcut feature detection. Our code is available at https://github.com/YangTianze009/CORTEX.
format Preprint
id arxiv_https___arxiv_org_abs_2506_00698
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Concept-Centric Token Interpretation for Vector-Quantized Generative Models
Yang, Tianze
Shi, Yucheng
Du, Mengnan
Wu, Xuansheng
Tan, Qiaoyu
Sun, Jin
Liu, Ninghao
Computer Vision and Pattern Recognition
Machine Learning
Vector-Quantized Generative Models (VQGMs) have emerged as powerful tools for image generation. However, the key component of VQGMs -- the codebook of discrete tokens -- is still not well understood, e.g., which tokens are critical to generate an image of a certain concept? This paper introduces Concept-Oriented Token Explanation (CORTEX), a novel approach for interpreting VQGMs by identifying concept-specific token combinations. Our framework employs two methods: (1) a sample-level explanation method that analyzes token importance scores in individual images, and (2) a codebook-level explanation method that explores the entire codebook to find globally relevant tokens. Experimental results demonstrate CORTEX's efficacy in providing clear explanations of token usage in the generative process, outperforming baselines across multiple pretrained VQGMs. Besides enhancing VQGMs transparency, CORTEX is useful in applications such as targeted image editing and shortcut feature detection. Our code is available at https://github.com/YangTianze009/CORTEX.
title Concept-Centric Token Interpretation for Vector-Quantized Generative Models
topic Computer Vision and Pattern Recognition
Machine Learning
url https://arxiv.org/abs/2506.00698