Hyperparameter Optimization for Deepfake Detection Improving CNN and Transformer Performance Using Bayesian Search
We provide an extensive study of hyperparameter optimization for CNN and Transformer models using the deepfake detection problem. We will use Optuna for Hyperparameter Tuning with Bayesian optimization in which we will tune some important parameters (dropout rate, learning rate, optimizer type). Experiments are performed with a real and fake image dataset extracted from video clips. We evaluate the optimized models in terms of accuracy, precision, recall, F1-score, and ROC-AUC metrics. With hyperparameter discovered by bayesian search, the Transformer model presta the best performance where it attained 94% accuracy and 0.94 AUC compare to optimized CNN for which we get 90% accuracy, 0.89 AUC. Moreover, we observe that Transformer has a better recall to identify fake videos (96% vs 83% for CNN), indicating its robustness in identifying deepfakes. In conclusion, the proposed framework for hyperparameter optimization can be a valuable addition to the growing body of research aimed at developing robust and effective models for deepfake detection.
Authors
- Khadeja M. Mohamed
- Ahmad Skaiky
- Zainab Mahdi (ORCID: https://orcid.org/0000-0001-5520-5626)
- Hanan Mahmood Shukur Ali
- Aymen Mohammed
Institutions
- University of Kirkuk (IQ)
- University of Baghdad (IQ)
- University of Technology - Iraq (IQ)
- Alsalam University College (IQ)
- Al-Farahidi University
- Al-Kunooze University College
- Nahrain University (IQ)
Publication Details
- Published
- 2025-04-05
- DOI
- https://doi.org/10.1109/icmi65310.2025.11141060
- Primary Topic
- Industrial Vision Systems and Defect Detection
- Type
- article
- Field-Weighted Citation Impact
- 0.00