kth.sePublications KTH
Change search
Link to record
Permanent link

Direct link
Publications (10 of 156) Show all publications
Gökstorp, S. G. E., Katsikeas, S. & Johnson, P. (2026). Machine Learning for Cybersecurity: A Comprehensive Literature Review. ACM Computing Surveys, 58(9), Article ID 243.
Open this publication in new window or tab >>Machine Learning for Cybersecurity: A Comprehensive Literature Review
2026 (English)In: ACM Computing Surveys, ISSN 0360-0300, E-ISSN 1557-7341, Vol. 58, no 9, article id 243Article, review/survey (Refereed) Published
Abstract [en]

In order to gain an overview of the state of research using machine learning for applications in cybersecurity, we have carried out a quantitative analysis of published works using the Scopus database. In total, 81,082 articles in this area were scraped from Scopus, from which we created an article-citation graph with articles as nodes and 547,998 citations among them as edges. We used the Louvain method for community-finding to organize the articles into research areas based on citation patterns, resulting in 12 identified research areas: intrusion detection: classical machine learning, intrusion detection: deep learning, privacy-preserving machine learning, malware detection, biometrics, adversarial machine learning, steganalysis, neural cryptography, phishing, traffic classification, software vulnerabilities, and smart grid. The state of the whole field and of each research area is discussed in detail, including highly cited and seminal works in each area and the relative size of each area compared to each other. Compared to previous literature surveys on this topic this review is not limited to specific subtopics and organizes the field by grouping works based on research topics, giving a more comprehensive overview of the field and the areas it encompasses.

Place, publisher, year, edition, pages
Association for Computing Machinery (ACM), 2026
Keywords
Clustering, Cybersecurity, Systematic Literature Review
National Category
Computer Sciences
Identifiers
urn:nbn:se:kth:diva-379310 (URN)10.1145/3796543 (DOI)001721716700001 ()2-s2.0-105034488121 (Scopus ID)
Note

QC 20260416

Available from: 2026-04-16 Created: 2026-04-16 Last updated: 2026-04-17Bibliographically approved
Nyberg, J. & Johnson, P. (2026). Vejde: A Framework for Inductive Deep Reinforcement Learning Based on Factor Graph Color Refinement. Transactions on Machine Learning Research, 1
Open this publication in new window or tab >>Vejde: A Framework for Inductive Deep Reinforcement Learning Based on Factor Graph Color Refinement
2026 (English)In: Transactions on Machine Learning Research, ISSN 2835-8856, Vol. 1Article in journal (Refereed) Published
Abstract [en]

We present and evaluate Vejde; a framework which combines data abstraction, graphneural networks and reinforcement learning to produce inductive policy functions for decisionproblems with richly structured states, such as object classes and relations. MDP statesare represented as data bases of facts about entities, and Vejde converts each state to abipartite graph, which is mapped to latent states through neural message passing. Thefactored representation of both states and actions allows Vejde agents to handle problemsof varying size and structure. We tested Vejde agents on eight problem domains defined inRDDL, with ten problem instances each, where policies were trained using both supervisedand reinforcement learning. To test policy generalization, we separate problem instances intwo sets, one for training and the other solely for testing. Test results on unseen instancesfor the Vejde agents were compared to MLP agents trained on each problem instance,as well as the online planning algorithm Prost. Our results show that Vejde policies inaverage generalize to the test instances without a significant loss in score. Additionally, theinductive agents received scores on unseen test instances that on average were close to theinstance-specific MLP agents.

Keywords
reinforcement learning, graph neural networks, relational modeling
National Category
Artificial Intelligence
Research subject
Computer Science
Identifiers
urn:nbn:se:kth:diva-376758 (URN)2-s2.0-105028411355 (Scopus ID)
Note

QC 20260216

Available from: 2026-02-16 Created: 2026-02-16 Last updated: 2026-04-24Bibliographically approved
Gökstorp, S. G. E., Nyberg, J., Kim, Y., Johnson, P. & Dán, G. (2024). Anomaly Detection in Security Logs using Sequence Modeling. In: Proceedings of IEEE/IFIP Network Operations and Management Symposium 2024, NOMS 2024: . Paper presented at 2024 IEEE/IFIP Network Operations and Management Symposium, NOMS 2024, Seoul, Korea, May 6 2024 - May 10 2024. Institute of Electrical and Electronics Engineers (IEEE)
Open this publication in new window or tab >>Anomaly Detection in Security Logs using Sequence Modeling
Show others...
2024 (English)In: Proceedings of IEEE/IFIP Network Operations and Management Symposium 2024, NOMS 2024, Institute of Electrical and Electronics Engineers (IEEE) , 2024Conference paper, Published paper (Refereed)
Abstract [en]

