Malware Classification by Learning Semantic and Structural Features of Control Flow Graphs

This is the source code of our paper accepted by TrustCom2021: Malware Classification by Learning Semantic and Structural Features of Control Flow Graphs.

Citation

@inproceedings{wu2021malware,
  title={Malware Classification by Learning Semantic and Structural Features of Control Flow Graphs},
  author={Bolun Wu, Yuanhang Xu, Futai Zou},
  booktitle={2021 IEEE 20th International Conference on Trust, Security and Privacy in Computing and Communications (TrustCom)},
  pages={540--547},
  year={2021},
  organization={IEEE}
}

Code Structure

.
├── README.md
└── src
    ├── bert_tidy     # BERT training code by Transformers
    ├── dataset       # parse BIG2015 to CFGs and BBs
    └── gnn           # gnn code

Dataset

• BIG2015

Reference

• GNN
  • GIN
  • DGCNN
• NLP
  • BERT
• Instruction Normalization
  • DEEPBINDIFF
  • SAFE
• Malware Classification
  • Classifying Malware Represented as Control Flow Graphs using Deep Graph Convolutional Neural Network