Remote Training in Task-Oriented Communication: Supervised or Self-Supervised with Fine-Tuning?

Fuente: arXiv
Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Li, Hongru, Zhao, Hang, He, Hengtao, Song, Shenghui, Zhang, Jun, Letaief, Khaled B.
Format: Preprint
Veröffentlicht: 2025
Schlagworte:
Online-Zugang:
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866915171673309184
author Li, Hongru
Zhao, Hang
He, Hengtao
Song, Shenghui
Zhang, Jun
Letaief, Khaled B.
author_facet Li, Hongru
Zhao, Hang
He, Hengtao
Song, Shenghui
Zhang, Jun
Letaief, Khaled B.
contents Task-oriented communication focuses on extracting and transmitting only the information relevant to specific tasks, effectively minimizing communication overhead. Most existing methods prioritize reducing this overhead during inference, often assuming feasible local training or minimal training communication resources. However, in real-world wireless systems with dynamic connection topologies, training models locally for each new connection is impractical, and task-specific information is often unavailable before establishing connections. Therefore, minimizing training overhead and enabling label-free, task-agnostic pre-training before the connection establishment are essential for effective task-oriented communication. In this paper, we tackle these challenges by employing a mutual information maximization approach grounded in self-supervised learning and information-theoretic analysis. We propose an efficient strategy that pre-trains the transmitter in a task-agnostic and label-free manner, followed by joint fine-tuning of both the transmitter and receiver in a task-specific, label-aware manner. Simulation results show that our proposed method reduces training communication overhead to about half that of full-supervised methods using the SGD optimizer, demonstrating significant improvements in training efficiency.
format Preprint
id arxiv_https___arxiv_org_abs_2502_17922
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Remote Training in Task-Oriented Communication: Supervised or Self-Supervised with Fine-Tuning?
Li, Hongru
Zhao, Hang
He, Hengtao
Song, Shenghui
Zhang, Jun
Letaief, Khaled B.
Information Theory
Signal Processing
Task-oriented communication focuses on extracting and transmitting only the information relevant to specific tasks, effectively minimizing communication overhead. Most existing methods prioritize reducing this overhead during inference, often assuming feasible local training or minimal training communication resources. However, in real-world wireless systems with dynamic connection topologies, training models locally for each new connection is impractical, and task-specific information is often unavailable before establishing connections. Therefore, minimizing training overhead and enabling label-free, task-agnostic pre-training before the connection establishment are essential for effective task-oriented communication. In this paper, we tackle these challenges by employing a mutual information maximization approach grounded in self-supervised learning and information-theoretic analysis. We propose an efficient strategy that pre-trains the transmitter in a task-agnostic and label-free manner, followed by joint fine-tuning of both the transmitter and receiver in a task-specific, label-aware manner. Simulation results show that our proposed method reduces training communication overhead to about half that of full-supervised methods using the SGD optimizer, demonstrating significant improvements in training efficiency.
title Remote Training in Task-Oriented Communication: Supervised or Self-Supervised with Fine-Tuning?
topic Information Theory
Signal Processing
url https://arxiv.org/abs/2502.17922