Robust ensemble machine learning model for filtering phishing URLs: Expandable Random Gradient Stacked Voting Classifier (ERG-SVC)

R. D.P.L. Indrasiri, Malka N. Halgamuge, Azeem Mohammad

Research output: Contribution to journalArticlepeer-review

1 Citation (Scopus)
2 Downloads (Pure)

Abstract

As cyber-attacks grow fast and complicated, the cybersecurity industry faces challenges to utilize state-of-the-art technology and strategies to battle the consistently present malicious threats. Phishing is a sort of social engineering attack produced technically and classified as identity theft and complicated attack vectors to steal information of internet users. In this perspective, our main objective of this study is to propose a unique, robust ensemble machine learning model architecture that provides the highest prediction accuracy with a low error rate while proposing few other robust machine learning models. Both supervised and unsupervised techniques were used for the detection process. For our experiments, seven classification algorithms, one clustering algorithm, two ensemble techniques, and two large standard legitimate datasets with 73,575 URLs and 100,000 URLs were used. Two test modes (percentage split, K-Fold cross-validation) were utilized for conducting experiments and final predictions. Mechanisms were developed to (i) identify the best N, which is the optimal heuristic-based threshold value for splitting words into subwords for each classifier, (ii) tune hyperparameters for each classifier to specify the best parameter combination, (iii) select prominent features using various feature selection techniques, (iv) propose a robust ensemble model (classifier) called the Expandable Random Gradient Stacked Voting Classifier (ERG-SVC) utilizing a voting classifier along with a model architecture, (v) analyze possible clusters of the dataset using k-means clustering, (vi) thoroughly analyze the gradient boost classifier (GB) with respect to utilizing the “criterion" parameter with the Mean Absolute Error (MAE), Mean Squared Error (MSE), and Friendman_MSE, and(vii) propose a lightweight preprocessor to reduce computational cost and preprocessing time. Initial experiments were carried out with 46 features; the number of features was reduced to 22 after the experiments. The results show that the GB classifier outperformed with the least number of NLP based features by achieving a 98.118% prediction accuracy. Furthermore, our stacking ensemble model and proposed voting ensemble model (ERG-SVC) outperformed other tested approaches and yielded reliable prediction accuracy results in detecting malicious URLs at rates of 98.23% and 98.27%, respectively.

Original languageEnglish
Pages (from-to)150142-150161
Number of pages20
JournalIEEE Access
Volume9
DOIs
Publication statusPublished - 01 Nov 2021

Fingerprint

Dive into the research topics of 'Robust ensemble machine learning model for filtering phishing URLs: Expandable Random Gradient Stacked Voting Classifier (ERG-SVC)'. Together they form a unique fingerprint.

Cite this