用扩散模型生成高质量负样本,提升超边预测效果
Scalable and Effective Negative Sample Generation for Hyperedge Prediction
- 基于扩散模型生成负样本,通过边界感知损失迭代优化
- 在多个数据集上准确率超越现有方法,训练速度显著提升
- 适合需要高效处理大规模超图的推荐系统与社交网络研究
超边预测在理解社交网络、电商系统等复杂多实体交互中至关重要。传统方法因正负样本不平衡,难以生成高质量负样本。本文提出可扩展且高效的负样本生成框架SEHP,利用扩散模型解决该问题。SEHP采用边界感知损失函数,迭代优化负样本使其靠近决策边界,提升分类性能;通过采样正例构建子超图,实现可扩展批量处理;将子超图结构信息作为扩散过程的条件,有效捕捉全局模式。为提升效率,方法直接在隐空间操作,避免离散ID生成,大幅加速训练过程。大量实验表明,SEHP在准确率、效率和可扩展性上均优于现有方法,代表了超边预测技术的重要进展。代码已公开。
原文摘要 · Abstract (English)
Hyperedge prediction is crucial in hypergraph analysis for understanding complex multi-entity interactions in various web-based applications, including social networks and e-commerce systems. Traditional methods often face difficulties in generating high-quality negative samples due to the imbalance between positive and negative instances. To address this, we present the Scalable and Effective Negative Sample Generation for Hyperedge Prediction (SEHP) framework, which utilizes diffusion models to tackle these challenges. SEHP employs a boundary-aware loss function that iteratively refines negative samples, moving them closer to decision boundaries to improve classification performance. SEHP samples positive instances to form sub-hypergraphs for scalable batch processing. By using structural information from sub-hypergraphs as conditions within the diffusion process, SEHP effectively captures global patterns. To enhance efficiency, our approach operates directly in latent space, avoiding the need for discrete ID generation and resulting in significant speed improvements while preserving accuracy. Extensive experiments show that SEHP outperforms existing methods in accuracy, efficiency, and scalability, representing a substantial advancement in hyperedge prediction techniques. Our code is available here.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。