Abstract
Identifying and addressing anomalies in complex, distributed systems can be challenging for reliable execution of scientific workflows. We model these workflows as directed acyclic graphs (DAGs), where the nodes and edges of the DAGs represent jobs and their dependencies, respectively. We develop graph neural networks (GNNs) to learn patterns in the DAGs and to detect anomalies at the node (job) and graph (workflow) levels. We investigate workflow-specific GNN models that are trained on a particular workflow and workflow-agnostic GNN models that are trained across the workflows. Our GNN models, which incorporate both individual job features and topological information from the workflow, show improved accuracy and efficiency compared to conventional learning methods for detecting anomalies. While joint trained with multiple scientific workflows, our GNN models reached an accuracy more than 80% for workflow level and 75% for job level anomalies. In addition, we illustrate the importance of hyperparameter tuning method in our study that can significantly improve the metric(s) measure of evaluating the GNN models. Finally, we integrate explainable GNN methods to provide insights on job features in the workflow that cause an anomaly.
Keywords
Introduction
Distributed computing infrastructures are increasingly being leveraged for the execution of complex scientific applications. These execution platforms are heterogeneous, geographically distributed, and combine compute, network and storage resources from (often) disparate scientific facilities, forming a “superfacility,” to execute large-scale scientific workloads—from data collection on scientific instruments to moving the data, processing, storing, curating, disseminating, and visualizing the results. Being inherently distributed and heterogeneous, these superfacilities do not have centralized control, which results in limited visibility and thus limited understanding of how scientific workloads perform in the entirety of the execution platform. Hence, operators of these infrastructures and scientists who use them struggle to understand the performance of their systems and applications. The problem is exacerbated by anomalies and system performance degradations that might occur in such complex infrastructures, for example, network congestion, system I/O bottlenecks, and file system overload. Detecting and diagnosing these anomalies in the execution platforms, and how they propagate all the way up to the scientists’ workflows, remains a significant challenge.
Not only are the execution platforms complex but also the scientific applications and workloads running on these platforms are getting more sophisticated, with scientists designing and describing their computational campaigns using large-scale scientific workflows. Scientific workflows abound in domain sciences—weather modeling, climate science, seismology, geodesy, biology, astronomy, ecology, high-energy physics, and ocean sciences, just to name a few (Taylor et al., 2014). Workflows provide an appropriate abstraction to describe these campaigns, and workflow management systems like Pegasus (Deelman et al., 2015) automate and orchestrate the execution of these workflows on distributed, heterogeneous infrastructures, carefully managing the computations and data movements on behalf of the scientists, thereby improving the efficiency, robustness, and scientific productivity. Often, large-scale workflows are constituted of thousands of compute and/or data-intensive tasks, with complex data and control dependencies, which perform modeling, simulation, and data analysis by processing vast amounts of data. Given the scale and complexity, the scientific applications or the data movements in these workflows can exhibit anomalous behavior, and this convolution of complex workflows with distributed superfacilities used to execute them presents an even more significant challenge when it comes to anomaly detection and diagnosis.
Recent works in the literature have tackled the anomaly detection problem (Deelman et al., 2019; Rodriguez et al., 2018; Li and Song 2019; Singh et al., 2018; Papadimitriou et al., 2019). However, they have limited capability in correlating data from infrastructure sources, with each other, and with application-level data. Some methods are threshold based or rule based (Deelman et al., 2019; Stevens et al., 2020), which fail to uncover longitudinal patterns. Several machine learning (ML) and deep learning (DL) approaches have been applied more recently to address the anomaly detection problem. Although Li and Song (2019) use DL to forecast anomalies in high-energy physics jobs by leveraging minimal data during early parts of the job’s execution and achieved improvements of up to 14% in resource utilization, this work only considers application-level metrics. Similarly, Wang et al. (2020) also used only application-level metrics to apply clustering and decision trees to detect anomalies. Gaikwad et al. (2016) used autoregression techniques to detect I/O bottlenecks. These works, although promising, fail to take a comprehensive approach that holistically addresses correlations and/or leverages workflow-, task-, and infrastructure-level performance data. Krawczuk et al. (2021) utilized convolutional neural network (CNN) classifiers for anomaly detection in Gantt charts that represented the features of the workflow job. But, this work suffered the weakness that it did not use the topological information present in workflows. For specific IO anomaly, the Taufer (2021) discussed machine learning approaches to be used to prevent and mitigate IO contention in High Performance Computing (HPC) systems while dealing with IO bandwidth constraints.
We model the workflows as directed acyclic graphs (DAGs), wherein the nodes and edges represent jobs and their dependencies. We present a new approach that uses graph neural networks (GNNs) to holistically model the problem of anomaly detection in distributed scientific workflow executions. First, by leveraging the Pegasus workflow management system, we capture key performance metrics for different jobs in the workflow, both from the application tasks and the infrastructure. Using GNN-based approaches allows us to intrinsically capture the control and data dependencies between jobs inherent in the directed acyclic graph (DAG) representation of scientific workflows (Jin et al., 2022). We develop and train the GNN-based models of the workflows by using the data captured in distributed infrastructures, and use them to predict whether a workflow (graph) or a job (node) is anomalous or not. One advantage of our approach over related works in the literature is that our method is generalizable to a wide range of workflows because there are no restrictions on the input to the graphs (e.g., number of jobs in the workflow), and hence, our GNN approach can simultaneously use different workflows for training.
Compared to deep neural network (DNN) models for image and text data, GNNs are less generalizable because of the diversity of the data and domains that they come from. Consequently, it is not easy to use a GNN developed and trained for one domain (such as molecular property prediction) to a different domain (workflow anomaly prediction). Moreover, anomaly detection is relatively a harder problem than the regular classification problem. While it is feasible to develop a GNN relatively in a short time, improving the accuracy of GNN anomaly detection is typically a trial and error and expert-driven method. To that end, a promising approach to improve the performance of GNNs is hyperparameter tuning, where the crucial hyperparameters of the GNNs such as number of layers, learning rate, and batch size are automatically tuned. We leverage an automated hyperparameter tuning approach to tune the hyperparameters of the GNNs that we developed for workflow anomaly detection and demonstrate the importance and effectiveness compared to the baseline.
While GNN models provide the flexibility to model workflow graphs and detect anomalies, they cannot provide explanations about the predictions. The ability to explain anomaly predictions is critical for several reasons: (1) it improves the transparency and interpretability of the model, so that the insights derived from the explanation can be used to increase trust in the GNN model; (2) it can allow researchers and infrastructure operators to gain an understanding of the workflow characteristics, and identify and correct inaccurate predictions made by the models before deployment. Given a trained GNN model and a workflow graph, we seek to derive information about which node and edge features have significantly contributed to an anomaly. To that end, we will focus on explainable GNN methods that seek to provide explanations and attribute node and edge features to anomalies.
To summarize, the article makes the following main contributions: • We adopt a simple and efficient GNN-based approach to learn the node embedding from both the workflow job features and local dependencies. • We explore and evaluate the workflow anomalies from both the graph level (workflows) and the node level (jobs). • We build an anomaly detection model from different types of workflows simultaneously. • We adopt hyperparameter tuning to improve the basic graph neural networks. • Finally, we provide explainable results from the machine learning model to domain experts.
Workflow model
This work utilizes the Pegasus Workflow Management System (WMS) (Deelman et al., 2015), to describe workflows and manage their execution. Pegasus workflows have been deployed on various types of distributed and high-performance computing resources (e.g., NERSC, 2022 and OLCF, 2022), shared computing resources (e.g., XSEDE (Towns et al., 2014) and OSG (Pordes et al., 2007)), local clusters, and clouds.
Pegasus enables the development of workflows abstracted from the compute resources and from the input data, resulting in easily portable workflows. Given the specification of the compute resources and the location of the input data, Pegasus converts the abstract workflows into executable ones, tailored to be suitable for the provided infrastructure. For the rest of this article, we refer to “executable workflows” as “workflows.”
In Pegasus, workflows are represented as directed acyclic graphs (DAGs), where nodes represent jobs and edges represent sequential dependencies and data dependencies between the jobs. A job can only be submitted when all its dependencies have been met.
For a more formal definition, a workflow is described as a DAG • • •
Moreover, Pegasus collects provenance data and events during the execution of a workflow and associates them with the jobs that produced them. Pegasus Panorama extensions (Deelman et al., 2017; Papadimitriou and Deelman 2018; Papadimitriou et al., 2021) allow us to enhance the workflow DAG with features containing execution metrics and infrastructure statistics per job (v
i
). To characterize the end-to-end execution of the workflows, the DAG is enhanced with the following types of features: • • • •
In Listing 1 we present the list of features that we add as metadata to the workflow DAG, which are used to train the GNN models (Section 3).
Graph neural network model for anomaly detection
Our objective is to identify whether a workflow has anomalies, which can be caused by abnormalities in CPU or hard disk performance and can occur in any individual job (as discussed in Section 4). To accomplish this, we designed a graph neural network (GNN) model that uses workflows modeled as directed acyclic graphs (DAGs) with features as input and detects the presence of anomalies. We follow a supervised learning formulation, where we generate training data with normal and anomalous workflow runs and use the trained model to detect anomalies in workflows.
We developed the graph neural network using graph convolutional layers (GCNs, Welling and Kipf (2016)). Our GNN, shown in Figure 1, is composed of two modules: the GCN module, which generates the hidden layer embedding, and the multilayer perceptron (MLP) module, which detects the anomaly. We use the GNN for anomaly detection at both the job level (by framing it as a node classification problem) and the workflow level (by framing it as a graph classification problem). The GCN module is made up of two layers, each containing a GCN operation followed by a rectified linear unit (ReLU) activation function. The MLP module is then applied to the hidden embedding generated by the GCN module to calculate the probability of a workflow/job (graph/node) being anomalous or not. For workflow anomaly detection, a global average pooling (mean of the hidden embedding across all jobs) is applied prior to the MLP module. The aim of the mean pooling is to integrate information across jobs. In the job anomaly detection scenario, the MLP module is applied to each node and we evaluate whether each job is anomalous or not. Graph neural network architecture.
Listing 1: Features describing a job.
In this case, given a graph representation of a workflow
The GCN module extracts the embedding of the hidden node
Hyperparameter tuning
GNNs have a number of hyperparameters that need to be set prior to training. Hyperparameter search (HPS) is the process of systematically searching for the best combination of hyperparameters for a GNN model. Typically, this can be done through techniques such as grid search, random search, or Bayesian optimization (Jamieson and Talwalkar 2016), with the aim of achieving the best performance on the validation set. By carefully tuning the hyperparameters, it is possible to significantly improve the performance of GNNs on various tasks including anomaly detection via supervised classification.
In this article, we adopt DeepHyper (Balaprakash et al., 2018), an open-source scalable hyperparameter tuning software. It supports both single- and multi-objective optimization and also supports distributed computing, making it easy to scale up the computation. This is particularly useful in our setting when we have multiple metrics to consider for the imbalanced data, which will be discussed in Section 5.1. The hyperparameter tuning in DeepHyper seeks to find the best hyperparameter values such that the metric measure of the model can be improved. Given a candidate set of hyperparameters
DeepHyper adopts an asynchronous model-based search (AMBS) method that consists of sampling a number of hyperparameter configurations and progressively fitting a surrogate model over the input–output space until exhausting the user-defined maximum number of evaluations. The asynchronous aspect allows the search to avoid waiting for all the evaluation results before proceeding to the next iteration. As soon as an evaluation is finished, the data are used to retrain the surrogate model, which is then used to bias the search toward the promising configurations. The framework is designed to operate in the master–worker computational paradigm, where one master node fits the surrogate model and generates promising input configurations and worker nodes perform the computationally expensive evaluations and return the outputs to the manager node.
Explainable GNN models
Explanation of anomaly detection predictions from GNNs is more challenging than traditional machine learning models due to their high complexity and black-box nature. One of the key features of GNNs is that they take into account the relationships between nodes in the graph, unlike traditional neural networks that only process individual examples. This is achieved by applying a propagation step to the graph, which is used to update the embeddings of each node based on its neighbors.
GNNExplainer (Ying et al., 2019) is an open-source library for explaining GNN predictions. It aims to provide a unified interface for interpreting the decision-making process of GNN models, and also it provides various techniques for explaining the predictions of GNN models. Specific to the anomaly detection task, we focus on (1) node-level explanations, which allow users to understand how a specific node’s features and graph topology contributed to the final prediction; and (2) graph-level explanations, which allow users to understand how the graph structure as a whole contributed to the final prediction. This is indeed useful for identifying and quantifying the importance of features and nodes in the scientific workflow where the anomaly occurs.
For the node-level explanation, given a node v, the explanation tries to identify a subgraph G
S
⊆ G and the associated node features X
S
= {x
j
| x
i
∈ G
S
} that are important for the prediction of the trained GNN model. The GNNExplainer takes the mutual information (MI) as the measurement of the importance, formulated as
For the graph-level explanation, the explainer chooses a reference node v, and takes explanation G
S
(v) for reference node and aligns it to explanation of other nodes associated on the graph. More specifically, we extract the 1-hop subgraph from v and explain the nodes extracted from the topological structure, extending the formulation of equation (6) to
Experimental setup
Representative workflows
To evaluate our GNN approach, we use the following science workflows orchestrated by Pegasus.
1000 Genome Pegasus workflow.
CASA Nowcast Pegasus workflow.
CASA Wind Pegasus workflow.
Task clustering
Pegasus enables the clustering of workflow tasks into larger jobs via horizontal or label task clustering. This technique helps to optimize for the time spent waiting in the queue instead of doing computations, which is essential when a workflow has small short running tasks. With horizontal clustering, tasks on the same level are grouped together; with label clustering, Pegasus groups together tasks that carry the same label in their metadata. By employing the task clustering technique, the shape of the final executable workflow DAG changes, without changing the semantics of the workflow. And we can analyze the new DAG independently, as a new instance.
Executable workflows
In our experiments, we used the three workflows described in Section 4.1, but we invoked the workflows with different input sizes and different configuration settings to create six different executable workflow DAGs.
More specifically, we invoked the 1000 Genome workflow with two different input sizes (2 GBs vs 5 GBs). This resulted in a different number of nodes and edges in the two executable workflow instances, as seen in Table 1 and Table 2. • • Statistics of executable workflows—anomalies affecting all jobs. Statistics of executable workflows—anomalies affecting some jobs.
Additionally, we enabled horizontal clustering for the CASA Nowcast workflow and label clustering for the CASA Wind workflow. The clustering details are as follows. • • • •
The different sizes of the executable workflow DAGs for CASA Nowcast and CASA Wind workflows can be found in Table 1.
Data collection
To collect the data, we used the Pegasus Panorama extension (Deelman et al., 2017; Papadimitriou and Deelman, 2018), which offers advanced monitoring capabilities (Papadimitriou et al., 2021). It enables end-to-end online workflow execution monitoring and provides execution traces of the computational tasks, statistics for individual transfers, and infrastructure-related metrics, and stores the measurement data in an Elasticsearch instance (ELK, 2018). In this article, we collected two datasets, where in the first one anomalies are introduced to the entire workflow, while in the second one anomalies are introduced in some of the workflow jobs.
Dataset 1
For the first dataset, we executed the five workflows of Table 1 on the ExoGENI testbed (Baldin et al., 2016). We used seven virtual machines, with one submit node, five worker nodes, and one data node. The worker nodes were located within the same ExoGENI region, while the submit node and the data node were located in a different regions. Each virtual machine had four 2.2 GHz vCPUs, 10 GB RAM, and 75 GB storage. The connectivity between the two ExoGENI regions was established over a high-speed layer 2 VLAN, with 1 Gbps links for each node. To facilitate workflow execution, we configured our nodes with Pegasus and HTCondor. On the data node, we used a standard web server to enable file transfers over HTTP.
To introduce synthetic network and I/O anomalies, we used the Linux Traffic Control (TC) toolset (Hubert et al., 2002). TC is able to replicate network anomalies such as delay, packet loss, and jitter, by configuring the Linux kernel packet scheduler. Additionally, to reduce the performance of the worker nodes, we used the stress tool (Waterland, 2013), a simple workload generator that can impose a configurable amount of CPU, memory, I/O, and disk stress on the system.
In this set of experiments, the injected anomalies were affecting all worker nodes, and as a result all jobs of the workflow. During our data collection, we generated 6,000 traces of the above workflows for four main classes, as seen in Table 1. Examples of how these anomalies affect workflow execution can be found in our prior work (Papadimitriou et al., 2021). • • • •
Dataset 2
Dataset 2 differs from Dataset 1 in the following ways: (1) as summarized in Table 2, our second dataset is limited to the 1000 Genome workflow, which has been scaled up to more jobs (1000 Genome B); (2) the anomalies are now injected to some of the worker nodes, and, as a result, this affects only a subset of workflow jobs, but not all of them; and (3) as we will explain in the next paragraph, the experiments are conducted on Chameleon testbed (Keahey et al., 2020).
To generate the second dataset, we provisioned three bare-metal nodes on Chameleon testbed, with one submit node and two container executor nodes. By using Docker containers (Docker Inc. 2022), we virtualized the 2 container executor nodes to 20 workers (10 workers per node). The worker nodes were located within the same Chameleon region (Texas Advanced Computing Center—TACC), while the submit node was located at the University of Chicago region. Each worker node had 4 cores and 16 GB RAM, that were assigned to them via Docker’s runtime options (cpuset, memory). Connectivity between the two Chameleon regions was established over a high-speed layer 2 VLAN, with capped speed of 1 Gbps for each worker node. To facilitate the workflow execution, we configured our nodes with Pegasus and HTCondor, and the data was served by the submit node over HTTP.
In this dataset, to introduce synthetic anomalies, we used Docker’s runtime options to limit and shape the performance of the spawned worker containers. This way, we can configure the amount of CPU time a worker node (container) gets, and limit the average I/O each worker node can perform. These capabilities are supported via Linux’s control groups version 2 (cgroups v2) (Linux Kernel Organization 2023). This approach offers sufficient isolation among the workers and allows us to obtain reproducible results from each type of experiment. For this dataset, we generated 350 traces of the scaled up version of the 1000 Genome workflow for three main classes, as seen in Table 2. • • •
Evaluation of the GNN model for anomaly detection
Model setup and metrics
Graph-level classification.
Node-level classification.
We split the data into training, validation, and testing at 60%, 20%, and 20%, respectively. All the layers within the GNN architecture take the hidden dimension of 64 with bias terms and apply the cross-entropy loss to quantify the wellness of the trained model. For optimization, we employ Adam optimizer (Kingma and Ba, 2015) with a learning rate of 1e−3.
It is worth noting that our collected data are quite imbalanced, that is, the number of normal and anomaly samples differs significantly. Figure 5 shows the statistics of 1000 Genome B with partial anomalies for the binary classification problem. Instead of feeding the data into the training model, we explicitly specify the weights of the labels according to the number of samples in the training set. We adopt cross-entropy loss as the final prediction of probabilities for each label from outputs. Here, we introduce the weights as Imbalanced data (1000 Genome B with partial anomaly).
Graph-level anomaly detection
To identify abnormalities in the workflow, we used a GCN module to analyze the graph, followed by an MLP module to make probabilistic predictions for each graph. The results of the graph classification on the test set are shown in Table 3 for both binary and multi-label settings. Furthermore, in order to apply our approach to various types of workflow, we also tested the ability of a single model to predict anomalies in multiple workflows at once, which is labeled as “ALL” in the final row of the table.
We demonstrated the accuracies and loss values during training of a single model for the 1000 Genome A workflows in Figures 6(a, binary) and (b, multi-label). We presented the mean accuracy and loss values for each epoch during the 10 rounds of evaluations. To start with, a randomly generated GNN model, which is presented as the first epoch in the figures, can be improved during the backpropagation update on parameters of the GNN model. The results indicate that binary classification outperforms multi-label anomaly detection in detecting normal versus anomaly, achieving 10% to 20% higher accuracies across various workflows. This suggests that binary classification is more effective in detecting anomalies compared to identifying specific anomaly categories. Training for graph level anomaly detection (1000 Genome A).
Moreover, we investigated the rationale of the performance from the perspective of workflow structures. Figure 7 shows the relationship between the accuracy of the binary setting and the number of jobs in the workflows. Clearly, the accuracy is proportional to the workflow size; that is, with more complex structures, anomaly detection reaches higher accuracy. Even among the full anomaly workflows, the 1000 Genome A reached best and has more jobs involved. This is largely due to the intrinsic property of graph neural networks, where more complex structures help to aggregate from local neighbors through layer-wise propagation, overcoming the over-smoothing problems (Chen et al., 2020). Rationale behind accuracy versus the number of jobs in the workflow.
Node-level anomaly detection
Another important issue in science workflows is identifying abnormal jobs (nodes) within a workflow (graph). Instead of using a single label for each run in a graph-level approach, we assign labels to individual nodes by adapting them from the entire run. This means that the labels of the jobs are the same as the label of the corresponding run. We randomly select jobs from all runs to use as training, validation, and testing sets, allowing the model to be trained with information from a variety of labels.
Similar to the graph-level setting, we report the performance on the testing set in Table 4. “ALL” in the last row represents the performance of a single model that trains different workflows at the same time. We observe that the binary (normal vs anomaly) setting reaches better accuracy than the multi-label setting does, the same observation we had in graph-level anomaly detection. More specifically, the recall score, which is a widely used metric for imbalanced data for binary problems, reaches 0.777 even in the case of a single model that utilizes all workflows together. Figures 8(a) (a, binary) and (b, multi-label) show the accuracy and loss values on the training of a single model for the 1000 Genome A workflows in terms of node-level anomalies. Again, the results are reported with 10 repetitions. Training for node level anomaly detection (1000 Genome A).
Anomaly detection by category
Graph-level binary classification by anomaly categories.
Figures 9(a) and 9(b) show the confusion matrix, also known as the error matrix, to evaluate the multi-label classification for 1000 Genome A workflow with the single-model graph-level and node-level classifications, respectively. The value in each cell indicates the ratio of observations known to be in group i and predicted to be in group j, with row-wise summing being 1. The diagonal parts indicate the corrected predictions, and the higher, the better. These results also match our findings, given in Table 5: HDD-related anomalies achieve a higher score than the others do (on the diagonal). Confusion matrix of multi-label classification.
Anomaly detection for different anomaly levels
Graph-level binary classification by anomaly levels.
Model comparison
Performance comparison on 1000 Genome A.
As seen in the table, the GNNs outperform both the standard machine learning models and the Gantt chart representation of workflows, due to their ability to learn embeddings from both local structural information and node features. This is made possible by the inclusion of additional structural information from the workflow. It is worth noting that all the other hyperparameters were kept the same as in Section 5.1 and no fine-tuning was done. We will explicitly discuss the hyperparameter tuning to boost the performance of GNNs in next subsection.
Runtime (sec.) of training deep learning models.
Hyperparameter search
To improve the overall performance of GNNs in anomaly detection, we make use of DeepHyper, a framework that uses the Bayesian optimization method to search for the best hyperparameter settings. To demonstrate this approach, we apply it to the 1000 Genome A workflow with the task of binary node-level anomaly detection. As introduced in Section 3.1, our first target is the accuracy score, a single objective of the metric to measure the performance of our GNN model. We tune for the number of hidden dimensions in {16, 32, 64, 128, 256} and dropout rate in HPS for accuracy score.
Furthermore, considering that the data is quite imbalanced, we also tune the hyperparameter for both accuracy and AUC-ROC score. The latter computes the Compute Area Under the Receiver Operating Characteristic Curve (ROC AUC) from prediction scores, and it is typically used in imbalanced binary classification. Figure 11 shows the results after HPS with 500 model evaluations. Among all the results (black dots), the red ones represent the best choices after considering the trade-off between the accuracy and ROC-AUC scores. Both the single-objective and multi-objective HPS demonstrate that the performance of our GNN model can be improved effectively and efficiently. HPS for both accuracy and ROC-AUC scores.
Explanation of GNNs
To explain the results from GNNs for anomaly detection, we apply the GNNExplainer to demonstrate both node-level and graph-level explanations. We use 1000 Genome B to provide the explanation of prediction on a real workflow. Recall in Section 3.2, we take advantage of GNNExplainer, which optimizes an upper bound of mutual information to identify the importance of both structure and feature information. To start, we pick node 23 of the workflow, which serves as a transfer node. Figure 12 shows its position and the 1-hop subgraph in the workflow. Figure 13 demonstrates the comparison between the normal job and anomaly job selected randomly from our collected data. Each column represents the raw node features, and its value is the log probabilities of the prediction associated with each feature. The higher value indicates a higher rate of importance in the prediction of anomalies. Figure 14, which is the feature-wise distribution between normal and anomaly of the same job across different simulations, supports our explanation from the GNNExplainer. For example, the runtime is the running time to complete the associated job. Intuitively, the job is more likely to be an anomaly when the runtime is higher than usual, which is identified by the explainer. This is essentially useful when we want to make decisions to optimize the workflow and reschedule the jobs. Demonstration of node 23 in 1000 Genome. Node-level explanation. Normalized features of 1000 Genome B (Job: 23).


