arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2403.02439cs.LGcs.AI

使用可解释人工智能定位预测异常的根本原因

Root Causing Prediction Anomalies Using Explainable AI

Meta

Ramanathan Vishnampet, Rajesh Shenoy, Jianhui Chen, Anuj Gupta

AI总结:

本研究将可解释人工智能应用于持续学习模型的性能下降根因定位,通过全局特征重要性分布的时间偏移识别异常成因,召回率优于传统相关性方法,已部署用于生产数据管线监控。

摘要:

本文提出了可解释人工智能(XAI)的一种新颖应用,用于定位从用户参与数据中持续学习的机器学习模型的性能下降根本原因。在此类系统中,单一特征损坏可能引发级联的特征漂移、标签漂移和概念漂移。我们已成功应用该技术提升了个性化广告所用模型的可靠性。\n此类系统的性能下降表现为模型的预测异常。这些模型通常持续使用由数百个实时数据处理管线生成或从其他上游模型派生的特征进行训练。任何管线故障或上游模型的不稳定性都可能导致特征损坏,进而使模型的预测输出偏离实际输出,并造成训练数据损坏。特征与预测输出之间的因果关系十分复杂,且由于系统的规模和动态性,定位根本原因极具挑战性。\n我们证明了全局特征重要性分布的时间偏移能够有效隔离预测异常的成因,其召回率优于模型-特征相关性方法。即便使用简单的基于扰动的方法近似局部特征重要性,并在数千个样本上进行聚合,该技术依然有效。我们发现,该技术是一种模型无关、低成本且高效的生产环境复杂数据管线监控方式,并且已部署了一套系统,用于持续分析持续训练模型的全局特征重要性分布。

英文摘要:

This paper presents a novel application of explainable AI (XAI) for root-causing performance degradation in machine learning models that learn continuously from user engagement data. In such systems a single feature corruption can cause cascading feature, label and concept drifts. We have successfully applied this technique to improve the reliability of models used in personalized advertising. Performance degradation in such systems manifest as prediction anomalies in the models. These models are typically trained continuously using features that are produced by hundreds of real time data processing pipelines or derived from other upstream models. A failure in any of these pipelines or an instability in any of the upstream models can cause feature corruption, causing the model's predicted output to deviate from the actual output and the training data to become corrupted. The causal relationship between the features and the predicted output is complex, and root-causing is challenging due to the scale and dynamism of the system. We demonstrate how temporal shifts in the global feature importance distribution can effectively isolate the cause of a prediction anomaly, with better recall than model-to-feature correlation methods. The technique appears to be effective even when approximating the local feature importance using a simple perturbation-based method, and aggregating over a few thousand examples. We have found this technique to be a model-agnostic, cheap and effective way to monitor complex data pipelines in production and have deployed a system for continuously analyzing the global feature importance distribution of continuously trained models.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