Image-Based Malware Classification
Malware Mugshots
Three vision models vote on malware rendered as images, across 17 families. Now a first-author paper on arXiv.
My thesis. I turned malware files into pictures and taught three vision models to recognize the family on sight. They vote together and agree 94 percent of the time across 17 families. It became a first-author paper, now on arXiv.
- Python
- PyTorch
- OpenCV
- ViT
- Docker
- CUDA