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

利用知识图谱生成可解释特征

Leveraging Knowlegde Graphs for Interpretable Feature Generation

SAP Labs Paris(思爱普巴黎实验室);Université Paris Cité(巴黎西岱大学)

Mohamed Bouadi, Arta Alavi, Salima Benbernou, Mourad Ouziri

AI总结:

提出KRAFT框架,利用知识图谱指导生成可解释特征,结合神经生成器与基于知识的推理器,通过深度强化学习训练,在真实数据集上显著提升准确率并确保高可解释性。

摘要:

机器学习(ML)模型的质量高度依赖于输入数据,因此在ML中通常需要进行特征工程(FE)。此外,随着ML驱动的系统日益普及,特别是在关键场景中,对可解释性和说明性的需求变得愈发重要。由于手动FE耗时且需要特定领域的知识,我们提出了KRAFT,这是一个利用知识图谱来指导生成可解释特征的自动特征工程(AutoFE)框架。我们的混合AI方法结合了一个神经生成器,通过一系列转换来变换原始特征,以及一个基于知识的推理器,利用描述逻辑(DL)评估特征的可解释性。该生成器通过深度强化学习(DRL)进行训练,以最大化预测准确率和生成特征的可解释性。在真实数据集上的大量实验表明,KRAFT在确保高可解释性的同时显著提升了准确率。

英文摘要:

The quality of Machine Learning (ML) models strongly depends on the input data, as such Feature Engineering (FE) is often required in ML. In addition, with the proliferation of ML-powered systems, especially in critical contexts, the need for interpretability and explainability becomes increasingly important. Since manual FE is time-consuming and requires case specific knowledge, we propose KRAFT, an AutoFE framework that leverages a knowledge graph to guide the generation of interpretable features. Our hybrid AI approach combines a neural generator to transform raw features through a series of transformations and a knowledge-based reasoner to evaluate features interpretability using Description Logics (DL). The generator is trained through Deep Reinforcement Learning (DRL) to maximize the prediction accuracy and the interpretability of the generated features. Extensive experiments on real datasets demonstrate that KRAFT significantly improves accuracy while ensuring a high level of interpretability.

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

↑