Integrating Pre-Trained Language Model with Physical Layer Communications

Ju-Hyung Lee, Dong-Ho Lee, Joohan Lee, Jay Pujara
2/18/2024
cs.ITcs.CLcs.LG

No official code found. Best implementation: abman23/on-device-ai-comm (25 stars, MIT, last commit 2024-05-26).

Abstract

The burgeoning field of on-device AI communication, where devices exchange information directly through embedded foundation models, such as language models (LMs), requires robust, efficient, and generalizable communication frameworks. However, integrating these frameworks with existing wireless systems and effectively managing noise and bit errors pose significant challenges. In this work, we introduce a practical ondevice AI communication framework, integrated with physical layer (PHY) communication functions, demonstrated through its performance on a link-level simulator. Our framework incorporates end-to-end training with channel noise to enhance resilience, incorporates vector quantized variational autoencoders (VQ-VAE) for efficient and robust communication, and utilizes pre-trained encoder-decoder transformers for improved generalization capabilities. Simulations, across various communication scenarios, reveal that our framework achieves a 50% reduction in transmission size while demonstrating substantial generalization ability and noise robustness under standardized 3GPP channel models.

View on arXivView PDF

Code Implementations(1)

MIT

On-device AI/LLM comm system | Pre-trained LLM integrated with 5G-NR PHY | Fine-tuned BART on noisy 3GPP CDL channels | 50 % compression via quantization | Evaluated with NVIDIA Sionna LLS

253Shell, PythonFeb 15, 20242 years agoMIT

Similar Papers

Cite this paper

@article{lee2024integrating,
  title  = {Integrating Pre-Trained Language Model with Physical Layer Communications},
  author = {Ju-Hyung Lee and Dong-Ho Lee and Joohan Lee and Jay Pujara},
  year   = {2024},
  eprint = {2402.11656},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2402.11656v2}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2402.11656.svg)](https://codeofpaper.com/paper/2402.11656)

Discussion