Zero-Day Attack Detection Using Deep Learning with Focal Loss and Threshold Optimization
Authors: Umair Ashraf, Abdul Rehman, Aoun Muhammad, Umar Fayyaz, Sehrish Raza
Abstract
Zero-day attacks are attacks which exploit unknown vulnerabilities and they are difficult to find by signature-based methods. In this paper, a deep learning model will be presented for zero day attack identification on the UNSW-NB15 data. We propose a new architecture, based on Focal Loss, and is able to use it to address both extreme imbalance of classes and threshold optimization to reach optimal detection. This model proposed gave an accuracy of 81.26% accuracy. The ROC-AUC grew at 0.9039 with excellent discriminative ability of normal traffic and zero-day attacks. The accuracy of the system is equal (76.42%) and recall (70.87%) with very slight overfitting (7.97% error), which makes it able good enough to go over to the field. Comparative analysis shows our scheme on Focal Loss performs better than conventional Binary Cross Entropy with only 15,297 parameters.
