Improving Network Representation Learning via Dynamic Random Walk, Self-Attention and Vertex Attributes-Driven Laplacian Space Optimization
Open Access
Abstract
Network data analysis is a crucial method for mining complicated object interactions. In recent years, random walk and neural-language-model-based network representation learning (NRL) approaches have been widely used for network data analysis. However, these NRL approaches suffer from the following deficiencies: firstly, because the random walk procedure is based on symmetric node similarity and fixed probability distribution, the sampled vertices’ sequences may lose local community structure information; secondly, because the feature extraction capacity of the shallow neural language model is limited, they can only extract the local structural features of networks; and thirdly, these approaches require specially designed mechanisms for different downstream tasks to integrate vertex attributes of various types. We conducted an in-depth investigation to address the aforementioned issues and propose a novel general NRL framework called dynamic structure and vertex attribute fusion network embedding, which firstly defines an asymmetric similarity and h-hop dynamic random walk strategy to guide the random walk process to preserve the network’s local community structure in walked vertex sequences. Next, we train a self-attention-based sequence prediction model on the walked vertex sequences to simultaneously learn the vertices’ local and global structural features. Finally, we introduce an attributes-driven Laplacian space optimization to converge the process of structural feature extraction and attribute feature extraction. The proposed approach is exhaustively evaluated by means of node visualization and classification on multiple benchmark datasets, and achieves superior results compared to baseline approaches.Keywords
Funding Information
- National Key R&D Program of China (2017YFC0907 505)
This publication has 25 references indexed in Scilit:
- Homophily, Structure, and Content Augmented Network Representation Learning2016
- Structural Deep Network Embedding2016
- node2vec2016
- Discriminative Deep Random Walk for Network Classification2016
- Robust Vertex ClassificationIEEE Transactions on Pattern Analysis and Machine Intelligence, 2015
- LINE2015
- Extreme Learning Machine for Multilayer PerceptronIEEE Transactions on Neural Networks and Learning Systems, 2015
- DeepWalk2014
- Inductive matrix completion for predicting gene–disease associationsBioinformatics, 2014
- Laplacian Eigenmaps for Dimensionality Reduction and Data RepresentationNeural Computation, 2003