Submitted:
04 November 2025
Posted:
05 November 2025
You are already at the latest version
Abstract

Keywords:
1. Introduction
2. Related Work
3. Novelty and Contribution
- Contextual TTP Embeddings: Builds on semantic advances such as SBERT [10] to encode technique descriptions into dense vectors, overcoming limitations of static embeddings (e.g., Attck2Vec). This enables generalisation across vocabulary variants and more nuanced behavioural similarity.
- Heterogeneous GNN Attribution: Uses relation-specific message passing across APT, TTP, and CKC nodes, enabling multi-hop reasoning. Unlike Deepro [12] (focused on campaign detection) or IPAttributor [6] (infrastructure clustering), our approach supports actor-level classification with both semantic and procedural depth.
- Operational Readiness: Provides an automated graph pipeline that ingests unstructured reports, normalises them to ATT&CK/CKC, and delivers attribution in interpretable form. This reduces analyst workload and ensures applicability in real-world SOC environments.
4. Methodology
- Feature Extraction: contextualising TTPs with SBERT embeddings [10] and Cyber Kill Chain semantics [9] to capture both behavioural meaning and procedural role (Section 4.1).
- Graph Construction: assembling a tripartite heterogeneous graph of APTs, TTPs, and CKC stages, where relations encode observed usage, lifecycle mapping, and temporal sequence (Section 4.2).
- Classification: applying relation-aware message passing with GraphSAGE layers to learn actor-level embeddings and optimise attribution decisions (Section 4.4).
4.1. Feature Extraction
- Semantic meaning: Sentence-BERT (SBERT) embeddings are generated from the MITRE ATT&CK descriptions of TTPs, giving a 384-dimensional vector that captures semantic similarity between related behaviours (e.g., “credential dumping” and “password extraction”).
- Procedural context: Each TTP is also mapped to its corresponding Cyber Kill Chain (CKC) stage (e.g., Initial Access, Execution, Lateral Movement). A one-hot vector encodes the stage, providing temporal and functional grounding.
4.2. Graph Construction
- APT–TTP edges: link actors to techniques they have employed in real campaigns, as reported in CTI sources such as ThreatFox, MISP, and OTX.
- TTP–CKC edges: map each technique to its corresponding stage in the Cyber Kill Chain, providing procedural grounding and attack-context reasoning.
- Temporal TTP–TTP edges: preserve the observed order of technique usage within campaigns, supporting temporal reasoning during model training.

4.3. Model Input
4.4. Classification
5. Training and Evaluation
5.1. Evaluation Metrics
5.2. Architectural Comparison
6. Performance Analysis
6.1. Overall Performance
6.2. Representation Quality (t-SNE and Class-Wise )

