任务驱动的因果特征蒸馏:迈向可信的风险预测
机器学习
2024-01-23 v2 人工智能
摘要
由于人工智能最近在许多领域取得了巨大成功,引发了人们对其在可信且可解释的风险预测方面潜力的极大兴趣。然而,大多数模型缺乏因果推理能力且难以应对类别不平衡问题,导致精确率和召回率较低。为解决这一问题,我们提出了一种任务驱动的因果特征蒸馏模型(TDCFD),将原始特征值转换为针对特定风险预测任务的因果特征归因。因果特征归因有助于描述该特征的值对风险预测结果的贡献程度。在因果特征蒸馏之后,应用深度神经网络生成具有因果可解释性且高精确率/召回率的可信预测结果。我们在多个合成和真实数据集上评估了 TDCFD 方法的性能,结果表明其在精确率、召回率、可解释性和因果性方面均优于最先进的方法。
引用
@article{arxiv.2312.16113,
title = {Task-Driven Causal Feature Distillation: Towards Trustworthy Risk Prediction},
author = {Zhixuan Chu and Mengxuan Hu and Qing Cui and Longfei Li and Sheng Li},
journal= {arXiv preprint arXiv:2312.16113},
year = {2024}
}
备注
Proceedings of the 2024 AAAI Conference on Artificial Intelligence