Furthermore, to identify the anomalies of the entire workflow, that is, a single anomaly job in the workflow leads to abnormal behavior of the graph, we also pick the same node 23. Instead of investing its own, the explainer extracts a subgraph associated with it. Figure 15 demonstrates the difference between the normal workflow and anomaly workflow. Each row represents the job in the workflow, and each column represents the node feature. As we can see, the explainer identifies the anomaly node and the causes from its ancestors. In the scenario where a couple of its parent nodes are anomalous, it may affect the job itself, resulting in significant delays in collecting and transferring results to its successors. This aligns with our findings from the structure. Graph-level explanation.
Conclusion and future work
In this study, we examined anomaly detection in workflows by representing them as directed acyclic graphs and utilizing graph neural networks (GNNs) to identify anomalous workflows. By utilizing both node features and local structural information, our GNN models outperformed traditional machine learning and computer vision techniques. In addition, we boost the model performance by tuning hyperparameters in the GNN model and integrate explainable approaches to provide insights on job and graph level anomalies. In the future, we plan to investigate the efficiency of larger scientific workflows and apply our models to new workflow datasets. Additionally, since creating intentional anomalies can be time consuming, we plan to use generative models to generate synthetic anomaly data.
Footnotes
Declaration of conflicting interests
The author(s) declared no potential conflicts of interest with respect to the research, authorship, and/or publication of this article.
Funding
The author(s) disclosed receipt of the following financial support for the research, authorship, and/or publication of this article: This work is funded by the Department of Energy under the Integrated Computational and Data Infrastructure (ICDI) for Scientific Discovery, grant #DE-SC0022328. Experimental data was collected on the ExoGENI testbed supported by NSF. This material is based upon work supported by the U.S. Department of Energy, Office of Science, under contract number DE-AC02-06CH11357.
Author biographies
Hongwei Jin received his Ph.D. in computer science from the University of Illinois at Chicago in 2022. Before that, he got his M.S. in applied mathematics from the Illinois Institute of Technology. He had a couple of works that have been published in top-tier machine learning conferences including NeurIPS, UAI, IJCAI, ECML-PKDD, etc.
Krishnan Raghavan received his Ph.D. in computer engineering from Missouri University of Science and Technology in May 2019. He has co-authored several papers on deep neural networks with applications to big data.
George Papadimitriou is a Computer Science PhD candidate at the University of Southern California, and a Graduate Research Assistant in the Science Automation Technologies group at the USC Information Sciences Institute. His research interests lie within the intersection of Data Intensive Applications and Distributed Computing. He received his BS in Electrical and Computer Engineering from the National Technical University of Athens.
Cong Wang is a senior network and systems researcher at RENCI, University of North Carolina at Chapel Hill. His research focuses on cloud computing, networking, and distributed systems. He obtained his PhD in the department of Electrical and Computer Engineering at University of Massachusetts Amherst.
Anirban Mandalserves as the Assistant Director for network research and infrastructure at Renaissance Computing Institute (RENCI) at University of North Carolina, Chapel Hill. He leads several efforts in cyberinfrastructure research in support of science. His research interests lie in the areas of distributed systems, cloud computing, networking, and data-driven scientific workflows. His research deals with resource provisioning, scheduling, performance analysis, machine learning, and anomaly detection for large scale scientific cyberinfrastructures, next generation networks and experimental testbeds. Prior to joining RENCI, he earned his PhD degree in Computer Science from Rice University in 2006 and a Bachelor’s degree in Computer Science & Engineering from IIT Mumbai, India in 2000.
Mariam Kiran is a research scientist with shared positions with Energy Sciences Network and the Scientific Data Management (SDM) group in Computational Research Division. Her work specifically concentrates on using advanced software and machine learning techniques to advance system architectures, particularly high-speed networks such as DOE networks. Her current work is exploring reinforcement learning, unsupervised clustering and classification techniques to optimally control distributed network resources, improving high-speed big data transfers for exascale science applications and optimize how current network infrastructure is utilized. Kiran is the recipient of the DOE ASCR Early Career Award in 2017. Before joining LBNL in 2016, Kiran held positions as a lecturer and research fellow at the Universities of Sheffield and Leeds in the UK. She earned her undergrad and PhD degree in software engineering and computer science from the University of Sheffield, UK in 2011.
Prasanna Balaprakash is the director of AI Program and Distinguished R&D Scientist at Oak Ridge National Laboratory. His research interests span the areas of artificial intelligence, machine learning, optimization, and high-performance computing. Currently, he seeks to deliver foundational, scalable, and applied AI/ML capabilities supporting Oak Ridge National Laboratory's broad mission and provides world-class solutions in computer and computational science, neutron science, materials science, biology and health science, nuclear engineering, isotopes, manufacturing, energy, and climate science. He is a recipient of U.S. Department of Energy 2018 Early Career Award. He is the machine-learning team lead and data-understanding team co-lead in RAPIDS, the SciDAC Computer Science institute. Prior to ORNL, he was the R&D Group Leader and Computer Scientist in the Mathematics and Computer Science Division with a joint appointment in the Leadership Computing Facility at Argonne National Laboratory.
Ewa Deelman received her PhD in Computer Science from the Rensselaer Polytechnic Institute in 1998. Following a postdoc at the UCLA Computer Science Department she joined the University of Southern California’s Information Sciences Institute (ISI) in 2000, where she is serving as a Research Director and is leading the Science Automation Technologies group. She is also a Research Professor at the USC Computer Science Department and an AAAS and IEEE Fellow. Dr. Deelman’s research interests include the design and exploration of collaborative, distributed scientific environments, with particular emphasis on workflow management as well as the management of large amounts of data and metadata. At ISI, Dr. Deelman is leading the Pegasus project, which designs and implements workflow mapping techniques for large-scale applications running in distributed environments. Pegasus is being used today in a number of scientific disciplines, enabling researchers to formulate complex computations in a declarative way.