6.3. Confusion Matrix Analysis
6.4. Classification Report
6.5. Learning Dynamics
6.6. Summary Metrics
7. Benchmarking Against Prior Work
8. Future Work and Recommendations
- Few-shot and Meta-learning: Future work should explore meta-learning approaches that allow the model to rapidly adapt to new or underrepresented APT groups with very few examples. This is particularly important for emerging adversaries, where labelled CTI is scarce.
- Adversarial Augmentation: Incorporating adversarial augmentation strategies can expose the model to perturbed or rephrased variants of TTPs during training. This would increase robustness against false-flag tactics and deceptive reporting, where actors deliberately alter their observable behaviour or language.
- Synthetic CTI Generation: Generative methods (e.g., large language models or GANs) could be used to create synthetic but realistic CTI samples for rare APT classes. This would mitigate the long-tail imbalance problem, improving stability and accuracy across minority classes.
- Multi-task Learning: Jointly predicting CKC stages and APT attribution may enable the model to capture richer temporal-procedural dynamics, further aligning with adversary lifecycle reasoning.
- Hybrid Architectures: Combining GNNs with sequential models such as LSTMs or Transformers could enhance cross-report temporal reasoning and capture evolving campaign traces that unfold across multiple CTI sources.
- Self-supervised Pretraining: Leveraging large unlabelled CTI corpora for contrastive or self-supervised training may improve embedding quality and resilience against sparse labelled datasets.
- Explainability and SOC Integration: Embedding-level visualisations (e.g., UMAP/t-SNE projections) and confusion-matrix dashboards should be integrated into SOC workflows, helping analysts interpret adversary clustering and attribution confidence in real time.
9. Conclusions
References
- Irshad, E.; Siddiqui, A.B. Cyber threat attribution using unstructured reports in cyber threat intelligence. Egyptian Informatics Journal 2023, 24, 43–59. [CrossRef]
- Xiao, N.; Lang, B.; Wang, T.; Chen, Y. APT-MMF: An advanced persistent threat actor attribution method based on multimodal and multilevel feature fusion. Computers & Security 2024, 144, 103960. [CrossRef]
- Ren, Y.; Xiao, Y.; Zhou, Y.; Zhang, Z.; Tian, Z. CSKG4APT: A cybersecurity knowledge graph for advanced persistent threat organization attribution. IEEE Transactions on Knowledge and Data Engineering 2023, 35, 5695–5709. [CrossRef]
- Hwang, S.; Kim, T.S. An exploratory study on artifacts for cyber attack attribution considering false flag: Using Delphi and AHP methods. IEEE Access 2023, 11, 74533–74544. [CrossRef]
- Goel, S.; Nussbaum, B. Attribution across cyber attack types: Network intrusions and information operations. IEEE Open Journal of the Communications Society 2021, 2, 942–953. [CrossRef]
- Xiang, X.; Liu, H.; Zeng, L.; Zhang, H.; Gu, Z. IPAttributor: Cyber attacker attribution with threat intelligence-enriched intrusion data. Mathematics 2024, 12, 1364. [CrossRef]
- Zhang, S.; Xue, X.; Su, X. DeepOP: A hybrid framework for MITRE ATT&CK sequence prediction via deep learning and ontology. Electronics 2025, 14, 257. [CrossRef]
- Rosenberg, I.; Sicard, G.; David, E. DeepAPT: Nation-state APT attribution using end-to-end deep neural networks. In Proceedings of the Artificial Neural Networks and Machine Learning – ICANN 2017, Cham, 2017; Vol. 10614, Lecture Notes in Computer Science, pp. 91–99. [CrossRef]
- Hutchins, E.M.; Cloppert, M.J.; Amin, R.M. Intelligence-Driven Computer Network Defense Informed by Analysis of Adversary Campaigns and Intrusion Kill Chains. Technical Report LM White Paper, Lockheed Martin, 2011.
- Reimers, N.; Gurevych, I. Sentence-BERT: Sentence Embeddings using Siamese BERT-Networks. In Proceedings of the Proceedings of the 2019 Conference on Empirical Methods in Natural Language Processing and the 9th International Joint Conference on Natural Language Processing (EMNLP-IJCNLP). Association for Computational Linguistics, 2019, pp. 3982–3992. [CrossRef]
- Xiao, F.; Chen, S.; Yang, J.; He, H.; Jiang, X.; Tan, X.; Jin, D. GRAIN: Graph neural network and reinforcement learning aided causality discovery for multi-step attack scenario reconstruction. Computers & Security 2025, 148, 104180. [CrossRef]
- Yan, N.; Zhu, H.; Zhang, J.; Peng, T.; Zhang, X.; Zhang, H.; Huang, T.; Lin, X.; Liu, S.; Liu, X. Deepro: Provenance-based APT campaigns detection via GNN. In Proceedings of the Proc. IEEE Int. Conf. Trust, Security and Privacy in Computing and Communications (TrustCom), 2022, pp. 747–758. [CrossRef]
- Hamilton, W.L.; Ying, R.; Leskovec, J. Inductive Representation Learning on Large Graphs. In Proceedings of the Advances in Neural Information Processing Systems (NeurIPS). Curran Associates, Inc., 2017, pp. 1024–1034.
- Gilmer, J.; Schoenholz, S.S.; Riley, P.F.; Vinyals, O.; Dahl, G.E. Neural Message Passing for Quantum Chemistry. In Proceedings of the Proceedings of the 34th International Conference on Machine Learning (ICML). PMLR, 2017, Vol. 70, Proceedings of Machine Learning Research, pp. 1263–1272.
- He, H.; Garcia, E.A. Learning from imbalanced data. IEEE Transactions on Knowledge and Data Engineering 2009, 21, 1263–1284. [CrossRef]
- Cui, Y.; Jia, M.; Lin, T.Y.; Song, Y.; Belongie, S. Class-Balanced Loss Based on Effective Number of Samples. In Proceedings of the CVPR, 2019.
- Kingma, D.P.; Ba, J. Adam: A Method for Stochastic Optimization. CoRR 2014, abs/1412.6980.








