Dropout Sampling for Robust Object Detection in Open-Set Condition

Investigates dropout sampling object detection

Summary

This papers shows object detection in open-set. They show that dropout sampling can be effective to reduce the false-positive for the unknown classes (those classes which were not in the training dataset) in the testing dataset.

Methodology

Dropout sampling

In this, multiple inferences are made with same input with dropout layer being active. Hence, we get multiple output for the same input. This is a variational inference technique as it helps to obtain rather intractable infereces.

Bayesian Perpective

Basic Idea: Model the network's wights WWas a distribution p(W∣T)p(W|T)conditioned on the training data TT, instead of a deterministic point estimate vairable.

How: Placing a prior over the weights e.g. W∼N(0,I)W \sim N(0,I), the network training can be interpreted as determining a posterior over the weights given the training data: p(W∣T)p(W|T). However, evaluating this posterior is not tractable without approximation techniques. Where TTis the training data

See What's Bayesian here: - Prior: W∼N(0,I)W \sim N(0,I) - Liklihood: Training of network is basically liklihood estimation - Posterior: Final trained wieght distribution p(W∣T)p(W|T) is the posterior

Approximation to Intractable Inference: Let IIbe the input to the network, then bayesian inference of the ouput is:

p(y∣I,T)=∫p(y∣I,w)p(w∣T)dw≈1n∑i=1nsip(y|I,T) = \int p(y|I,w)p(w|T)dw \approx \frac{1}{n}\sum_{i=1}^n s_i

where nnis the number of times the inference is done and sis_iis the ouput at ithi^{th}inference in which wiw_iis used which is sampled using dropout.

Final: Then basically they use the Entropy of the above classification output p(y∣I,T)p(y|I,T) to threshold between the unknown and known classes. If the entropy is high, then the detected output is mostly from an outlier class.

Insights

  • Using entropy over average of output is better than single output. As aerage gives better approximation of posterior.

Last updated