mlsanityData to 14 Oct 2026

Adaptive Token Merging for Efficient Transformer Semantic Communication at the Edge

arXiv 2509.09955 ↗First posted 12 Sep 2025cs.LG cs.AI cs.CVcited by 7

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

  1. 2023Token Fusion: Bridging the Gap between Token Pruning and Token Merging2 Dec
  2. 2022Token Merging: Your ViT But Faster17 Oct
  3. 2022Adaptive Sparse ViT: Towards Learnable Adaptive Token Pruning by Fully Exploiting Self-Attention28 Sep

Other references · 34

  1. 2017Attention Is All You Needcited by 60,361
  2. 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,744
  3. 2023GPT-4 Technical Reportcited by 18,396
  4. 2023Visual Instruction Tuningcited by 7,869
  5. 2021Masked Autoencoders Are Scalable Vision Learnerscited by 7,403
  6. 2022BLIP: Bootstrapping Language-Image Pre-training for Unified Vision-Language Understanding and Generationcited by 4,322
  7. 2015VQA: Visual Question Answeringcited by 2,506
  8. 2022Learn to Explain: Multimodal Reasoning via Thought Chains for Science Question Answeringcited by 1,919
  9. 2018A Tutorial on Bayesian Optimizationcited by 744
  10. 2023A Comprehensive Overview of Large Language Modelscited by 648
  11. 2021MobileViT: Light-weight, General-purpose, and Mobile-friendly Vision Transformercited by 642
  12. 2024An Image is Worth 1/2 Tokens After Layer 2: Plug-and-Play Inference Acceleration for Large Vision-Language Modelscited by 642
22 more
  1. 2024SparseVLM: Visual Token Sparsification for Efficient Vision-Language Model Inferencecited by 391
  2. 2017Adversarial Examples: Attacks and Defenses for Deep Learningcited by 351
  3. 2018A Comprehensive Survey of Deep Learning for Image Captioningcited by 214
  4. 2020Differentiable Expected Hypervolume Improvement for Parallel Multi-Objective Bayesian Optimizationcited by 179
  5. 2019A Vision of 6G Wireless Systems: Applications, Trends, Technologies, and Open Research Problemscited by 160
  6. 2019Edge AI: On-Demand Accelerating Deep Neural Network Inference via Edge Computingcited by 152
  7. 2022On The Computational Complexity of Self-Attentioncited by 139
  8. 2024Fit and Prune: Fast and Training-free Visual Token Pruning for Multi-modal Large Language Modelscited by 126
  9. 2021Learning Task-Oriented Communication for Edge Inference: An Information Bottleneck Approachcited by 110
  10. 2022Q-ViT: Accurate and Fully Quantized Low-bit Vision Transformercited by 108
  11. 2021Edge Artificial Intelligence for 6G: Vision, Enabling Technologies, and Applicationscited by 97
  12. 2022Less Data, More Knowledge: Building Next Generation Semantic Communication Networkscited by 62
  13. 2023SwinJSCC: Taming Swin Transformer for Deep Joint Source-Channel Codingcited by 45
  14. 2020AdvFlow: Inconspicuous Black-box Adversarial Attacks using Normalizing Flowscited by 36
  15. 2025Token Communications: A Large Model-Driven Framework for Cross-modal Context-aware Semantic Communicationscited by 35
  16. 2024Large Multi-Modal Models (LMMs) as Universal Foundation Models for AI-Native Wireless Systemscited by 27
  17. 2015A Bayesian approach to constrained single- and multi-objective optimizationcited by 23
  18. 2021Latency-Aware Neural Architecture Search with Multi-Objective Bayesian Optimizationcited by 15
  19. 2024Adaptive Semantic Token Selection for AI-native Goal-oriented Communicationscited by 10
  20. 2018User preferences in Bayesian multi-objective optimization: the expected weighted hypervolume improvement criterioncited by 4
  21. 2024Contrastive Learning and Adversarial Disentanglement for Privacy-Aware Task-Oriented Semantic Communicationcited by 3
  22. 2025Adaptive Pareto-Optimal Token Merging for Edge Transformer Models in Semantic Communicationcited by 2

Built on by

Cited by year (2026 to 14 Oct)

Citing papers per year: 2025: 1; 2026: 6.

7 papers.

Similar, not cited

Close in wording, no citation either way

  1. 2026Decoder-Agnostic Token Merging for Vision Transformers: A Systematic Study of G2TM
  2. 2025Similarity-Aware Token Pruning: Your VLM but Faster
  3. 2026Fusion: A Framework for Unified Sequential Token AdaptatIon in VisiOn TraNsformers
  4. 2026RAPID: Layer-Wise Redundancy-Aware Pruning and Importance-Driven Token Merging for Efficient ViT
  5. 2025ASTRA: Communication-Efficient Acceleration for Multi-Device Transformer Inference
  6. 2025Efficient Token Compression for Vision Transformer with Spatial Information Preserved
  7. 2026DORA: Dynamic Online Reinforcement Agent for Token Merging in Vision Transformers
  8. 2025ToSA: Token Merging with Spatial Awareness