Home

Awesome

Uncertainty Estimation by Fisher Information-based Evidential Deep Learning

Paper Project

Authors: Danruo Deng, Guangyong Chen, Yu Yang, Furui Liu, Pheng-Ann Heng
Affiliations: CUHK, Zhejiang Lab

Uncertainty estimation is a key factor that makes deep learning reliable in practical applications. Recently proposed evidential neural networks explicitly account for different uncertainties by treating the network's outputs as evidence to parameterize the Dirichlet distribution, and achieve impressive performance in uncertainty estimation. However, for high data uncertainty samples but annotated with the one-hot label, the evidence-learning process for those mislabeled classes is over-penalized and remains hindered. To address this problem, we propose a novel and simple method, Fisher Information-based Evidential Deep Learning ($\mathcal{I}$-EDL). In particular, we introduce Fisher Information Matrix (FIM) to measure the informativeness of evidence carried by each sample, according to which we can dynamically reweight the objective loss terms to make the network more focused on the representation learning of uncertain classes. The generalization ability of our network is further improved by optimizing the PAC-Bayesian bound. As demonstrated empirically, our proposed method consistently outperforms traditional EDL-related algorithms in multiple uncertainty estimation tasks, especially in the more challenging few-shot classification settings.

<img src="iedl.jpg" alt="drawing" width="100%"/>

🔥 Updates

🛠️ Requirement

git clone --recursive https://github.com/danruod/IEDL.git
conda env create -f environment.yml
conda activate IEDL

🚀 File Structure & Quick-starter code

This repository mainly contains two folders:

🤝 Acknowledgement

📑 Citation

If you find that our work is helpful in your research, please consider citing our paper:

@article{deng2023uncertainty,
  title={Uncertainty Estimation by Fisher Information-based Evidential Deep Learning},
  author={Deng, Danruo and Chen, Guangyong and Yu, Yang and Liu, Furui and Heng, Pheng-Ann},
  journal={arXiv preprint arXiv:2303.02045},
  year={2023}
}

✉️ Contact

Feel free to contact me (Danruo DENG: drdeng@link.cuhk.edu.hk) if anything is unclear or you are interested in potential collaboration.