Malware Detection Using GCN

• This project was developed to evaluate the performance of Malware Detection using GCN.
• To enable GPU training and ease of use, we employed the PyTorch framework.

Requirements

• Python 3.9
• Pytorch 1.9.1+cu111
• Torchvision 0.10.1+cu111
• RTX 3080 GPU

pip install torch==1.9.1+cu111 torchvision==0.10.1+cu111 -f https://download.pytorch.org/whl/torch_stable.html

• Torch-geometric
• Torch-scatter 2.0.9
• Torch-sparse 0.6.12
• Torch-cluster 1.5.9
• Torch-spline-conf 1.2.1

pip install torch-scatter==2.0.9 -f https://pytorch-geometric.com/whl/torch-1.9.1+cu111.html
pip install torch-sparse==0.6.12 -f https://pytorch-geometric.com/whl/torch-1.9.1+cu111.html
pip install torch-cluster==1.5.9 -f https://pytorch-geometric.com/whl/torch-1.9.1+cu111.html
pip install torch-spline-conv==1.2.1 -f https://pytorch-geometric.com/whl/torch-1.9.1+cu111.html
pip install torch-geometric

Dataset

• TIANCHI: Alibaba Cloud Malware Detection Based On Behaviors

https://tianchi.aliyun.com/competition/entrance/231694/information