As cyberattacks are becoming more sophisticated, automated activity logging and anomaly detection are becoming important tools for defending computer systems. Recent deep learning-based approaches have demonstrated promising results in cybersecurity contexts, typically using supervised learning combined with large amounts of labeled data. Self-supervised learning has seen growing interest as a method of training models because it does not require labeled training data, which can be difficult and expensive to collect. However, existing self-supervised approaches to anomaly detection in user authentication logs either suffer from low precision or rely on large pre-trained natural language models. This makes them slow and expensive both during training and inference. Building on previous works, we therefore propose an end-to-end trained self-supervised transformer-based sequence model for anomaly detection in user authentication events. Thanks in part to an adapted masked-language modeling (MLM) learning task and domain knowledge-based improvements to the anomaly detection method, our proposed model outperforms previous long short-term memory (LSTM)-based approaches at detecting red-team activity in the "Comprehensive, Multi-Source Cyber-Security Events"authentication event dataset, improving the area under the receiver operating characteristic curve (AUC) from 0.9760 to 0.9989 and achieving an average precision of 0.0410. Our work presents the first application of end-to-end trained self-supervised transformer models to user authentication data in a cybersecurity context, and demonstrates the potential of transformer-based approaches for anomaly detection.

Place, publisher, year, edition, pages
Institute of Electrical and Electronics Engineers (IEEE), 2024
Keywords
anomaly detection, cybersecurity, LSTM, machine learning, misuse detection, sequence modeling, transformer
National Category
Computer Sciences Computer graphics and computer vision
Identifiers
urn:nbn:se:kth:diva-351008 (URN)10.1109/NOMS59830.2024.10575561 (DOI)001270140300136 ()2-s2.0-85198335240 (Scopus ID)
Conference
2024 IEEE/IFIP Network Operations and Management Symposium, NOMS 2024, Seoul, Korea, May 6 2024 - May 10 2024
Note

Part of ISBN 979-8-3503-2793-9

QC 20240724

Available from: 2024-07-24 Created: 2024-07-24 Last updated: 2026-02-16Bibliographically approved
Nyberg, J. & Johnson, P. (2024). Structural Generalization in Autonomous Cyber Incident Response with Message-Passing Neural Networks and Reinforcement Learning. In: 2024 IEEE INTERNATIONAL CONFERENCE ON CYBER SECURITY AND RESILIENCE, CSR: . Paper presented at 4th IEEE Annual International Conference on Cyber Security and Resilience (IEEE CSR), SEP 02-04, 2024, London, ENGLAND (pp. 282-289).
Open this publication in new window or tab >>Structural Generalization in Autonomous Cyber Incident Response with Message-Passing Neural Networks and Reinforcement Learning
2024 (English)In: 2024 IEEE INTERNATIONAL CONFERENCE ON CYBER SECURITY AND RESILIENCE, CSR, 2024, p. 282-289Conference paper, Published paper (Refereed)
Abstract [en]

We believe that agents for automated incident response based on machine learning need to handle changes in network structure. Computer networks are dynamic, and can naturally change in structure over time. Retraining agents for small network changes costs time and energy. We attempt to address this issue with an existing method of relational agent learning, where the relations between objects are assumed to remain consistent across problem instances. The state of the computer network is represented as a relational graph and encoded through a message passing neural network. The message passing neural network and an agent policy using the encoding are optimized end-to-end using reinforcement learning. We evaluate the approach on the second instance of the Cyber Autonomy Gym for Experimentation (CAGE 2), a cyber incident simulator that simulates attacks on an enterprise network. We create variants of the original network with different numbers of hosts and agents are tested without additional training on them. Our results show that agents using relational information are able to find solutions despite changes to the network, and can perform optimally in some instances. Agents using the default vector state representation perform better, but need to be specially trained on each network variant, demonstrating a trade-off between specialization and generalization.

