How to get a better model

When I use gcn or graphsage model in dgl/example to train on my dataset for node dichotomy semi-supervised training, I find the validation/test set result is not stable and I think it’s because of not enough training . I set a very big training epoch and find the validation/test set result even get worse with the loss down in a small value.
So is this because overfitting? I really confused about the loss and the accuracy is both down. Does it mean my dataset(mask label 0: mask label1 = 1:1 ) not so good ? Or should I change a loss function(now is torch.nn.CrossEntropyLoss())?

In addition to the training loss and test accuracy, can you also plot curves for training accuracies and validation accuracies? If the training accuracy increases while the validation/test accuracies decrease, then this is likely an overfitting issue and you may need to tune things like early stopping, L2 regularization and dropout.

@mufeili Thanks for your reply. I have watched validation accuracies as the test accuracies. Both of them decreased. I have other training experience like a lightgbm model or a tensorflow cnn model, but neither of them have a such unstable performance. So any example for me to make a early stopping with the loss and accuracy is unstable?

This example may not be exactly what you want, but does contain a part for early stopping. Basically we check if the validation accuracy no longer improves for a pre-specified number of epochs. If so, early stopping will be performed. For test evaluation, we load model saved which has achieved the highest validation accuracy during training.

@mufeili Follow your advice. I use and to train,the result as follows:

               precision    recall  f1-score   support

 class_0       0.89      0.94      0.91     42498
 class_1       0.32      0.19      0.24      6212

 accuracy                           0.85     48710
 macro avg      0.60      0.56      0.58     48710
weighted avg    0.82      0.85      0.83     48710

As you can see, the performance on class_1 is bad. I think this may caused by the imbanlance node of 0 and 1 on the graph (which 0:1:unknow = 425001 : 62133 : 2109839). Anyway to improve the performance?

See it the technique described in this article helps:

Hey @yang,

How do you generate those details? I need it to evaluate the model


Thanks for the pointer @mufeili,

However, I’m afraid I cannot access that article because I’m not a member.

Does this one work?

Thanks @mufeili. It does work and i have a question about it.
I believe I can do random over/under sampling to my giant graph that I want to train. And I think it will not cause any problem if i do node classification based on the features of nodes.

However, if i do node classification that consider the edge’s feature too and I do over sampling, how should I treat the new edges (it’s features and its connection) after I add more nodes by doing over sampling?

Thanks so much

If you are working on node classification, I think you can simply perform down sampling/up sampling by weighting the loss terms for different nodes in loss computation. In this way, you do not need to explicitly modify the graph topology or features.

thanks @mufeili,

by weighting the loss terms, do you mean modifying class CrossEntropyLoss() or another loss functions I use in training?

Do you have reference or example of doing such a thing?


Take a look at the pos_weight argument of PyTorch’s BCEWithLogitsLoss and see if it’s applicable here.