/MCBG

Source code of Malware Classification by Learning Semantic and Structural Features of Control Flow Graphs (TrustCom 2021)

Primary LanguagePython

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

Reference