PROB icon indicating copy to clipboard operation
PROB copied to clipboard

Thoughts on final prediction

Open luckylukez opened this issue 1 year ago • 0 comments

Hi Orr,

I have a couple thoughts on the final prediction and would love to hear your thoughts.

If I understand things correctly the C+1 logit of the classification head is the logit for "not known" objects, which can be either unknown or background. The objectness head on the other hand, learns a general objectness from the known classes in order to give a higher objectness score to objects that are known or similar to known objects. The objects that are similar to known objects but not known are what we hope will match with unknown objects that are not background. Is that a fair description?

My thought is this: The objectness head will score higher for objects similar to known classes, but the C+1 logit might score higher the less similar an object is to any known object as it is essentially a score of "not known". So when detecting unknowns, there could be a risk that objects that are less similar to knowns actually achieve higher scores simply because the C+1 logit is high. In that case maybe it would be reasonable to let the classification head sort unknowns from knowns, but then let the final score of the unknowns be just the score from the objectness head as this should favor objects more similar to known objects.

Any thoughts on this?

luckylukez avatar May 14 '24 12:05 luckylukez