DiffSG: A Generative Solver for Network Optimization with Diffusion Model

Fuente: arXiv
Salvato in:
Dettagli Bibliografici
Autori principali: Liang, Ruihuai, Yang, Bo, Yu, Zhiwen, Guo, Bin, Cao, Xuelin, Debbah, Mérouane, Poor, H. Vincent, Yuen, Chau
Natura: Preprint
Pubblicazione: 2024
Soggetti:
Accesso online:
Tags: Aggiungi Tag
Nessun Tag, puoi essere il primo ad aggiungerne!!
_version_ 1866929749895413760
author Liang, Ruihuai
Yang, Bo
Yu, Zhiwen
Guo, Bin
Cao, Xuelin
Debbah, Mérouane
Poor, H. Vincent
Yuen, Chau
author_facet Liang, Ruihuai
Yang, Bo
Yu, Zhiwen
Guo, Bin
Cao, Xuelin
Debbah, Mérouane
Poor, H. Vincent
Yuen, Chau
contents Generative diffusion models, famous for their performance in image generation, are popular in various cross-domain applications. However, their use in the communication community has been mostly limited to auxiliary tasks like data modeling and feature extraction. These models hold greater promise for fundamental problems in network optimization compared to traditional machine learning methods. Discriminative deep learning often falls short due to its single-step input-output mapping and lack of global awareness of the solution space, especially given the complexity of network optimization's objective functions. In contrast, generative diffusion models can consider a broader range of solutions and exhibit stronger generalization by learning parameters that describe the distribution of the underlying solution space, with higher probabilities assigned to better solutions. We propose a new framework Diffusion Model-based Solution Generation (DiffSG), which leverages the intrinsic distribution learning capabilities of generative diffusion models to learn high-quality solution distributions based on given inputs. The optimal solution within this distribution is highly probable, allowing it to be effectively reached through repeated sampling. We validate the performance of DiffSG on several typical network optimization problems, including mixed-integer non-linear programming, convex optimization, and hierarchical non-convex optimization. Our results demonstrate that DiffSG outperforms existing baseline methods not only on in-domain inputs but also on out-of-domain inputs. In summary, we demonstrate the potential of generative diffusion models in tackling complex network optimization problems and outline a promising path for their broader application in the communication community. Our code is available at https://github.com/qiyu3816/DiffSG.
format Preprint
id arxiv_https___arxiv_org_abs_2408_06701
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle DiffSG: A Generative Solver for Network Optimization with Diffusion Model
Liang, Ruihuai
Yang, Bo
Yu, Zhiwen
Guo, Bin
Cao, Xuelin
Debbah, Mérouane
Poor, H. Vincent
Yuen, Chau
Networking and Internet Architecture
Machine Learning
Generative diffusion models, famous for their performance in image generation, are popular in various cross-domain applications. However, their use in the communication community has been mostly limited to auxiliary tasks like data modeling and feature extraction. These models hold greater promise for fundamental problems in network optimization compared to traditional machine learning methods. Discriminative deep learning often falls short due to its single-step input-output mapping and lack of global awareness of the solution space, especially given the complexity of network optimization's objective functions. In contrast, generative diffusion models can consider a broader range of solutions and exhibit stronger generalization by learning parameters that describe the distribution of the underlying solution space, with higher probabilities assigned to better solutions. We propose a new framework Diffusion Model-based Solution Generation (DiffSG), which leverages the intrinsic distribution learning capabilities of generative diffusion models to learn high-quality solution distributions based on given inputs. The optimal solution within this distribution is highly probable, allowing it to be effectively reached through repeated sampling. We validate the performance of DiffSG on several typical network optimization problems, including mixed-integer non-linear programming, convex optimization, and hierarchical non-convex optimization. Our results demonstrate that DiffSG outperforms existing baseline methods not only on in-domain inputs but also on out-of-domain inputs. In summary, we demonstrate the potential of generative diffusion models in tackling complex network optimization problems and outline a promising path for their broader application in the communication community. Our code is available at https://github.com/qiyu3816/DiffSG.
title DiffSG: A Generative Solver for Network Optimization with Diffusion Model
topic Networking and Internet Architecture
Machine Learning
url https://arxiv.org/abs/2408.06701