Attention Enhanced Entity Recommendation for Intelligent Monitoring in Cloud Systems

TL;DR

This paper introduces DiRecGNN, an attention-based graph neural network that improves entity recommendation for cloud service monitoring, capturing long-range dependencies and outperforming existing methods.

Contribution

The paper presents a novel attention-enhanced entity ranking model inspired by transformers, tailored for recommending monitoring attributes in cloud systems.

Findings

43.1% increase in MRR over existing methods

Product teams rated the feature 4.5 out of 5 for usefulness

Effective capture of long-range dependencies in heterogeneous graphs

Abstract

In this paper, we present DiRecGNN, an attention-enhanced entity recommendation framework for monitoring cloud services at Microsoft. We provide insights on the usefulness of this feature as perceived by the cloud service owners and lessons learned from deployment. Specifically, we introduce the problem of recommending the optimal subset of attributes (dimensions) that should be tracked by an automated watchdog (monitor) for cloud services. To begin, we construct the monitor heterogeneous graph at production-scale. The interaction dynamics of these entities are often characterized by limited structural and engagement information, resulting in inferior performance of state-of-the-art approaches. Moreover, traditional methods fail to capture the dependencies between entities spanning a long range due to their homophilic nature. Therefore, we propose an attention-enhanced entity ranking model inspired by transformer architectures. Our model utilizes a multi-head attention mechanism to focus on heterogeneous neighbors and their attributes, and further attends to paths sampled using random walks to capture long-range dependencies. We also employ multi-faceted loss functions to optimize for relevant recommendations while respecting the inherent sparsity of the data. Empirical evaluations demonstrate significant improvements over existing methods, with our model achieving a 43.1% increase in MRR. Furthermore, product teams who consumed these features perceive the feature as useful and rated it 4.5 out of 5.