Contrastive Quant: Quantization Makes Stronger Contrastive Learning

Yonggan Fu,Qixuan Yu,Meng Li,Xu Ouyang,Vikas Chandra,Yingyan Lin
DOI: https://doi.org/10.1145/3489517.3530419
2022-01-01
Abstract:Contrastive learning learns visual representations by enforcing feature consistency under different augmented views. In this work, we explore contrastive learning from a new perspective. Interestingly, we find that quantization, when properly engineered, can enhance the effectiveness of contrastive learning. To this end, we propose a novel contrastive learning framework, dubbed Contrastive Quant, to encourage feature consistency under both differently augmented inputs via various data transformations and differently augmented weights/activations via various quantization levels. Extensive experiments, built on top of two state-of-the-art contrastive learning methods SimCLR and BYOL, show that Contrastive Quant consistently improves the learned visual representation.
What problem does this paper attempt to address?