Keywords
cyber security, reinforcement learning, graph learning, relational learning, generalization
National Category
Computer Sciences
Identifiers
urn:nbn:se:kth:diva-356458 (URN)10.1109/CSR61664.2024.10679456 (DOI)001327167900042 ()2-s2.0-85206191645 (Scopus ID)
Conference
4th IEEE Annual International Conference on Cyber Security and Resilience (IEEE CSR), SEP 02-04, 2024, London, ENGLAND
Note

QC 20241119

Part of ISBN 979-8-3503-7536-7

Available from: 2024-11-19 Created: 2024-11-19 Last updated: 2024-11-19Bibliographically approved
Widel, W., Hacks, S., Ekstedt, M., Johnson, P. & Lagerström, R. (2023). The meta attack language-a formal description. Computers & Security, 130, 103284, Article ID 103284.
Open this publication in new window or tab >>The meta attack language-a formal description
Show others...
2023 (English)In: Computers & Security, ISSN 0167-4048, E-ISSN 1872-6208, Vol. 130, p. 103284-, article id 103284Article in journal (Refereed) Published
Abstract [en]

Nowadays, IT infrastructures are involved in making innumerable aspects of our lives convenient, starting with water or energy distribution systems, and ending with e-commerce solutions and online banking services. In the worst case, cyberattacks on such infrastructures can paralyze whole states and lead to losses in terms of both human lives and money.One of the approaches to increase security of IT infrastructures relies on modeling possible ways of compromising them by potential attackers. To facilitate creation and reusability of such models, domain specific languages (DSLs) can be created. Ideally, a user will employ a DSL for modeling their infrastruc-ture of interest, with the domain-specific threats and attack logic being already encoded in the DSL by the domain experts.The Meta Attack Language (MAL) has been introduced previously as a meta-DSL for development of security-oriented DSLs. In this work, we define formally the syntax and a semantics of MAL to ease a common understanding of MAL's functionalities and enable reference implementations on different tech-nical platforms. It's applicability for modeling and analysis of security of IT infrastructures is illustrated with an example.

Place, publisher, year, edition, pages
Elsevier BV, 2023
Keywords
Threat modeling, Attack simulation, Attack graphs, Domain specific language
National Category
Reliability and Maintenance Computer Systems
Identifiers
urn:nbn:se:kth:diva-329957 (URN)10.1016/j.cose.2023.103284 (DOI)001001451200001 ()2-s2.0-85156202278 (Scopus ID)
Note

QC 20230626

Available from: 2023-06-26 Created: 2023-06-26 Last updated: 2025-08-28Bibliographically approved
Engström, V., Johnson, P., Lagerström, R., Ringdahl, E. & Wällstedt, M. (2022). Automated Security Assessments of Amazon Web Service Environments. ACM Transactions on Privacy and Security, 26(2), 1-31, Article ID 20.
Open this publication in new window or tab >>Automated Security Assessments of Amazon Web Service Environments
Show others...
2022 (English)In: ACM Transactions on Privacy and Security, ISSN 2471-2566, Vol. 26, no 2, p. 1-31, article id 20Article in journal (Refereed) Published
Abstract [en]

Migrating enterprises and business capabilities to cloud platforms like Amazon Web Services (AWS) has become increasingly common. However, securing cloud operations, especially at large scales, can quickly become intractable. Customer-side issues such as service misconfigurations, data breaches, and insecure changes are prevalent. Furthermore, cloud-specific tactics and techniques paired with application vulnerabilities create a large and complex search space. Various solutions and modeling languages for cloud security assessments exist. However, no single one appeared sufficiently cloud-centered and holistic. Many also did not account for tactical security dimensions. This article, therefore, presents a domain-specific modeling language for AWS environments. When used to model AWS environments, manually or automatically, the language automatically constructs and traverses attack graphs to assess security. Assessments, therefore, require minimal security expertise from the user. The modeling language was primarily tested on four third-party AWS environments through securiCAD Vanguard, a commercial tool built around the AWS modeling language. The language was validated further by measuring performance on models provided by anonymous end users and a comparison with a similar open source assessment tool. As of March 2020, the modeling language could represent essential AWS structures, cloud tactics, and threats. However, the tests highlighted certain shortcomings. Data collection steps, such as planted credentials, and some missing tactics were obvious. Nevertheless, the issues covered by the DSL were already reminiscent of common issues with real-world precedents. Future additions to attacker tactics and addressing data collection should yield considerable improvements.

