An improved transformer-based model for detecting phishing, spam and ham emails: A large language model approach

被引:8
作者
Jamal, Suhaima [1 ]
Wimmer, Hayden [1 ]
Sarker, Iqbal H. [2 ]
机构
[1] Georgia Southern Univ, Dept Informat Technol, Statesboro, GA USA
[2] Edith Cowan Univ, Ctr Securing Digital Futures, Perth, WA 6027, Australia
关键词
artificial intelligence; cyber security; DistilBERT; fine tuning; large language model; model optimization; phishing; RoBERTA; spam;
D O I
10.1002/spy2.402
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
Phishing and spam have been a cybersecurity threat with the majority of breaches resulting from these types of social engineering attacks. Therefore, detection has been a long-standing challenge for both academic and industry researcher. New and innovative approaches are required to keep up with the growing sophistication of threat actors. One such illumination which has vast potential are large language models (LLM). LLM emerged and already demonstrated their potential to transform society and provide new and innovative approaches to solve well-established challenges. Phishing and spam have caused financial hardships and lost time and resources to email users all over the world and frequently serve as an entry point for ransomware threat actors. While detection approaches exist, especially heuristic-based approaches, LLMs offer the potential to venture into a new unexplored area for understanding and solving this challenge. LLMs have rapidly altered the landscape from business, consumers, and throughout academia and demonstrate transformational potential to profoundly impact the society. Based on this, applying these new and innovative approaches to email detection is a rational next step in academic research. In this work, we present IPSDM, an improved phishing spam detection model based on fine-tuning the BERT family of models to specifically detect phishing and spam emails. We demonstrate our fine-tuned version, IPSDM, is able to better classify emails in both unbalanced and balanced datasets. Moreover, IPSDM consistently outperforms the baseline models in terms of classification accuracy, precision, recall, and F1-score, while concurrently mitigating overfitting concerns.
引用
收藏
页数:21
相关论文
共 47 条
  • [1] Spam Email Detection Using Deep Learning Techniques
    AbdulNabi, Isra'a
    Yaseen, Qussai
    [J]. 12TH INTERNATIONAL CONFERENCE ON AMBIENT SYSTEMS, NETWORKS AND TECHNOLOGIES (ANT) / THE 4TH INTERNATIONAL CONFERENCE ON EMERGING DATA AND INDUSTRY 4.0 (EDI40) / AFFILIATED WORKSHOPS, 2021, 184 : 853 - 858
  • [2] Transformer models for text-based emotion detection: a review of BERT-based approaches
    Acheampong, Francisca Adoma
    Nunoo-Mensah, Henry
    Chen, Wenyu
    [J]. ARTIFICIAL INTELLIGENCE REVIEW, 2021, 54 (08) : 5789 - 5829
  • [3] Phishing website detection: How effective are deep learning-based models and hyperparameter optimization
    Almousa, May
    Zhang, Tianyang
    Sarrafzadeh, Abdolhossein
    Anwar, Mohd
    [J]. SECURITY AND PRIVACY, 2022, 5 (06):
  • [4] Time efficient variants of Twin Extreme Learning Machine
    Anand, Pritam
    Bharti, Amisha
    Rastogi, Reshma
    [J]. INTELLIGENT SYSTEMS WITH APPLICATIONS, 2023, 17
  • [5] Araci D., 2019, FINBERT FINANCIAL SE
  • [6] Aslam M., 2024, INT J ADV ENG TECHNO, V1, P52
  • [7] Baaqeel H, 2020, INT ARAB CONF INF TE
  • [8] A comprehensive survey of AI-enabled phishing attacks detection techniques
    Basit, Abdul
    Zafar, Maham
    Liu, Xuan
    Javed, Abdul Rehman
    Jalil, Zunera
    Kifayat, Kashif
    [J]. TELECOMMUNICATION SYSTEMS, 2021, 76 (01) : 139 - 154
  • [9] Cabrera-Leon Y., 2016, INT JOINT C COMP INT
  • [10] A Performance Evaluation of Machine Learning-Based Streaming Spam Tweets Detection
    Chen, Chao
    Zhang, Jun
    Xie, Yi
    Xiang, Yang
    Zhou, Wanlei
    Hassan, Mohammad Mehedi
    AlElaiwi, Abdulhameed
    Alrubaian, Majed
    [J]. IEEE TRANSACTIONS ON COMPUTATIONAL SOCIAL SYSTEMS, 2015, 2 (03) : 65 - 76