Adaptive Token Merging for Efficient Transformer Semantic Communication at the Edge

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Erak, Omar, Alhussein, Omar, Abou-Zeid, Hatem, Bennis, Mehdi, Muhaidat, Sami
Format: Preprint
Veröffentlicht: 2025
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866912584634990592
author Erak, Omar
Alhussein, Omar
Abou-Zeid, Hatem
Bennis, Mehdi
Muhaidat, Sami
author_facet Erak, Omar
Alhussein, Omar
Abou-Zeid, Hatem
Bennis, Mehdi
Muhaidat, Sami
contents 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.
format Preprint
id arxiv_https___arxiv_org_abs_2509_09955
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Adaptive Token Merging for Efficient Transformer Semantic Communication at the Edge
Erak, Omar
Alhussein, Omar
Abou-Zeid, Hatem
Bennis, Mehdi
Muhaidat, Sami
Machine Learning
Artificial Intelligence
Computer Vision and Pattern Recognition
Image and Video Processing
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.
title Adaptive Token Merging for Efficient Transformer Semantic Communication at the Edge
topic Machine Learning
Artificial Intelligence
Computer Vision and Pattern Recognition
Image and Video Processing
url https://arxiv.org/abs/2509.09955