Task-Driven Causal Feature Distillation: Towards Trustworthy Risk Prediction

Zhixuan Chu,Mengxuan Hu,Qing Cui,Longfei Li,Sheng Li
DOI: https://doi.org/10.1609/aaai.v38i10.29047
2024-01-01
Abstract:Since artificial intelligence has seen tremendous recent successes in manyareas, it has sparked great interest in its potential for trustworthy andinterpretable risk prediction. However, most models lack causal reasoning andstruggle with class imbalance, leading to poor precision and recall. To addressthis, we propose a Task-Driven Causal Feature Distillation model (TDCFD) totransform original feature values into causal feature attributions for thespecific risk prediction task. The causal feature attribution helps describehow much contribution the value of this feature can make to the risk predictionresult. After the causal feature distillation, a deep neural network is appliedto produce trustworthy prediction results with causal interpretability and highprecision/recall. We evaluate the performance of our TDCFD method on severalsynthetic and real datasets, and the results demonstrate its superiority overthe state-of-the-art methods regarding precision, recall, interpretability, andcausality.
What problem does this paper attempt to address?