TokenCom: Vision-Language Model for Multimodal and Multitask Token Communications
Authors: Feibo Jiang, Siwei Tu, Li Dong, Xiaolong Li, Kezhi Wang, Cunhua Pan, Zhu Han, Jiangzhou Wang
Organizations: Hunan Provincial Key Laboratory of Intelligent Computing and Language Information Processing, Hunan Normal University, Changsha, China. · School of Information Science and Engineering, Hunan Normal University, Changsha, China. · Changsha Social Laboratory of Artificial Intelligence, Hunan University of Technology and Business, Changsha, China. · the School of Computer Science, Hunan University of Technology and Business, Changsha, China. · the Department of Computer Science, Brunel University London, UK. · the National Mobile Communications Research Laboratory, Southeast University, Nanjing 210096, China. · the Department of Electrical and Computer Engineering, University of Houston, Houston, TX, USA. · the National Mobile Communications Research Laboratory, Southeast University, Nanjing, China, and also with the Purple Mountain Laboratories, Nanjing, China.
Visual-Language Models (VLMs), with their strong capabilities in image and text understanding, offer a solid foundation for intelligent communications. However, their effectiveness is constrained by limited token granularity, overlong visual token sequences, and inadequate cross-modal alignment. To overcome these challenges, we propose TaiChi, a novel VLM framework designed for token communications. TaiChi adopts a dual-visual tokenizer architecture that processes both high- and low-resolution images to collaboratively capture pixel-level details and global conceptual features. A Bilateral Attention Network (BAN) is introduced to intelligently fuse multi-scale visual tokens, thereby enhancing visual understanding and producing compact visual tokens. In addition, a Kolmogorov Arnold Network (KAN)-based modality projector with learnable activation functions is employed to achieve precise nonlinear alignment from visual features to the text semantic space, thus minimizing information loss. Finally, TaiChi is integrated into a multimodal and multitask token communication system equipped with a joint VLM-channel coding scheme. Experimental results validate the superior performance of TaiChi, as well as the feasibility and effectiveness of the TaiChi-driven token communication system.
Figures & tables
Fig. 1 : The structure of the proposed TaiChi framework.
Fig. 2 : Dual-visual tokenizer architecture.
Fig. 3 : The computation pipeline of BAN.
Fig. 4 : The structure of KAN.
Fig. 5 : The structure of the TaiChi-driven token communication system.
Comparison Dimension
Token Communication
Semantic Communication
Information Unit
Discrete tokens serving as semantic symbols, generated via LLM/VLM-based semantic compression
Low-dimensional semantic features or embeddings extracted by task-specific neural networks
Encoding and Decoding
Tokenization and detokenization with generative semantic reconstruction
Semantic encoding and decoding through end-to-end trained models
Relation to LLMs/VLMs
Strong dependence on LLMs/VLMs for semantic understanding, reasoning, and generation
Large models are optional; typically relies on task-driven semantic compression networks
Cross-Modal Support
Native cross-modal support (text, image, audio) via unified token representation
Modality-specific encoders; lacks a unified cross-modal semantic representation
Transmission Robustness
High robustness enabled by generative contextual completion and token-level recovery
Robustness achieved mainly via end-to-end training; no explicit generative recovery mechanism
Task Generalization
Strong generalization benefiting from LLM/VLM knowledge and multi-task reasoning capability
Limited generalization due to task-specific training and weak transferability
TABLE I : Comparison Between TokCom and Semantic Communication
Fig. 6 : Different multimodal task examples for TaiChi.
Fig. 7 : The two training stages of TaiChi.
Fig. 8 : The joint coding process of TaiChi-driven token communication system.
Method
LLM
Res.
VQA T
MMB
MME
MobileVLM
MLLaMA 2.7B
336
47.5
59.6
1289
BLIP-2
Vicuna-13B
224
42.5
-
1293.8
InstructBLIP
Vicuna-7B
224
50.1
36.0
-
InstructBLIP
Vicuna-13B
224
50.7
-
1212.8
LLaVA-1.5
Vicuna-13B
336
61.3
69.2
1531/295
TaiChi
Gemma-2B
336
59.2
61.9
1323/240
TABLE II : Comparison on zero-shot benchmarks.
Method
VQA T
MMB
MM-Vet
TaiChi
59.2
65.1
33.3
TaiChi (w/o KAN)
55.0
60.7
24.5
TaiChi (w/o BAN)
59.0
60.9
28.5
TABLE III : Ablation study of TaiChi’s key components.
Fig. 9 : Performance of different tasks versus SNR: (a) VQAv2 dataset for VQA task; (b) SST2 dataset for text classification.
Fig. 10 : Performance of different tasks versus SNR: (a) CLEVR dataset for VQA task; (b) IMDB dataset for text classification.
Fig. 11 : Performance of token communication systems with TaiChi and other VLMs.
Fig. 12 : Performance of joint VLM-channel coding in token communication systems.
While Vision-Language Models (VLMs) demonstrate strong capabilities, they continue to suffer from a critical limitation: insufficient fine-grained visual perception, which fundamentally limits their multimodal understanding. We attribute this bottleneck to text-dominant optimization biases during pre-training, which encourage the model to overlook fine-grained visual details, thereby limiting the capability of multimodal understanding. We investigate that overcoming this bottleneck requires two key elements: (1) a unified token space paradigm that ensures stable training dynamics, and (2) a modality-aligned dense visual supervision signal enriched with both structural granularity and semantic information to capture critical visual representations. Based on these insights, we propose VIVAS, a framework built upon the unified token space paradigm, which introduces a dense-structural-semantic vision tokenizer, which expands the textual vocabulary into a unified vision-language vocabulary by incorporating a visual vocabulary. During pretraining, VIVAS performs vision-language unified autoregressive supervision over both visual details and linguistic content, thereby enhancing visual perception to improve multimodal understanding. Trained end-to-end on 12.4T tokens, VIVAS achieves state-of-the-art performance across 7 tasks and 39 multimodal benchmarks.
Zhehan Kan, Yubo Zhu, Xinghua Jiang +10
Tsinghua University · Tencent Youtu Lab · Nanjing University
Vision-Language Models (VLMs) have demonstrated strong capability in a wide range of tasks such as visual recognition, document parsing, and visual grounding. Nevertheless, recent work shows that while VLMs often manage to capture the correct image region corresponding to the question, they do not necessarily produce the correct answers. In this work, we demonstrate that this misalignment could be attributed to suboptimal information flow within VLMs, where text tokens distribute too much attention to irrelevant visual tokens, leading to incorrect answers. Based on the observation, we show that modulating the information flow during inference can improve the perception capability of VLMs. The idea is that text tokens should only be associated with important visual tokens during decoding, eliminating the interference of irrelevant regions. To achieve this, we propose a token dynamics-based method to determine the importance of visual tokens, where visual tokens that exhibit distinct activation patterns during different decoding stages are viewed as important. We apply our approach to representative open-source VLMs and evaluate on various datasets, including visual question answering, visual grounding and counting, optical character recognition, and object hallucination. The results show that our approach significantly improves the performance of baselines. Project page: https://cxliu0.github.io/AIF/.
Abundant visual information strengthens vision-language model (VLM) perception, yet massive visual tokens raise inference costs. Existing visual token pruning methods rely on similarity-based guidance, which exploits pairwise text-vision and vision-vision token correlations for compression. However, such methods only capture local layer-level signals and overlook the whole inference process in VLM. In this paper, we revisit VLM inference and present a new efficient guidance scheme that complements similarity-based guidance. In particular, we identify a key observation: as LLM layers deepen, text tokens continuously aggregate visual information via self-attention and progressively absorb partial visual content into textual representations. To quantify this phenomenon, we propose Cross Modal Absorption (CMA) from a geometric representation perspective to measure how much visual information is absorbed by text, revealing that more visual tokens in deeper layers can be approximately explained by the text subspace. We accordingly propose Cross Modal Residual (CMR). It projects visual tokens onto the text subspace via Tikhonov regularized least squares and exploits reconstruction residuals to quantify visual information that cannot be explained by text. Finally, based on CMR, we present SIEVE, a training-free visual token compression method that combines CMR, text-attention relevance, and residual-space diversity to retain task-relevant and complementary tokens. Experiments on diverse VLM architectures verify the effectiveness of SIEVE. For instance, on LLaVA-NeXT-7B, SIEVE keeps only 11.1% of visual tokens while preserving 97.5% of the original average performance, achieving 3.62× prefill speedup, 2.49× end-to-end speedup, and a 6.02× KV-cache reduction.
Congyang Ou, Ruike Song, Yang Zhou +3
Northwestern Polytechnical University · Chongqing University · MiLM Plus, Xiaomi Inc.