用轻量模型实现抗噪声的语义通信,性能接近大模型
Large-Scale Model Enabled Semantic Communication Based on Robust Knowledge Distillation
- 通过知识蒸馏搜索轻量化语义编码架构
- 模型参数减少同时保持高精度,抗信道噪声能力强
- 适合资源受限场景下的智能通信系统
大规模模型(LSMs)在语义表征与理解方面表现优异,可作为语义通信(SC)系统设计的有效框架。但其直接部署常受计算复杂度和资源需求制约。本文提出一种基于鲁棒知识蒸馏的语义通信(RKD-SC)框架,实现高效且抗信道噪声的LSM驱动通信。针对最优紧凑模型架构选择与知识有效迁移并保持鲁棒性两大挑战,首先提出基于知识蒸馏的轻量级可微分架构搜索(KDL-DARTS)算法,将知识蒸馏损失与复杂度惩罚融合于架构搜索过程,以发现高性能、轻量化的语义编码器结构;其次设计新型两阶段鲁棒知识蒸馏(RKD)算法,将大模型(教师)的语义能力迁移到紧凑编码器(学生)并提升系统鲁棒性;为进一步增强对信道损伤的抵抗能力,引入通道感知的Transformer(CAT)模块作为信道编解码器,在多样信道条件下训练且支持变长输出。图像分类任务的大量仿真结果表明,该框架显著降低模型参数,同时保持教师模型的高精度表现,并在抗干扰能力上优于现有方法。
原文摘要 · Abstract (English)
Large-scale models (LSMs) can be an effective framework for semantic representation and understanding, thereby providing a suitable tool for designing semantic communication (SC) systems. However, their direct deployment is often hindered by high computational complexity and resource requirements. In this paper, a novel robust knowledge distillation based semantic communication (RKD-SC) framework is proposed to enable efficient and \textcolor{black}{channel-noise-robust} LSM-powered SC. The framework addresses two key challenges: determining optimal compact model architectures and effectively transferring knowledge while maintaining robustness against channel noise. First, a knowledge distillation-based lightweight differentiable architecture search (KDL-DARTS) algorithm is proposed. This algorithm integrates knowledge distillation loss and a complexity penalty into the neural architecture search process to identify high-performance, lightweight semantic encoder architectures. Second, a novel two-stage robust knowledge distillation (RKD) algorithm is developed to transfer semantic capabilities from an LSM (teacher) to a compact encoder (student) and subsequently enhance system robustness. To further improve resilience to channel impairments, a channel-aware transformer (CAT) block is introduced as the channel codec, trained under diverse channel conditions with variable-length outputs. Extensive simulations on image classification tasks demonstrate that the RKD-SC framework significantly reduces model parameters while preserving a high degree of the teacher model's performance and exhibiting superior robustness compared to existing methods.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。