Place, publisher, year, edition, pages
Association for Computing Machinery (ACM), 2022
Keywords
Attack simulation, threat modeling, cloud security, enterprise modeling, automatic security assessment, attack graphs, domain-specific language
National Category
Computer Sciences
Research subject
Computer Science
Identifiers
urn:nbn:se:kth:diva-324723 (URN)10.1145/3570903 (DOI)000970870300010 ()2-s2.0-85154572778 (Scopus ID)
Note

QC 20230529

Available from: 2023-03-13 Created: 2023-03-13 Last updated: 2025-12-19Bibliographically approved
Nyberg, J., Johnson, P. & Mehes, A. (2022). Cyber threat response using reinforcement learning in graph-based attack simulations. In: Varga, P Granville, LZ Galis, A Godor, I Limam, N Chemouil, P Francois, J Pahl, M (Ed.), Proceedings of the IEEE/IFIP Network Operations and Management Symposium 2022: Network and Service Management in the Era of Cloudification, Softwarization and Artificial Intelligence, NOMS 2022. Paper presented at IEEE/IFIP Network Operations and Management Symposium, NOMS 2022, Budapest, 25 April 2022, through 29 April 2022. IEEE
Open this publication in new window or tab >>Cyber threat response using reinforcement learning in graph-based attack simulations
2022 (English)In: Proceedings of the IEEE/IFIP Network Operations and Management Symposium 2022: Network and Service Management in the Era of Cloudification, Softwarization and Artificial Intelligence, NOMS 2022 / [ed] Varga, P Granville, LZ Galis, A Godor, I Limam, N Chemouil, P Francois, J Pahl, M, IEEE, 2022Conference paper, Published paper (Refereed)
Abstract [en]

In this ongoing project we employ reinforcement learning in a simulation environment to learn policies for cyber defense. The environment is based on attack graphs produced using the Meta Attack Language, a modeling language used to assess the security of systems. Two RL algorithms are utilized to prevent a simulated attacker agent to reach a series of targets within attack graphs. The defensive agent has to make decisions based on the value of keeping assets enabled, or suffering the consequence of the attacker reaching its goal. The initial results are promising, and show that both algorithms are able to find distinct strategies for defense. However, further analysis is needed to evaluate policy quality, including the implementation of sensible baseline policies for comparison.

Place, publisher, year, edition, pages
IEEE, 2022
Series
IEEE IFIP Network Operations and Management Symposium, ISSN 1542-1201
Keywords
cyber-security, reinforcement learning, threat modeling, attack graph
National Category
Information Systems
Identifiers
urn:nbn:se:kth:diva-319058 (URN)10.1109/NOMS54207.2022.9789835 (DOI)000851572700090 ()2-s2.0-85133206723 (Scopus ID)
Conference
IEEE/IFIP Network Operations and Management Symposium, NOMS 2022, Budapest, 25 April 2022, through 29 April 2022
Note

Part of proceedings: ISBN 978-1-6654-0601-7

QC 20220926

Available from: 2022-09-26 Created: 2022-09-26 Last updated: 2023-01-16Bibliographically approved
Katsikeas, S., Johnson, P., Ekstedt, M. & Lagerström, R. (2021). Research communities in cyber security: A comprehensive literature review. Computer Science Review, 42, 100431-100431, Article ID 100431.
Open this publication in new window or tab >>Research communities in cyber security: A comprehensive literature review
2021 (English)In: Computer Science Review, ISSN 1574-0137, E-ISSN 1876-7745, Vol. 42, p. 100431-100431, article id 100431Article in journal (Refereed) Published
Place, publisher, year, edition, pages
Elsevier BV, 2021
Keywords
General Computer Science, Theoretical Computer Science
National Category
Electrical Engineering, Electronic Engineering, Information Engineering
Identifiers
urn:nbn:se:kth:diva-303119 (URN)10.1016/j.cosrev.2021.100431 (DOI)000707155400004 ()2-s2.0-85114841893 (Scopus ID)
Note

QC 20211103

