lorenmt / reco

The implementation of "Bootstrapping Semantic Segmentation with Regional Contrast" [ICLR 2022].
https://shikun.io/projects/regional-contrast
Other
162 stars 25 forks source link

The question about the selection of semantic segmentation network #12

Closed super233 closed 2 years ago

super233 commented 2 years ago

Hi, thanks for your awesome work.

I have a question about the selection of semantic segmentation networks in contrastive learning, which is not about the code.

I have noticed that recent contrastive learning based works are all implemented with DeepLab V3+, such as DCL, PC2Seg, RegionContrast and your paper, and I am curious about why there's no paper to implement contrastive learning on DeepLab V2.

In my view, it's mainly due to the shallow structure of DeepLab V2, which is not convenient to generate well representational features for subsequent contrastive learning. In your implementation of DeepLab V2, the features are generated from the backbone, however, the predicted logits are generated from the ASPP.

What do you think about this question? Looking forward to your reply. Please forgive me for interrupting. :-)

lorenmt commented 2 years ago

The design of DeepLabv2 directly uses ASPP module (based on encoder feature) for label prediction which is not well suited to attach an additional branch for representation learning.

In addition, DeepLabv2 trains significantly slower than DeepLabv3+ with worse performance. So choosing DeepLabv3+ seems to be a no-brainer.