arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2408.17397cs.ITeess.SPmath.IT
收藏

面向MIMO信道任务型语义通信的端到端学习:一种信息论框架

End-to-End Learning for Task-Oriented Semantic Communications Over MIMO Channels: An Information-Theoretic Framework

Chang Cai, Xiaojun Yuan, Ying-Jun Angela Zhang

AI总结:

本文提出一种信息论框架下的解耦预训练与深度展开预编码方法,用于MIMO多址信道上的任务型语义通信端到端学习,在CIFAR-10和ModelNet10上显著提升分类准确率。

摘要:

本文研究任务型语义通信系统中学习与通信的端到端(E2E)设计问题。具体而言,我们考虑一个基于无线多输入多输出(MIMO)多址信道的多设备协作边缘推理系统,其中多个设备将提取的特征传输到服务器以执行分类任务。我们将特征编码、MIMO预编码和分类的端到端设计建模为一个条件互信息最大化问题。然而,设计并训练一个能够同时适应任务数据集和不同信道实现的端到端网络是众所周知的困难问题。在网络训练方面,我们提出了一种解耦预训练框架,分别训练特征编码器和MIMO预编码器,并在服务器端采用最大后验(MAP)分类器生成推理结果。特征编码器仅使用任务数据集进行预训练,而MIMO预编码器仅基于信道和噪声分布进行预训练。尽管如此,我们成功地将每个单独组件的预训练目标与端到端学习目标对齐,从而逼近端到端学习的性能界。通过利用解耦预训练结果进行初始化,端到端学习可以以最小的训练开销进行。在网络架构设计方面,我们开发了两种深度展开预编码网络,它们有效地融入了解耦预编码问题解的领域知识。在CIFAR-10和ModelNet10数据集上的仿真结果表明,与各种基线方法相比,所提方法显著提高了分类准确率。

英文摘要:

This paper addresses the problem of end-to-end (E2E) design of learning and communication in a task-oriented semantic communication system. In particular, we consider a multi-device cooperative edge inference system over a wireless multiple-input multiple-output (MIMO) multiple access channel, where multiple devices transmit extracted features to a server to perform a classification task. We formulate the E2E design of feature encoding, MIMO precoding, and classification as a conditional mutual information maximization problem. However, it is notoriously difficult to design and train an E2E network that can be adaptive to both the task dataset and different channel realizations. Regarding network training, we propose a decoupled pretraining framework that separately trains the feature encoder and the MIMO precoder, with a maximum a posteriori (MAP) classifier employed at the server to generate the inference result. The feature encoder is pretrained exclusively using the task dataset, while the MIMO precoder is pretrained solely based on the channel and noise distributions. Nevertheless, we manage to align the pretraining objectives of each individual component with the E2E learning objective, so as to approach the performance bound of E2E learning. By leveraging the decoupled pretraining results for initialization, the E2E learning can be conducted with minimal training overhead. Regarding network architecture design, we develop two deep unfolded precoding networks that effectively incorporate the domain knowledge of the solution to the decoupled precoding problem. Simulation results on both the CIFAR-10 and ModelNet10 datasets verify that the proposed method achieves significantly higher classification accuracy compared to various baselines.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