无需手工特征,用消息迭代训练模型高效拆解复杂网络。
Learning Network Dismantling Without Handcrafted Inputs
- 引入注意力机制和消息迭代谱,自动提取网络结构特征。
- 在百万节点真实网络上超越现有方法,准确率显著提升。
- 适合需要高效识别关键节点的网络分析场景。
消息传递图神经网络在重要网络科学问题上取得突破,但其优异性能通常依赖手工设计的结构特征,增加了计算开销并引入偏差。本文通过引入注意力机制与消息迭代谱,并结合有效算法生成结构多样化的微小合成网络训练集,构建了表达能力强的消息传递框架。该框架被用于高效求解NP难的网络拆解问题(等价于关键节点识别),具有广泛实际应用价值。所提出的MIND模型仅在多样化合成网络上训练,即可泛化至包含数百万节点的未见真实网络,性能优于当前最先进方法。该模型的高效率与强泛化能力可拓展应用于多种复杂网络问题。
原文摘要 · Abstract (English)
The application of message-passing Graph Neural Networks has been a breakthrough for important network science problems. However, the competitive performance often relies on using handcrafted structural features as inputs, which increases computational cost and introduces bias into the otherwise purely data-driven network representations. Here, we eliminate the need for handcrafted features by introducing an attention mechanism and utilizing message-iteration profiles, in addition to an effective algorithmic approach to generate a structurally diverse training set of small synthetic networks. Thereby, we build an expressive message-passing framework and use it to efficiently solve the NP-hard problem of Network Dismantling, virtually equivalent to vital node identification, with significant real-world applications. Trained solely on diversified synthetic networks, our proposed model -- MIND: Message Iteration Network Dismantler -- generalizes to large, unseen real networks with millions of nodes, outperforming state-of-the-art network dismantling methods. Increased efficiency and generalizability of the proposed model can be leveraged beyond dismantling in a range of complex network problems.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。