论文

HyperFuse 提出快速超图节点嵌入方法,速度最高提升 179 倍

HyperFuse: Fast Self-Supervised Node Embeddings for Attributed Hypergraphs

精选理由

超图嵌入要训几百轮太慢了,HyperFuse 只要 100 轮,快 13 到 179 倍,精度还不输 TriCL 和 SE-HSSL,做图研究的可以看看。

HyperFuse 是一个无需标注的超图表征学习流程,先用 Banerjee 超图邻接矩阵的谱松弛求解结构坐标,再按成员稳定性给超边分配效用权重,最后用轻量编码器训练 100 轮。在 9 个公开超图数据集上与 TriCL、SE-HSSL、VilLain、HypeBoy 对比,平均每个数据集只需 8.7 秒,几何平均提速 13-179 倍。6 个下游分类器中 5 个取得最高平均准确率,且比 HypeBoy 快 13 倍、准确率高 2.1-4.1 个百分点。

原文 · arXiv cs.LG

HyperFuse: Fast Self-Supervised Node Embeddings for Attributed Hypergraphs

Self-supervised hypergraph representation learning can produce informative node embeddings, but existing methods often require deep encoders trained for hundreds of epochs, making embedding generation costly even for hypergraphs with a few thousand nodes. This limits applications requiring embeddings for many or evolving hypergraphs. We present HyperFuse, a label-free pipeline for fast hypergraph representation learning. HyperFuse (i) computes structural node coordinates by maximizing a spectral relaxation of hypergraph modularity using Banerjee's hypergraph adjacency and a matrix-free operator with cost linear in node-hyperedge incidences; (ii) constructs multi-scale feature summaries and assigns bounded utility weights to hyperedges based on member stability under feature and membership masking; and (iii) trains a lightweight utility-weighted hypergraph encoder for 100 epochs using an invariance-decorrelation objective. We compare HyperFuse with TriCL, SE-HSSL, VilLain, and HypeBoy on nine public hypergraphs using six downstream classifiers and k-means clustering. On the eight datasets where all methods completed, HyperFuse required 8.7 s per dataset on average, achieving 13-179x geometric-mean speed-ups over the baselines. It achieved the highest average accuracy with five of six classifiers, while classification and clustering performance was not significantly different from TriCL and SE-HSSL. Compared with HypeBoy, HyperFuse was 13x faster and 2.1-4.1 percentage points more accurate across all classifiers. HyperFuse provides a practical approach for fast, repeated hypergraph embedding generation.