arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2210.07500cs.SIcs.LG
收藏

ToupleGDD:一种基于深度强化学习的影响力最大化精细设计方案

ToupleGDD: A Fine-Designed Solution of Influence Maximization by Deep Reinforcement Learning

The University of Texas at Dallas(德克萨斯大学达拉斯分校);Beijing Normal University(北京师范大学);BNU-HKBU United International College(北京师范大学-香港浸会大学联合国际学院)

Tiantian Chen, Siwen Yan, Jianxiong Guo, Weili Wu

AI总结:

本文提出了一种新颖的端到端深度强化学习框架ToupleGDD,通过结合耦合图神经网络和双重深度Q网络解决影响力最大化问题,在小图训练大图测试的设置下展现出强大的泛化能力与优越性能。

摘要:

旨在从网络中选取一个具有最大影响力的小规模节点子集,影响力最大化(IM)问题已被广泛研究。由于给定种子集计算影响力传播是#P-hard的,包括启发式和近似算法在内的最先进方法面临理论保证、时间效率、泛化能力等巨大困难。这使其无法适应大规模网络和更复杂的应用。另一方面,随着深度强化学习(DRL)在人工智能等领域取得最新成果,许多工作致力于利用DRL解决组合优化问题。受此启发,本文提出了一种新颖的端到端DRL框架ToupleGDD来解决IM问题,该框架结合了三个耦合的图神经网络用于网络嵌入,以及双重深度Q网络用于参数学习。以往使用DRL解决IM问题的工作在整个网络的子图上训练模型,然后在整个图上测试,这使得其模型在不同网络间的性能不稳定。然而,我们的模型在几个随机生成的小图上以小预算进行训练,并在完全不同的网络上以各种大预算进行测试,能够获得非常接近IMM的结果,并在多个数据集上取得优于OPIM-C的结果,展现出强大的泛化能力。最后,我们在合成和现实数据集上进行了大量实验,实验结果证明了我们模型的有效性和优越性。

英文摘要:

Aiming at selecting a small subset of nodes with maximum influence on networks, the Influence Maximization (IM) problem has been extensively studied. Since it is #P-hard to compute the influence spread given a seed set, the state-of-the-art methods, including heuristic and approximation algorithms, faced with great difficulties such as theoretical guarantee, time efficiency, generalization, etc. This makes it unable to adapt to large-scale networks and more complex applications. On the other side, with the latest achievements of Deep Reinforcement Learning (DRL) in artificial intelligence and other fields, lots of works have been focused on exploiting DRL to solve combinatorial optimization problems. Inspired by this, we propose a novel end-to-end DRL framework, ToupleGDD, to address the IM problem in this paper, which incorporates three coupled graph neural networks for network embedding and double deep Q-networks for parameters learning. Previous efforts to solve IM problem with DRL trained their models on subgraphs of the whole network, and then tested on the whole graph, which makes the performance of their models unstable among different networks. However, our model is trained on several small randomly generated graphs with a small budget, and tested on completely different networks under various large budgets, which can obtain results very close to IMM and better results than OPIM-C on several datasets, and shows strong generalization ability. Finally, we conduct a large number of experiments on synthetic and realistic datasets, and experimental results prove the effectiveness and superiority of our model.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