Abstract:Even though machine learning algorithms already play a significant role in data science, many current methods pose unrealistic assumptions on input data. The application of such methods is difficult due to incompatible data formats, or heterogeneous, hierarchical or entirely missing data fragments in the dataset. As a solution, we propose a versatile, unified framework called `HMill' for sample representation, model definition and training. We review in depth a multi-instance paradigm for machine learning that the framework builds on and extends. To theoretically justify the design of key components of HMill, we show an extension of the universal approximation theorem to the set of all functions realized by models implemented in the framework. The text also contains a detailed discussion on technicalities and performance improvements in our implementation, which is published for download under the MIT License. The main asset of the framework is its flexibility, which makes modelling of diverse real-world data sources with the same tool possible. Additionally to the standard setting in which a set of attributes is observed for each object individually, we explain how message-passing inference in graphs that represent whole systems of objects can be implemented in the framework. To support our claims, we solve three different problems from the cybersecurity domain using the framework. The first use case concerns IoT device identification from raw network observations. In the second problem, we study how malicious binary files can be classified using a snapshot of the operating system represented as a directed graph. The last provided example is a task of domain blacklist extension through modelling interactions between entities in the network. In all three problems, the solution based on the proposed framework achieves performance comparable to specialized approaches.

Malicious Internet Entity Detection Using Local Graph Inference

Malicious Internet Entity Detection Using Local Graph Inference

Mapping the Internet: Modelling Entity Interactions in Complex Heterogeneous Networks

Android Malware Detection using Large-scale Network Representation Learning

Detecting Unknown Threat Based on Continuous-Time Dynamic Heterogeneous Graph Network

MalGraph: Hierarchical Graph Neural Networks for Robust Windows Malware Detection

Discovering Malicious Signatures in Software from Structural Interactions

A Heterogeneous Graph Learning Model for Cyber-Attack Detection

Temporal graph-based approach for behavioural entity classification

Using Graph Theory for Improving Machine Learning-based Detection of Cyber Attacks

AISLE: Self-supervised Representation Learning for the Investigation of Advanced Persistent Threat

Symbolic analysis meets federated learning to enhance malware identifier

An Efficient DenseNet-Based Deep Learning Model for Malware Detection

Detecting Malicious Domains with Behavioral Modeling and Graph Embedding

Graph based Encrypted Malicious Traffic Detection with Hybrid Analysis of Multi-view Features

Towards Explainable Network Intrusion Detection using Large Language Models

Knowledge Graph Based Large Scale Network Security Threat Detection Techniques

GraphMal: A Network Malicious Traffic Identification Method Based on Graph Neural Network

Deep Learning Criminal Networks

Detection of Malicious Agents in Social Learning

ECNet: Robust Malicious Network Traffic Detection With Multi-View Feature and Confidence Mechanism