Adaptive Token Merging for Efficient Transformer Semantic Communication at the Edge
On the map: Efficient LLM inference and serving → Token pruning and merging for ViTs
Abstract (arXiv)
Large-scale transformers are central to modern semantic communication, yet their high computational and communication costs hinder deployment on resource-constrained edge devices. This paper introduces a training-free framework for adaptive token merging, a novel mechanism that compresses transformer representations at runtime by selectively merging semantically redundant tokens under per-layer similarity thresholds. Unlike prior fixed-ratio reduction, our approach couples merging directly to input redundancy, enabling data-dependent adaptation that balances efficiency and task relevance without retraining. We cast the discovery of merging strategies as a multi-objective optimization problem and leverage Bayesian optimization to obtain Pareto-optimal trade-offs between accuracy, inference cost, and communication cost. On ImageNet classification, we match the accuracy of the unmodified transformer with 30% fewer floating-point operations per second and under 20% of the original communication cost, while for visual question answering our method achieves performance competitive with the full LLaVA model at less than one-third of the compute and one-tenth of the bandwidth. Finally, we show that our adaptive merging is robust across varying channel conditions and provides inherent privacy benefits, substantially degrading the efficacy of model inversion attacks. Our framework provides a practical and versatile solution for deploying powerful transformer models in resource-limited edge intelligence scenarios.
Key facts. Cited by 7 papers in the map (data through 14 Oct 2026). Placed in Token pruning and merging for ViTs, one of 12 directions in Efficient LLM inference and serving. Builds on 37 of its 57 references that are in the map. First posted on arXiv 12 Sep 2025.
Builds on
37 of 57 references.
Closest ancestors · 3
Same direction, newest first
- 2023Token Fusion: Bridging the Gap between Token Pruning and Token Merging2 Dec
- 2022Token Merging: Your ViT But Faster17 Oct
- 2022Adaptive Sparse ViT: Towards Learnable Adaptive Token Pruning by Fully Exploiting Self-Attention28 Sep
Other references · 34
- 2017Attention Is All You Needcited by 60,361
- 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,744
- 2023GPT-4 Technical Reportcited by 18,396
- 2023Visual Instruction Tuningcited by 7,869
- 2021Masked Autoencoders Are Scalable Vision Learnerscited by 7,403
- 2022BLIP: Bootstrapping Language-Image Pre-training for Unified Vision-Language Understanding and Generationcited by 4,322
- 2015VQA: Visual Question Answeringcited by 2,506
- 2022Learn to Explain: Multimodal Reasoning via Thought Chains for Science Question Answeringcited by 1,919
- 2018A Tutorial on Bayesian Optimizationcited by 744
- 2023A Comprehensive Overview of Large Language Modelscited by 648
- 2021MobileViT: Light-weight, General-purpose, and Mobile-friendly Vision Transformercited by 642
- 2024An Image is Worth 1/2 Tokens After Layer 2: Plug-and-Play Inference Acceleration for Large Vision-Language Modelscited by 642
22 more
- 2024SparseVLM: Visual Token Sparsification for Efficient Vision-Language Model Inferencecited by 391
- 2017Adversarial Examples: Attacks and Defenses for Deep Learningcited by 351
- 2018A Comprehensive Survey of Deep Learning for Image Captioningcited by 214
- 2020Differentiable Expected Hypervolume Improvement for Parallel Multi-Objective Bayesian Optimizationcited by 179
- 2019A Vision of 6G Wireless Systems: Applications, Trends, Technologies, and Open Research Problemscited by 160
- 2019Edge AI: On-Demand Accelerating Deep Neural Network Inference via Edge Computingcited by 152
- 2022On The Computational Complexity of Self-Attentioncited by 139
- 2024Fit and Prune: Fast and Training-free Visual Token Pruning for Multi-modal Large Language Modelscited by 126
- 2021Learning Task-Oriented Communication for Edge Inference: An Information Bottleneck Approachcited by 110
- 2022Q-ViT: Accurate and Fully Quantized Low-bit Vision Transformercited by 108
- 2021Edge Artificial Intelligence for 6G: Vision, Enabling Technologies, and Applicationscited by 97
- 2022Less Data, More Knowledge: Building Next Generation Semantic Communication Networkscited by 62
- 2023SwinJSCC: Taming Swin Transformer for Deep Joint Source-Channel Codingcited by 45
- 2020AdvFlow: Inconspicuous Black-box Adversarial Attacks using Normalizing Flowscited by 36
- 2025Token Communications: A Large Model-Driven Framework for Cross-modal Context-aware Semantic Communicationscited by 35
- 2024Large Multi-Modal Models (LMMs) as Universal Foundation Models for AI-Native Wireless Systemscited by 27
- 2015A Bayesian approach to constrained single- and multi-objective optimizationcited by 23
- 2021Latency-Aware Neural Architecture Search with Multi-Objective Bayesian Optimizationcited by 15
- 2024Adaptive Semantic Token Selection for AI-native Goal-oriented Communicationscited by 10
- 2018User preferences in Bayesian multi-objective optimization: the expected weighted hypervolume improvement criterioncited by 4
- 2024Contrastive Learning and Adversarial Disentanglement for Privacy-Aware Task-Oriented Semantic Communicationcited by 3
- 2025Adaptive Pareto-Optimal Token Merging for Edge Transformer Models in Semantic Communicationcited by 2
Built on by
Citing papers per year: 2025: 1; 2026: 6.
7 papers.
- 2025ZACH-ViT: A Zero-Token Vision Transformer with ShuffleStrides Data Augmentation for Robust Lung Ultrasound Classificationcited by 2
- 2026Ada-TokenCom: Rate-Adaptive Token Communications via Large-Model-Driven Token Compression and Generationcited by 1
- 2026Topology-Preserving Deep Joint Source-Channel Coding for Semantic Communicationcited by 0
- 2026Position Matters: Feature Inversion Attacks in ViT Split Inference with Token Reduction and Shufflingcited by 0
- 2026From Semantic to Token Communication: The Next Paradigm for Large-Model-Driven 6G Intelligent Connectivitycited by 0
- 2026Queue-Aware Token Communications: Bridging AI Representation and Wireless Servicecited by 0
- 2026Scaling Vision Transformers Without Training: Prune-and-Merge Token Compressioncited by 0
- 2026Scaling Vision Transformers Without Training: Prune-and-Merge Token Compression13 Oct 2026
- 2026Queue-Aware Token Communications: Bridging AI Representation and Wireless Service12 Oct 2026
- 2026From Semantic to Token Communication: The Next Paradigm for Large-Model-Driven 6G Intelligent Connectivity9 Sep 2026
- 2026Position Matters: Feature Inversion Attacks in ViT Split Inference with Token Reduction and Shuffling1 Sep 2026
- 2026Ada-TokenCom: Rate-Adaptive Token Communications via Large-Model-Driven Token Compression and Generation28 Aug 2026
- 2026Topology-Preserving Deep Joint Source-Channel Coding for Semantic Communication17 Mar 2026
- 2025ZACH-ViT: A Zero-Token Vision Transformer with ShuffleStrides Data Augmentation for Robust Lung Ultrasound Classification20 Oct 2025
Similar, not cited
Close in wording, no citation either way
- 2026Decoder-Agnostic Token Merging for Vision Transformers: A Systematic Study of G2TM
- 2025Similarity-Aware Token Pruning: Your VLM but Faster
- 2026Fusion: A Framework for Unified Sequential Token AdaptatIon in VisiOn TraNsformers
- 2026RAPID: Layer-Wise Redundancy-Aware Pruning and Importance-Driven Token Merging for Efficient ViT
- 2025ASTRA: Communication-Efficient Acceleration for Multi-Device Transformer Inference
- 2025Efficient Token Compression for Vision Transformer with Spatial Information Preserved
- 2026DORA: Dynamic Online Reinforcement Agent for Token Merging in Vision Transformers
- 2025ToSA: Token Merging with Spatial Awareness