卫星学习中联合优化数据重分配与训练,减少能耗和时延
SatDL: Jointly Optimizing Data Redistribution and Training for Satellite-Based Distributed Learning
- 设计分布式-批评者框架,协同优化数据传输与训练时间
- 在1584颗星链卫星上实测,总时延降低最多18.6%
- 适合低功耗、高实时性要求的太空边缘智能场景
基于卫星的分布式学习可通过全球分布的传感器数据直接在轨训练模型,避免大量数据下载至地面服务器。然而,由于各卫星观测区域地理差异导致标签分布不均(非独立同分布),严重影响训练收敛速度,延长训练周期并增加太阳能供电卫星的能耗。现有方法或完全重分配数据以实现独立同分布,虽加速收敛但带来显著通信延迟;或不进行重分配,仅通过算法调整缓解标签不平衡,仍导致训练时间长、能耗高。二者均造成过长的端到端学习时间(数据传输延迟+训练时间),提升星载能耗。本文提出SatDL,一种旨在最小化端到端学习时间的数据重分配框架。其核心为分布式-批评者框架,联合建模并优化数据传输延迟与训练时间。通过1,584颗星链卫星的追踪仿真及基于NVIDIA Jetson和A100 GPU的硬件模拟,在五个数据集上验证显示,SatDL可将总端到端学习时间减少最多18.6%,星载能耗降低12.23%-88.00%,同时保持推理精度与顶尖基准相当。
原文摘要 · Abstract (English)
Satellite-based distributed learning promises to train machine-learning models directly in orbit using massive, globally dispersed sensor data, thereby avoiding large-scale data downloads to ground servers. However, training convergence is significantly slowed by severe non-IID data, specifically label imbalance, as each satellite observes different geographic regions with distinct labels. This imbalance extends training duration and increases energy consumption for solar-powered satellites. Existing approaches either fully redistribute data to enforce IID conditions - accelerating convergence but incurring substantial communication delays - or avoid redistribution entirely by modifying local learning algorithms to mitigate the impact of label imbalance, which, however, still prolong training and increase energy use. Both extremes result in excessive total end-to-end learning time (data-transfer delay plus training time) and thus elevated onboard energy consumption. We present SatDL, a data-redistribution framework designed to minimize total end-to-end learning time. At its core, SatDL develops a Distributor-Critic framework that jointly models and optimizes data-transfer delay and training time. Evaluations through trace-driven simulations of a 1,584-satellite Starlink constellation and hardware emulations using NVIDIA Jetson and A100 GPUs across five datasets show SatDL reduces total end-to-end learning time by up to 18.6% and onboard energy consumption by 12.23-88.00%, while maintaining inference accuracy within a few percentage points of state-of-the-art baselines.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。