用图扩散模型从次优数据中生成最优解,提升边缘计算任务调度效率。
GDSG: Graph Diffusion-based Solution Generator for Optimization Problems in MEC Networks
- 基于图神经网络的扩散模型,从次优解中学习最优解分布。
- 在次优数据上训练仍逼近100%最优解,超越传统方法。
- 适合研究边缘计算、智能交通系统中的优化问题者阅读。
为提升移动边缘计算(MEC)网络的运行效率,针对普遍存在的NP难优化问题,本文提出图扩散式解生成方法(GDSG)。以多服务器多用户计算卸载(MSCO)为例,该方法无需依赖大量最优数据,而是利用高效生成的次优解构建训练集,通过多任务扩散模型结合图神经网络,学习高质量解的分布。模型在训练中实现接近100%的任务正交性,确保离散与连续生成互不干扰。实验表明,无论在最优或次优数据集上,GDSG均显著优于现有基准方法。相关数据集和代码已开源,可访问 https://github.com/qiyu3816/GDSG。
原文摘要 · Abstract (English)
Optimization is crucial for MEC networks to function efficiently and reliably, most of which are NP-hard and lack efficient approximation algorithms. This leads to a paucity of optimal solution, constraining the effectiveness of conventional deep learning approaches. Most existing learning-based methods necessitate extensive optimal data and fail to exploit the potential benefits of suboptimal data that can be obtained with greater efficiency and effectiveness. Taking the multi-server multi-user computation offloading (MSCO) problem, which is widely observed in systems like Internet-of-Vehicles (IoV) and Unmanned Aerial Vehicle (UAV) networks, as a concrete scenario, we present a Graph Diffusion-based Solution Generation (GDSG) method. This approach is designed to work with suboptimal datasets while converging to the optimal solution large probably. We transform the optimization issue into distribution-learning and offer a clear explanation of learning from suboptimal training datasets. We build GDSG as a multi-task diffusion model utilizing a Graph Neural Network (GNN) to acquire the distribution of high-quality solutions. We use a simple and efficient heuristic approach to obtain a sufficient amount of training data composed entirely of suboptimal solutions. In our implementation, we enhance the backbone GNN and achieve improved generalization. GDSG also reaches nearly 100\% task orthogonality, ensuring no interference between the discrete and continuous generation tasks. We further reveal that this orthogonality arises from the diffusion-related training loss, rather than the neural network architecture itself. The experiments demonstrate that GDSG surpasses other benchmark methods on both the optimal and suboptimal training datasets. The MSCO datasets has open-sourced at this http URL, as well as the GDSG algorithm codes at https://github.com/qiyu3816/GDSG.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。