Available from: 2021-10-06 Created: 2021-10-06 Last updated: 2025-12-16Bibliographically approved
Katsikeas, S., Hacks, S., Johnson, P., Ekstedt, M., Lagerström, R., Jacobsson, J., . . . Eliasson, P. (2020). An Attack Simulation Language for the IT Domain. In: Graphical Models for Security: 7th International Workshop, GraMSec 2020, Boston, MA, USA, June 22, 2020, Revised Selected Papers. Paper presented at 7th International Workshop on Graphical Models for Security, GramSec 2020; Boston; United States; 22 June 2020 through 22 June 2020 (pp. 67-86). Springer Nature, 12419
Open this publication in new window or tab >>An Attack Simulation Language for the IT Domain
Show others...
2020 (English)In: Graphical Models for Security: 7th International Workshop, GraMSec 2020, Boston, MA, USA, June 22, 2020, Revised Selected Papers, Springer Nature , 2020, Vol. 12419, p. 67-86Conference paper, Published paper (Refereed)
Abstract [en]

Cyber-attacks on IT infrastructures can have disastrous consequences for individuals, regions, as well as whole nations. In order to respond to these threats, the cyber security assessment of IT infrastructures can foster a higher degree of security and resilience against cyber-attacks. Therefore, the use of attack simulations based on system architecture models is proposed. To reduce the effort of creating new attack graphs for each system under assessment, domain-specific languages (DSLs) can be employed. DSLs codify the common attack logics of the considered domain. Previously, MAL (the Meta Attack Language) was proposed, which serves as a framework to develop DSLs and generate attack graphs for modeled infrastructures. In this article, we propose coreLang as a MAL-based DSL for modeling IT infrastructures and analyzing weaknesses related to known attacks. To model domain-specific attributes, we studied existing cyber-attacks to develop a comprehensive language, which was iteratively verified through a series of brainstorming sessions with domain modelers. Finally, this first version of the language was validated against known cyber-attack scenarios.

Place, publisher, year, edition, pages
Springer Nature, 2020
Series
Lecture Notes in Computer Science (including subseries Lecture Notes in Artificial Intelligence and Lecture Notes in Bioinformatics), ISSN 0302-9743 ; 12419
Keywords
Attack graphs, Attack simulation, Domain specific language, IT Infrastructure, Meta Attack Language, Threat modeling, Computer simulation languages, Crime, Digital subscriber lines, Graphic methods, Network security, Problem oriented languages, Attack graph, Brainstorming sessions, Cyber security, Cyber-attacks, Domain specific languages, IT infrastructures, Model domains, System architectures, Computer crime
National Category
Computer Systems
Identifiers
urn:nbn:se:kth:diva-290393 (URN)10.1007/978-3-030-62230-5_4 (DOI)2-s2.0-85097391406 (Scopus ID)
Conference
7th International Workshop on Graphical Models for Security, GramSec 2020; Boston; United States; 22 June 2020 through 22 June 2020
Note

QC 20210219

Available from: 2021-02-19 Created: 2021-02-19 Last updated: 2022-12-20Bibliographically approved
Kakouros, N., Johnson, P. & Lagerström, R. (2020). Detecting plagiarism in penetration testing education. In: : . Paper presented at Nordsec 2020, The 25th Nordic Conference on Secure IT Systems, November 23-24, Online.
Open this publication in new window or tab >>Detecting plagiarism in penetration testing education
2020 (English)Conference paper, Poster (with or without abstract) (Refereed)
Abstract [en]

Plagiarism in hands-on ethical hacking education is challeng-ing due to student work being delivered almost exclusively virtually viaactions on a cyber range. We are developing the Cheat Detection Sys-tem, a solution that uses system administration and two approaches oflog analysis to infer about the cheating behavior of students; one basedon canary detection and another based on traces of student actions whichwe have termed the intrusion deficit approach.

National Category
Computer Systems
Research subject
Computer Science
Identifiers
urn:nbn:se:kth:diva-286103 (URN)
Conference
Nordsec 2020, The 25th Nordic Conference on Secure IT Systems, November 23-24, Online
Note

QC 20201124

Available from: 2020-11-19 Created: 2020-11-19 Last updated: 2022-06-25Bibliographically approved
Organisations
Identifiers
ORCID iD: ORCID iD iconorcid.org/0000-0002-3293-1681

Search in DiVA

Show all publications