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)