Adaptive Pareto-Optimal Token Merging for Edge Transformer Models in Semantic Communication
On the map: Efficient LLM inference and serving → Early exit and edge inference
Abstract (arXiv)
Large-scale transformer models have emerged as a powerful tool for semantic communication systems, enabling edge devices to extract rich representations for robust inference across noisy wireless channels. However, their substantial computational demands remain a major barrier to practical deployment in resource-constrained 6G networks. In this paper, we present a training-free framework for adaptive token merging in pretrained vision transformers to jointly reduce inference time and transmission resource usage. We formulate the selection of per-layer merging proportions as a multi-objective optimization problem to balance accuracy and computational cost. We employ Gaussian process-based Bayesian optimization to construct a Pareto frontier of optimal configurations, enabling flexible runtime adaptation to dynamic application requirements and channel conditions. Extensive experiments demonstrate that our method consistently outperforms other baselines and achieves significant reductions in floating-point operations while maintaining competitive accuracy across a wide range of signal-to-noise ratio (SNR) conditions. Additional results highlight the effectiveness of adaptive policies that adjust merging aggressiveness in response to channel quality, providing a practical mechanism to trade off latency and semantic fidelity on demand. These findings establish a scalable and efficient approach for deploying transformer-based semantic communication in future edge intelligence systems.
Key facts. Cited by 2 papers in the map (data through 14 Oct 2026). Placed in Early exit and edge inference, one of 12 directions in Efficient LLM inference and serving. Builds on 16 of its 23 references that are in the map. First posted on arXiv 11 Sep 2025.
Builds on
16 of 23 references.
References · 16
- 2017Attention Is All You Needcited by 60,361
- 2020An Image is Worth 16x16 Words: Transformers for Image Recognition at Scalecited by 30,186
- 2019Optuna: A Next-generation Hyperparameter Optimization Frameworkcited by 2,501
- 2022Token Merging: Your ViT But Fastercited by 1,021
- 2018A Tutorial on Bayesian Optimizationcited by 744
- 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
- 2022On The Computational Complexity of Self-Attentioncited by 139
- 2021Edge Artificial Intelligence for 6G: Vision, Enabling Technologies, and Applicationscited by 97
- 2022Less Data, More Knowledge: Building Next Generation Semantic Communication Networkscited by 62
- 2022Adaptive Sparse ViT: Towards Learnable Adaptive Token Pruning by Fully Exploiting Self-Attentioncited by 35
- 2025Token Communications: A Large Model-Driven Framework for Cross-modal Context-aware Semantic Communicationscited by 35
4 more
- 2021Latency-Aware Neural Architecture Search with Multi-Objective Bayesian Optimizationcited by 15
- 2024Adaptive Semantic Token Selection for AI-native Goal-oriented Communicationscited by 10
- 2024Contrastive Learning and Adversarial Disentanglement for Privacy-Aware Task-Oriented Semantic Communicationcited by 3
- 2023Accelerating and Compressing Deep Neural Networks for Massive MIMO CSI Feedbackcited by 2
Built on by
Citing papers per year: 2025: 1; 2026: 1.
2 papers.
Similar, not cited
Close in wording, no citation either way
- 2025TOAST: Task-Oriented Adaptive Semantic Transmission over Dynamic Wireless Environments
- 2026Adaptive Semantic Communication for Wireless Image Transmission Leveraging Mixture-of-Experts Mechanism
- 2026Channel-Adaptive Edge AI: Maximizing Inference Throughput by Adapting Computational Complexity to Channel States
- 2024Vision Transformer-based Semantic Communications With Importance-Aware Quantization
- 2025EdgeFlex-Transformer: Transformer Inference for Edge Devices
- 2026Progressive Semantic Communication for Efficient Edge-Cloud Vision-Language Models
- 2023Adaptive Resource Allocation for Semantic Communication Networks
- 2026Closing the Semantic-Edge Gap: Tiny Language Models for 6G Wireless Intelligence ↗