Improving spam email classification accuracy using ensemble techniques: a stacking approach

被引:7
作者
Adnan, Muhammad [1 ]
Imam, Muhammad Osama [2 ]
Javed, Muhammad Furqan [2 ]
Murtza, Iqbal [2 ]
机构
[1] UiT Arctic Univ Norway, Dept Technol & Safety, Tromso, Norway
[2] Air Univ Islamabad, Fac Comp & AI, Islamabad, Pakistan
关键词
Spam; Email; Classification; Machine learning; Ensemble; Stacking method;
D O I
10.1007/s10207-023-00756-1
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Spam emails pose a substantial cybersecurity danger, necessitating accurate classification to reduce unwanted messages and mitigate risks. This study focuses on enhancing spam email classification accuracy using stacking ensemble machine learning techniques. We trained and tested five classifiers: logistic regression, decision tree, K-nearest neighbors (KNN), Gaussian naive Bayes and AdaBoost. To address overfitting, two distinct datasets of spam emails were aggregated and balanced. Evaluating individual classifiers based on recall, precision and F1 score metrics revealed AdaBoost as the top performer. Considering evolving spam technology and new message types challenging traditional approaches, we propose a stacking method. By combining predictions from multiple base models, the stacking method aims to improve classification accuracy. The results demonstrate superior performance of the stacking method with the highest accuracy (98.8%), recall (98.8%) and F1 score (98.9%) among tested methods. Additional experiments validated our approach by varying dataset sizes and testing different classifier combinations. Our study presents an innovative combination of classifiers that significantly improves accuracy, contributing to the growing body of research on stacking techniques. Moreover, we compare classifier performances using a unique combination of two datasets, highlighting the potential of ensemble techniques, specifically stacking, in enhancing spam email classification accuracy. The implications extend beyond spam classification systems, offering insights applicable to other classification tasks. Continued research on emerging spam techniques is vital to ensure long-term effectiveness.
引用
收藏
页码:505 / 517
页数:13
相关论文
共 37 条
[1]  
Agarwal D. K., 2016, Int J Comput Appl, V136, P16, DOI 10.5120/ijca2016908395
[2]  
[Anonymous], 2022, APACHE SPAM ASSASSIN
[3]  
Awad W. A., 2011, International Journal of Computer Science & Information Technology, V3, P173, DOI 10.5121/ijcsit.2011.3112
[4]  
Cohen W., 2009, ENRON EMAIL DATASET
[5]   Spam filtering using a logistic regression model trained by an artificial bee colony algorithm [J].
Dedeturk, Bilge Kagan ;
Akay, Bahriye .
APPLIED SOFT COMPUTING, 2020, 91
[6]  
El-Kareem A., 2017, J AL AZHAR U ENG SEC, V12, P1242, DOI [10.21608/auej.2017.19151, DOI 10.21608/AUEJ.2017.19151]
[7]   A decision-theoretic generalization of on-line learning and an application to boosting [J].
Freund, Y ;
Schapire, RE .
JOURNAL OF COMPUTER AND SYSTEM SCIENCES, 1997, 55 (01) :119-139
[8]  
Frias-Blanco Isvani., 2016, P 31 ANN ACM S APPL, P929, DOI DOI 10.1145/2851613.2851655
[9]   Efficient Model for Coronary Artery Disease Diagnosis: A Comparative Study of Several Machine Learning Algorithms [J].
Garavand, Ali ;
Salehnasab, Cirruse ;
Behmanesh, Ali ;
Aslani, Nasim ;
Zadeh, Amin Hassan ;
Ghaderzadeh, Mustafa .
JOURNAL OF HEALTHCARE ENGINEERING, 2022, 2022
[10]   X-Ray Equipped with Artificial Intelligence: Changing the COVID-19 Diagnostic Paradigm during the Pandemic [J].
Ghaderzadeh, Mustafa ;
Aria, Mehrad ;
Asadi, Farkhondeh .
BIOMED RESEARCH INTERNATIONAL, 2021, 2021