Gespeichert in:
Bibliographische Detailangaben
Hauptverfasser: Gaudenzi, Alessandro, Nodari, Lorenzo, Kaplan, Lance, Russo, Alessandra, Sensoy, Murat, Cerutti, Federico
Format: Preprint
Veröffentlicht: 2025
Schlagworte:
Online-Zugang:https://arxiv.org/abs/2508.00368
Tags: Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
_version_ 1866916875095506944
author Gaudenzi, Alessandro
Nodari, Lorenzo
Kaplan, Lance
Russo, Alessandra
Sensoy, Murat
Cerutti, Federico
author_facet Gaudenzi, Alessandro
Nodari, Lorenzo
Kaplan, Lance
Russo, Alessandra
Sensoy, Murat
Cerutti, Federico
contents Advanced Persistent Threats (APTs) represent a significant challenge in cybersecurity due to their prolonged, multi-stage nature and the sophistication of their operators. Traditional detection systems typically focus on identifying malicious activity in binary terms (benign or malicious) without accounting for the progression of an attack. However, effective response strategies depend on accurate inference of the attack's current stage, as countermeasures must be tailored to whether an adversary is in the early reconnaissance phase or actively conducting exploitation or exfiltration. This work addresses the problem of attack stage inference under uncertainty, with a focus on robustness to out-of-distribution (OOD) inputs. We propose a classification approach based on Evidential Deep Learning (EDL), which models predictive uncertainty by outputting parameters of a Dirichlet distribution over possible stages. This allows the system not only to predict the most likely stage of an attack but also to indicate when it is uncertain or the input lies outside the training distribution. Preliminary experiments in a simulated environment demonstrate that the proposed model can accurately infer the stage of an attack with calibrated confidence while effectively detecting OOD inputs, which may indicate changes in the attackers' tactics. These results support the feasibility of deploying uncertainty-aware models for staged threat detection in dynamic and adversarial environments.
format Preprint
id arxiv_https___arxiv_org_abs_2508_00368
institution arXiv
publishDate 2025
record_format arxiv
spellingShingle Preliminary Investigation into Uncertainty-Aware Attack Stage Classification
Gaudenzi, Alessandro
Nodari, Lorenzo
Kaplan, Lance
Russo, Alessandra
Sensoy, Murat
Cerutti, Federico
Cryptography and Security
Machine Learning
Advanced Persistent Threats (APTs) represent a significant challenge in cybersecurity due to their prolonged, multi-stage nature and the sophistication of their operators. Traditional detection systems typically focus on identifying malicious activity in binary terms (benign or malicious) without accounting for the progression of an attack. However, effective response strategies depend on accurate inference of the attack's current stage, as countermeasures must be tailored to whether an adversary is in the early reconnaissance phase or actively conducting exploitation or exfiltration. This work addresses the problem of attack stage inference under uncertainty, with a focus on robustness to out-of-distribution (OOD) inputs. We propose a classification approach based on Evidential Deep Learning (EDL), which models predictive uncertainty by outputting parameters of a Dirichlet distribution over possible stages. This allows the system not only to predict the most likely stage of an attack but also to indicate when it is uncertain or the input lies outside the training distribution. Preliminary experiments in a simulated environment demonstrate that the proposed model can accurately infer the stage of an attack with calibrated confidence while effectively detecting OOD inputs, which may indicate changes in the attackers' tactics. These results support the feasibility of deploying uncertainty-aware models for staged threat detection in dynamic and adversarial environments.
title Preliminary Investigation into Uncertainty-Aware Attack Stage Classification
topic Cryptography and Security
Machine Learning
url https://arxiv.org/abs/2508.00368