LLM2D
通过特征归因增强人工智能回归任务中的特征选择和可解释性
Enhancing Feature Selection and Interpretability in AI Regression Tasks Through Feature Attribution
作者: Alexander Hinterleitner, Thomas Bartz-Beielstein, Richard Schulz, Sebastian Spengler, Thomas Winter, Christoph Leitenmeier
发布日期: 9/26/2024
arXiv ID: oai:arXiv.org:2409.16787v1

摘要

可解释人工智能(XAI)的研究正在不断发展,旨在使深度学习模型更加透明。大多数XAI方法侧重于为安全相关应用中人工智能(AI)系统的决策提供理由。然而,很少有人关注使用这些方法来提高深度学习算法的性能和鲁棒性。此外,许多现有的XAI工作主要针对分类问题。在本研究中,我们研究了特征归因方法在回归问题中过滤掉无信息特征的潜力,从而提高预测的准确性和稳定性。我们介绍了一种特征选择管道,它将集成梯度与k均值聚类相结合,从初始数据空间中选择最佳变量集。为了验证这种方法的有效性,我们将其应用于一个现实世界的工业问题——涡轮机械开发过程中的叶片振动分析。