Adaptive Pareto-Optimal Token Merging for Edge Transformer Models in Semantic Communication

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Erak, Omar, Alhussein, Omar, Abou-Zeid, Hatem, Bennis, Mehdi
Format: Preprint
Published: 2025
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909901481050112
author Erak, Omar
Alhussein, Omar
Abou-Zeid, Hatem
Bennis, Mehdi
author_facet Erak, Omar
Alhussein, Omar
Abou-Zeid, Hatem
Bennis, Mehdi
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2509_09168
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Adaptive Pareto-Optimal Token Merging for Edge Transformer Models in Semantic Communication
Erak, Omar
Alhussein, Omar
Abou-Zeid, Hatem
Bennis, Mehdi
Machine Learning
Artificial Intelligence
Computer Vision and Pattern Recognition
Image and Video Processing
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.
title Adaptive Pareto-Optimal Token Merging for Edge Transformer Models in Semantic Communication
topic Machine Learning
Artificial Intelligence
Computer Vision and Pattern Recognition
Image and Video Processing
url https://arxiv.org/abs/2509.09168