Question about Contrastive Learning Collapse

[EdgarAllanPoo]

Thank you very much for your work. May I ask have you ever encounter a condition during experiments where your model collapses (only able to reconstruct certain classes only)? I have tried to replicate your work as a basis on other datasets . However, my experiments only able to reconstruct 2 types of images regardless of the pair in the test set. Looking forward to your reply!