MalConv-Pytorch

A Pytorch implementation of MalConv

Desciprtion

This is the implementation of MalConv proposed in Malware Detection by Eating a Whole EXE.

Dependency

Please make sure each of them is installed with the correct version

• numpy
• pytorch (0.3.0.post4)
• pandas (0.20.3)

Setup

Preparing data

For the training data, please place PE files under data/train/ and build the label table for training set with each row being

    <File Name>, <Label>

where label = 1 refers to malware. Validation set should be handled in the same way.

Training

Run the following command for training progress

    python3 train.py <config_file_path> <random_seed>
    Example : python3 train.py config/example.yaml 123

Training Log & Checkpoint

Log file, prediction on validation set & Model checkpoint will be stored at the path specified in config file.

Parameters & Model Options

For parameters and options availible, please refer to config/example.yaml.