Deep learning approach to link weight prediction
Deep learning approach to link weight prediction
复制标题
DOI:
10.1109/ijcnn.2017.7966076
复制
发表时间:
2017-05
期刊:
影响因子:
--
通讯作者:
Yuchen Hou;L. Holder
中科院分区:
文献类型:
--
作者:
Yuchen Hou;L. Holder
Deep learning has been successful in various domains including image recognition, speech recognition and natural language processing. However, the research on its application in graph mining is still in an early stage. Here we present Model R, a neural network model created to provide a deep learning approach to link weight prediction problem. This model extracts knowledge of nodes from known links' weights and uses this knowledge to predict unknown links' weights. We demonstrate the power of Model R through experiments and compare it with stochastic block model and its derivatives. Model R shows that deep learning can be successfully applied to link weight prediction and it outperforms stochastic block model and its derivatives by up to 73% in terms of prediction accuracy. We anticipate this new approach to provide effective solutions to more graph mining tasks.