REFINE: Random RangE FInder for Network Embedding

被引:8
作者
Zhu, Hao [1 ,2 ]
Koniusz, Piotr [1 ,2 ]
机构
[1] Australian Natl Univ, Canberra, Australia
[2] Data61 CSIRO, Epping, Australia
来源
PROCEEDINGS OF THE 30TH ACM INTERNATIONAL CONFERENCE ON INFORMATION & KNOWLEDGE MANAGEMENT, CIKM 2021 | 2021年
关键词
Network Embedding; Low-Rank Approximation;
D O I
10.1145/3459637.3482168
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Network embedding approaches have recently attracted considerable interest as they learn low-dimensional vector representations of nodes. Embeddings based on the matrix factorization are effective but they are usually computationally expensive due to the eigen-decomposition step. In this paper, we propose a Random RangE FInder based Network Embedding (REFINE) algorithm, which can perform embedding on one million of nodes (YouTube) within 30 seconds in a single thread. REFINE is 10x faster than ProNE, which is 10 - 400x faster than other methods such as LINE, DeepWalk, Node2Vec, GraRep, and Hope. Firstly, we formulate our network embedding approach as a skip-gram model, but with an orthogonal constraint, and we reformulate it into the matrix factorization problem. Instead of using randomized tSVD (truncated SVD) as other methods, we employ the Randomized Blocked QR decomposition to obtain the node representation fast. Moreover, we design a simple but efficient spectral filter for network enhancement to obtain higher-order information for node representation. Experimental results prove that REFINE is very efficient on datasets of different sizes (from thousand to million of nodes and edges) for node classification, while enjoying a good performance.
引用
收藏
页码:3682 / 3686
页数:5
相关论文
共 32 条
[1]  
[Anonymous], 2017, ARXIV170205764
[2]  
[Anonymous], 2020, Uncertainty in Artificial Intelligence
[3]   LouvainNE: Hierarchical Louvain Method for High Quality and Scalable Network Embedding [J].
Bhowmick, Ayan Kumar ;
Meneni, Koushik ;
Danisch, Maximilien ;
Guillaume, Jean-Loup ;
Mitra, Bivas .
PROCEEDINGS OF THE 13TH INTERNATIONAL CONFERENCE ON WEB SEARCH AND DATA MINING (WSDM '20), 2020, :43-51
[4]  
Chen HC, 2018, AAAI CONF ARTIF INTE, P2127
[5]   A Survey on Network Embedding [J].
Cui, Peng ;
Wang, Xiao ;
Pei, Jian ;
Zhu, Wenwu .
IEEE TRANSACTIONS ON KNOWLEDGE AND DATA ENGINEERING, 2019, 31 (05) :833-852
[6]   node2vec: Scalable Feature Learning for Networks [J].
Grover, Aditya ;
Leskovec, Jure .
KDD'16: PROCEEDINGS OF THE 22ND ACM SIGKDD INTERNATIONAL CONFERENCE ON KNOWLEDGE DISCOVERY AND DATA MINING, 2016, :855-864
[7]   Finding Structure with Randomness: Probabilistic Algorithms for Constructing Approximate Matrix Decompositions [J].
Halko, N. ;
Martinsson, P. G. ;
Tropp, J. A. .
SIAM REVIEW, 2011, 53 (02) :217-288
[8]  
Klicpera Johannes, 2019, ARXIV191105485
[9]  
Koniusz P., 2020, IEEE Trans. Pattern Anal. Mach. Intell.
[10]  
Liang B, 2015, INT CONF SOFTW ENG, P894, DOI 10.1109/ICSESS.2015.7339198