Web Traffic Prediction via Temporal Graph Neural Network
Temporal Graph Neural Network or Temporal GNN is one of the variants of the GNN which handles the spatio-temporal data. The term "spatio-" refers to the nature of the graph that is closely related to the spatial relationship that exists in the image data (recall that an image is basically a graph), and the term "temporal" here indicates the data may be progressively changing in a sequence of time. In this project, the Chebysev GCNN+LSTM module and the Wiki Maths dataset are leveraged, which are provided by PyTorch Geometric Temporal. The complete Temporal GNN model contains the Chebysev GCNN+LSTM module, followed by a fully connected layer. Here, the model is trained to predict the daily user visits to Wikipedia's vital mathematics articles (represented by nodes/vertices). The graph's characteristic in the dataset is non-heterogenous and static. The details of the dataset can be seen here.
Experiment
To try out the experiment, the Google Colab notebook is provided.
Result
Quantitative Result
The quantitative performance of the model is given in the table below.
Metrics | Score |
---|---|
Loss | 0.452 |
MAPE | 4.110 |
MAE | 0.438 |
MSE | 0.519 |
Evaluation Metric Curve
Loss curve of the training and validation of the Temporal GNN model.
MAPE curve of the training and validation of the Temporal GNN model.
MAE curve of the training and validation of the Temporal GNN model.
MSE curve of the training and validation of the Temporal GNN model.
Qualitative Result
The figure below shows the qualitative result of the model.
The visitor prediction at one of the vital mathematics articles on Wikipedia.