1
$\begingroup$

In image classification, there are sometimes images that do not fit in any category.

For example, if I build a CNN in Keras to classify Dogs and Cats, does it help (in terms of training time and performance) to create an "other" (or "unclassified") category in which images of houses, people, birds, etc., are classified? Is there any research paper that discusses this?

A similar question was asked before here, but, unfortunately, it has no answer.

$\endgroup$
3

1 Answer 1

1
$\begingroup$

It is not advisable because if you use an "other" class, you are just increasing problems for your network. Since "other" means not dog and not cat, then, what common feature does it have? Most of the time the "other" images won't have many features in common. If they do, then go ahead and make an "other" class.

There is a better way: if the probabilities for both cat and dog are less than a threshold (you need to decide that, take, 0.5), then, you can say it is an "other" object.

$\endgroup$
2
  • $\begingroup$ Thank you. If I know that some image categories (for ex. horses) will appear often but they are not of practical interest (my goal is to only tag Cats and Dogs), does it in that case make sense for model performance to make a third category? $\endgroup$ Mar 9, 2021 at 10:37
  • $\begingroup$ It only makes sense if the third class has visual features similar to the classes of interest. Since, it will be a source of confusion in decision, it makes sense to make the network to learn to separate them out. On the other hand, if confusion is happening then you can increase the threshold for classification. $\endgroup$ Mar 9, 2021 at 10:45

You must log in to answer this question.

Not the answer you're looking for? Browse other questions tagged .