MCDropout

model

train

feature -> ReconNet_dropout -> drop.feature

cat[ drop.feature, init.feature] cat at dim=1

• ReconNet_dropout

dropout

​ shortcut = self.conv1(x)

​ shortcut.features = self.bn0(shortcut.features)

​ shortcut.features = self.act1(shortcut.features)

​ shortcut.features = self.dropout1(shortcut.features)

eval

metric

???predict_labels = torch.mean(outputs, dim=0, keepdim=True)

uncertainty_scores = torch.var(outputs, dim=0, keepdim=True)
uncertainty_scores = torch.mean(uncertainty_scores, dim=1)