Task-Driven Causal Feature Distillation: Towards Trustworthy Risk Prediction
Fuente:
arXiv
Gespeichert in:
| Hauptverfasser: | , , , , |
|---|---|
| Format: | Preprint |
| Veröffentlicht: |
2023
|
| Schlagworte: | |
| Online-Zugang: | |
| Tags: |
Tag hinzufügen
Keine Tags, Fügen Sie den ersten Tag hinzu!
|
| _version_ | 1866929217343586304 |
|---|---|
| author | Chu, Zhixuan Hu, Mengxuan Cui, Qing Li, Longfei Li, Sheng |
| author_facet | Chu, Zhixuan Hu, Mengxuan Cui, Qing Li, Longfei Li, Sheng |
| contents | Since artificial intelligence has seen tremendous recent successes in many areas, it has sparked great interest in its potential for trustworthy and interpretable risk prediction. However, most models lack causal reasoning and struggle with class imbalance, leading to poor precision and recall. To address this, we propose a Task-Driven Causal Feature Distillation model (TDCFD) to transform original feature values into causal feature attributions for the specific risk prediction task. The causal feature attribution helps describe how much contribution the value of this feature can make to the risk prediction result. After the causal feature distillation, a deep neural network is applied to produce trustworthy prediction results with causal interpretability and high precision/recall. We evaluate the performance of our TDCFD method on several synthetic and real datasets, and the results demonstrate its superiority over the state-of-the-art methods regarding precision, recall, interpretability, and causality. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2312_16113 |
| institution | arXiv |
| publishDate | 2023 |
| record_format | arxiv |
| spellingShingle | Task-Driven Causal Feature Distillation: Towards Trustworthy Risk Prediction Chu, Zhixuan Hu, Mengxuan Cui, Qing Li, Longfei Li, Sheng Machine Learning Artificial Intelligence Since artificial intelligence has seen tremendous recent successes in many areas, it has sparked great interest in its potential for trustworthy and interpretable risk prediction. However, most models lack causal reasoning and struggle with class imbalance, leading to poor precision and recall. To address this, we propose a Task-Driven Causal Feature Distillation model (TDCFD) to transform original feature values into causal feature attributions for the specific risk prediction task. The causal feature attribution helps describe how much contribution the value of this feature can make to the risk prediction result. After the causal feature distillation, a deep neural network is applied to produce trustworthy prediction results with causal interpretability and high precision/recall. We evaluate the performance of our TDCFD method on several synthetic and real datasets, and the results demonstrate its superiority over the state-of-the-art methods regarding precision, recall, interpretability, and causality. |
| title | Task-Driven Causal Feature Distillation: Towards Trustworthy Risk Prediction |
| topic | Machine Learning Artificial Intelligence |
| url | https://arxiv.org/abs/2312.16113 |