返回
Aleatory-aware deep uncertainty quantification for transfer learning
DOI:10.1016/j.compbiomed.2022.105246.png)
摘要
En 中文
The user does not have any idea about the credibility of outcomes from deep neural networks (DNN) when uncertainty quantification (UQ) is not employed. However, current Deep UQ classification models capture mostly epistemic uncertainty. Therefore, this paper aims to propose an aleatory-aware Deep UQ method for classifi-cation problems. First, we train DNNs through transfer learning and collect numeric output posteriors for all training samples instead of logical outputs. Then we determine the probability of happening a certain class from K-nearest output posteriors of the same DNN in training samples. We name this probability as opacity score, as the paper focuses on the detection of opacity on X-ray images. This score reflects the level of aleatory on the sample. When the NN is certain on the classification of the sample, the probability of happening a class becomes much higher than the probabilities of others. Probabilities for different classes become close to each other for a highly uncertain classification outcome. To capture the epistemic uncertainty, we train multiple DNNs with different random initializations, model selection, and augmentations to observe the effect of these training pa-rameters on prediction and uncertainty. To reduce execution time, we first obtain features from the pre-trained NN. Then we apply features to the ensemble of fully connected layers to get the distribution of opacity score during the test. We also train several ResNet and DenseNet DNNs to observe the effect of model selection on prediction and uncertainty. The paper also demonstrates a patient referral framework based on the proposed uncertainty quantification. The scripts of the proposed method are available at the following link: https://github. com/dipuk0506/Aleatory-aware-UQ.
Keyword:
Patient referral
Uncertainty
COVID
Aleatoric
Epistemic
Heteroscedastic
期刊
IF:
6.3
论文数:
8.3K
被引数:
3.3W
机构
引用论文
Learning-to-augment strategy using noisy and denoised data: Improving generalizability of deep CNN for the detection of COVID-19 in X-ray images使用噪声和去噪数据的学习增强策略: 提高深度CNN在x射线图像新型冠状病毒肺炎检测中的普遍性
Probabilistic modelling of wind turbine power curves with application of heteroscedastic Gaussian Process regression应用异方差高斯过程回归的风电机组功率曲线概率建模
RENEWABLE ENERGY
IF9.1
An Uncertainty-Aware Transfer Learning-Based Framework for COVID-19 Diagnosis基于不确定性感知迁移学习的新型冠状病毒肺炎诊断框架

