Improved Phishing Detection Algorithms using Adversarial Autoencoder Synthesized Data
Hossein Shirazi, Shashika R. Muramudalige, Indrakshi Ray, Anura P. Jayasumana
Abstract
Malicious actors often use phishing attacks to compromise legitimate users' credentials. Machine learning is a promising approach for phishing detection. While the accuracy of machine learning algorithms is often dependent on the training data, very little attack data for training is available. We propose an approach for augmenting existing datasets that can be used by machine learning algorithms. We use an Adversarial Autoencoder (AAE) to generate samples that mimic the phishing websites and provide metrics to assess the quality of the generated samples. We test these samples against models trained with real-world data. Some of generated samples are able to evade existing detection model. We then use a portion of these samples in training. The new machine learning models are more robust and have higher accuracy. In other words, real-world phishing site data augmented with AAE synthesized data used for training the model is more effective for phishing detection.