This repository contains a neural network model that detects whether a given application is a malware or not

TLDR;

I created the same deep learning model in Tensorflow and PyTorch to identify whether a given application is malicious or not. Check the notebooks or the website for more

Problem Description

The widespread proliferation of Android devices has led to a concerning increase in malware threats, which pose significant risks to users' personal data and digital security. Malicious apps often disguise themselves as legitimate software, making them difficult to identify without specialized tools.

The provided dataset, contains some of the features that an application may have or services that it may be using. Given this input, I developed an AI model that tries to find a pattern between the features that may reveal whether an application is malicious or not

This was done specifically for educational purposes to learn better the two biggest ML frameworks: Tensorflow and PyTorch.

Approach

In the process of learning more about the implementation of neural networks and their respective frameworks, this model was created initially in Tensorflow and later on in PyTorch.

In general, there is no reason nor a significant difference between the two implementations. However, for educational purposes the two frameworks have been used

For more details and the source code, please click here