Frozen is better than learning: A new design of prototype-based classifier for semantic segmentation

Jialei Chen,Daisuke Deguchi,Chenkai Zhang,Xu Zheng,Hiroshi Murase
DOI: https://doi.org/10.1016/j.patcog.2024.110431
IF: 8
2024-03-24
Pattern Recognition
Abstract:Semantic segmentation models comprise an encoder to extract features and a classifier for prediction. However, the learning of the classifier suffers from the ambiguity which is caused by two factors: (1) the weights of a classifier for similar categories may have positive similarities lowing the performance for similar categories, named correlation ambiguity , and (2) the classifier is prone to predict the category with a larger l2 norm and vice versa, termed prior ambiguity . To comedy the issues, we propose Category-Basis Prototype (CBP), frozen and mutually orthogonalized prototypes with equal l2 norm . Orthogonalization prevents the prototypes from being similar to each other and the equality decouples the prediction from the l2 norm. To better shape the feature space, we propose Online Centroid Contrastive Loss (OCCL) equipped with centroid and category-level losses. Experiments show that our method yields compelling results over two widely applied benchmarks indicating the effectiveness of our methods.
computer science, artificial intelligence,engineering, electrical & electronic
What problem does this paper attempt to address?