Aleatory-aware deep uncertainty quantification for transfer learning

ASIA unversity > 資訊學院 > 生物資訊與醫學工程學系 > 期刊論文 > Item 310904400/115015

jsp.display-item.identifier=請使用永久網址來引用或連結此文件: http://asiair.asia.edu.tw/ir/handle/310904400/115015

题名:	Aleatory-aware deep uncertainty quantification for transfer learning
作者:	Kab, H M Dipu;Kabir, H M Dipu;Rajendra, U.;Acharya, U. Rajendra
贡献者:	資訊電機學院生物資訊與醫學工程學系
关键词:	Patient referral；Uncertainty；COVID；Aleatoric；Epistemic；Heteroscedastic
日期:	2022-01-01
上传时间:	2023-03-28 01:58:35 (UTC+0)
出版者:	亞洲大學
摘要:	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 classification 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 parameters 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:
显示于类别:	[生物資訊與醫學工程學系 ] 期刊論文

文件中的档案:

档案	描述	大小	格式	浏览次数
index.html		0Kb	HTML	140	检视/开启

在ASIAIR中所有的数据项都受到原著作权保护.

数据加载中.....