Evaluation of Transformer-Based Large Language Models for Email Spam Detection Using BERT, Phi, and Gemma
In this paper, we study how LLMs based on the transformer architecture work and the possibility of adjusting these models to use only the body of email messages to classify them as spam or ham. The models studied are BERT, Gemma, and Phi. All of them underwent quantization stages, fine-tuning with a real dataset, and evaluation with metrics commonly used in binary classification problems. The Gemma model achieves over 99% accuracy in detecting spam, standing out as the best among the compared models.
Authors
- Kelton Augusto Pontara da Costa (ORCID: https://orcid.org/0000-0001-5458-3908)
- Juliana da Costa Feitosa (ORCID: https://orcid.org/0009-0005-6935-1022)
- José Remo Ferreira Brega (ORCID: https://orcid.org/0000-0002-2275-4722)
- Ana Clara C. Grassmann
Institutions
- Universidade Estadual Paulista (Unesp) (BR)
Publication Details
- Journal
- Statistics Optimization & Information Computing
- Published
- 2024-12-26
- DOI
- https://doi.org/10.19139/soic-2310-5070-2267
- Citations
- 1
- Primary Topic
- Spam and Phishing Detection
- Type
- article
- Field-Weighted Citation Impact
- 0.78
Funders
- Fundação de Amparo à Pesquisa do Estado de São Paulo