arXiv:2409.08538cs.LGcs.CR2024-09被引 25

提出隐私保护的分层学习框架,提升卫星通信效率与安全

An Efficient Privacy-aware Split Learning Framework for Satellite Communications

  • 结合差分隐私与图模型剪枝,动态优化神经网络结构
  • 在Amazon2M上降低50%算力消耗,保持0.82准确率
  • 适合关注卫星通信隐私与效率的研究者和工程师

在快速发展的卫星通信领域,引入先进机器学习技术,特别是分层学习(Split Learning),对提升卫星、空间站与地面站间的数据处理和模型训练效率至关重要。传统机器学习方法在卫星网络中常受限于带宽和计算资源。为此,我们提出一种新型高效分层学习框架——动态拓扑引导剪枝(DTIP),融合差分隐私与图结构及模型剪枝,以优化分布式学习中的图神经网络。DTIP对原始图数据施加差分隐私,并进行图神经网络剪枝,从而在各网络层级上同时优化模型规模与通信负载。在多个数据集上的实验表明,该方法显著提升了隐私保护、准确率与计算效率。具体而言,在Amazon2M数据集上,DTIP保持0.82准确率的同时实现每秒浮点运算量减少50%;在ArXiv数据集上,同样条件下达到0.85准确率。该框架不仅显著提升卫星通信的运行效率,还为隐私感知的分布式学习树立了新基准,有望革新空间网络中的数据处理方式。

原文摘要 · Abstract (English)

In the rapidly evolving domain of satellite communications, integrating advanced machine learning techniques, particularly split learning, is crucial for enhancing data processing and model training efficiency across satellites, space stations, and ground stations. Traditional ML approaches often face significant challenges within satellite networks due to constraints such as limited bandwidth and computational resources. To address this gap, we propose a novel framework for more efficient SL in satellite communications. Our approach, Dynamic Topology Informed Pruning, namely DTIP, combines differential privacy with graph and model pruning to optimize graph neural networks for distributed learning. DTIP strategically applies differential privacy to raw graph data and prunes GNNs, thereby optimizing both model size and communication load across network tiers. Extensive experiments across diverse datasets demonstrate DTIP's efficacy in enhancing privacy, accuracy, and computational efficiency. Specifically, on Amazon2M dataset, DTIP maintains an accuracy of 0.82 while achieving a 50% reduction in floating-point operations per second. Similarly, on ArXiv dataset, DTIP achieves an accuracy of 0.85 under comparable conditions. Our framework not only significantly improves the operational efficiency of satellite communications but also establishes a new benchmark in privacy-aware distributed learning, potentially revolutionizing data handling in space-based networks.

分层学习隐私保护卫星通信图神经网络

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。