Aleatory-aware deep uncertainty quantification for transfer learning

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

請使用永久網址來引用或連結此文件: 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	141	檢視/開啟

在ASIAIR中所有的資料項目都受到原著作權保護.

資料載入中.....