| Study | Limitations | How Our Work Responds |
|---|---|---|
| Irshad & Siddiqui [1] | Relies on static embeddings (Attck2Vec) and handcrafted features; weak adaptability to evolving TTP vocabularies. | Uses SBERT embeddings for contextual semantics, reducing vocabulary bias and capturing nuanced behavioural similarity. |
| Hwang & Kim [4] | Identifies false-flag risks but remains expert-driven and unscalable. | Moves beyond expert-driven analysis by leveraging SBERT embeddings and CKC-aware GNN reasoning, enabling scalable detection of false-flag operations and deceptive actor behaviours. |
| CSKG4APT [3] | Profile matching over large graphs; brittle against unseen or evolving behaviours. | Employs GNN reasoning with temporal and CKC semantics, generalising to novel actor strategies. |
| APT-MMF [2] | Multimodal GNN with attention, but lacks lifecycle grounding; actors with similar TTP sets remain indistinguishable. | Incorporates CKC-aware TTP embeddings to differentiate techniques by lifecycle stage. |
| DeepOP [7] and DeepAPT [8] | Learn temporal order but omit procedural role; weak interpretability. | Adds CKC stage semantics to temporal modelling for operational interpretability and improved attribution. |
| GRAIN [11] | Models heterogeneous relations but ignores CKC and temporal progression. | Adds CKC nodes and temporal TTP sequencing, resolving stage-order ambiguity. |
| Deepro [12] | Strong for campaign detection but not fine-grained attribution; lacks procedural features. | Extends GNN classification to actor-level with lifecycle-aware reasoning. |
| IPAttributor [6] | Effective at infra-level clustering (IPs/domains), but blind to behavioural semantics. | Operates at the operational layer (APT–TTP–CKC paths) for strategic attribution. |
| Goel & Nussbaum [5] | Broad policy analysis; lacks temporal or technical modelling. | Provides reproducible, data-driven attribution grounded in temporal and procedural CTI. |
| Model | Macro-F | Accuracy |
|---|---|---|
| LSTM (Temporal baseline) | 0.481 | 0.553 |
| HeteroGNN (GATConv) | 0.544 | 0.671 |
| HeteroGNN (GraphSAGE) | 0.842 | 0.850 |
| HeteroGNN (Hybrid: GAT+SAGE) | 0.596 | 0.611 |
| Metric | Value |
|---|---|
| Accuracy | 85.0% |
| Macro-F | 84.7% |
| Weighted-F | 84.2% |
| Model | Approach | Features | Graph Type | Metric(s) | Strengths |
|---|---|---|---|---|---|
| APT-MMF [2] | Triple-attention heterogeneous GNN | Text + Topology + Metapaths | Heterogeneous | Macro- | Multimodal fusion. |
| DeepOP [7] | Transformer + causal modelling | MITRE ATT&CK TTP sequences | Sequential | Strong path prediction. | |
| CSKG4APT [3] | BERT-based text modelling | CTI report sentences | – | CTI text relation extraction. | |
| GRAIN [11] | Heterogeneous GNN + embeddings | IoCs + semantic embeddings (FastText best) | Heterogeneous | Heterogeneous reasoning. | |
| This paper | SBERT GNN (GraphSAGE) | TTP + CKC + edges | Tripartite GNN | Macro- | High accuracy and lifecycle-aware attribution. |
Disclaimer/Publisher’s Note: The statements, opinions and data contained in all publications are solely those of the individual author(s) and contributor(s) and not of MDPI and/or the editor(s). MDPI and/or the editor(s) disclaim responsibility for any injury to people or property resulting from any ideas, methods, instructions or products referred to in the content. |
© 2025 by the authors. Licensee MDPI, Basel, Switzerland. This article is an open access article distributed under the terms and conditions of the Creative Commons Attribution (CC BY) license (http://creativecommons.org/licenses/by/4.0/).