KillChainGraph: ML Framework for Predicting and Mapping ATT&CK Techniques
Singh, Chitraksh, Dhanraj, Monisha, Huang, Ken
–arXiv.org Artificial Intelligence
--The escalating complexity and volume of cyber-attacks demand proactive detection strategies that go beyond traditional rule-based systems. This paper presents a phase-aware, multi-model machine learning framework that emulates adversarial behavior across the seven phases of the Cyber Kill Chain using the MITRE A TT&CK Enterprise dataset. T ech-niques are semantically mapped to phases via A TT ACK-BERT, producing seven phase-specific datasets. We evaluate LightGBM, a custom Transformer encoder, fine-tuned BERT, and a Graph Neural Network (GNN), integrating their outputs through a weighted soft voting ensemble. Inter-phase dependencies are modeled using directed graphs to capture attacker movement from reconnaissance to objectives. The ensemble consistently achieved the highest scores, with F1-scores ranging from 97.47% to 99.83%, surpassing GNN performance (97.36% to 99.81%) by 0.03%-0.20% This graph-driven, ensemble-based approach enables interpretable attack path forecasting and strengthens proactive cyber defense.
arXiv.org Artificial Intelligence
Aug-26-2025
- Country:
- Asia > India
- Karnataka > Bengaluru (0.04)
- Maharashtra > Mumbai (0.04)
- Europe > Slovenia
- Drava > Municipality of Benedikt > Benedikt (0.04)
- North America > United States (0.28)
- Asia > India
- Genre:
- Research Report (0.82)
- Industry:
- Government > Military
- Cyberwarfare (0.49)
- Information Technology > Security & Privacy (1.00)
- Government > Military
- Technology: