Predicate Classification Using Optimal Transport Loss in Scene Graph Generation

Sorachi Kurita, Satoshi Oyama, Itsuki Noda

In scene graph generation (SGG), learning with cross-entropy loss yields biased predictions owing to the severe imbalance in the distribution of the relationship labels in the data…