Hongbo Bo
Social influence prediction with train and test time augmentation for graph neural networks
Bo, Hongbo; McConville, Ryan; Hong, Jun; Liu, Weiru
Abstract
Data augmentation has been widely used in machine learning for natural language processing and computer vision tasks to improve model performance. However, little research has studied data augmentation on graph neural networks, particularly using augmentation at both train- and test-time. Inspired by the success of augmentation in other domains, we have designed a method for social influence prediction using graph neural networks with train- and test-time augmentation, which can effectively generate multiple augmented graphs for social networks by utilising a variational graph autoencoder in both scenarios. We have evaluated the performance of our method on predicting user influence on multiple social network datasets. Our experimental results show that our end-to-end approach, which jointly trains a graph autoencoder and social influence behaviour classification network, can outperform state-of-the-art approaches, demonstrating the effectiveness of train-and test-time augmentation on graph neural networks for social influence prediction. We observe that this is particularly effective on smaller graphs.
Presentation Conference Type | Conference Paper (published) |
---|---|
Conference Name | The International Joint Conference on Neural Networks 2021 (IJCNN 2021) |
Start Date | Jun 18, 2021 |
End Date | Jun 22, 2021 |
Acceptance Date | Apr 10, 2021 |
Online Publication Date | Sep 20, 2021 |
Publication Date | Sep 20, 2021 |
Deposit Date | Nov 5, 2021 |
Publisher | Institute of Electrical and Electronics Engineers (IEEE) |
Book Title | Proceedings of the International Joint Conference on Neural Networks 2021 (IJCNN 2021) |
ISBN | 9781665445979 |
DOI | https://doi.org/10.1109/IJCNN52387.2021.9533437 |
Keywords | Index Terms-graph neural networks; social network analysis; social influence analysis; augmentation |
Public URL | https://uwe-repository.worktribe.com/output/7450433 |
You might also like
A survey of location inference techniques on Twitter
(2015)
Journal Article
Privacy preserving record linkage in the presence of missing values
(2017)
Journal Article
A novel ensemble learning approach to unsupervised record linkage
(2017)
Journal Article
A collaborative multiagent framework based on online risk-aware planning and decision-making
(2017)
Journal Article
Downloadable Citations
About UWE Bristol Research Repository
Administrator e-mail: repository@uwe.ac.uk
This application uses the following open-source libraries:
SheetJS Community Edition
Apache License Version 2.0 (http://www.apache.org/licenses/)
PDF.js
Apache License Version 2.0 (http://www.apache.org/licenses/)
Font Awesome
SIL OFL 1.1 (http://scripts.sil.org/OFL)
MIT License (http://opensource.org/licenses/mit-license.html)
CC BY 3.0 ( http://creativecommons.org/licenses/by/3.0/)
Powered by Worktribe © 2024
Advanced Search