Robust Contrastive Learning With Theory Guarantee
Tran, Ngoc N., Tran, Lam, Phan, Hoang, Bui, Anh, Pham, Tung, Tran, Toan, Phung, Dinh, Le, Trung
–arXiv.org Artificial Intelligence
Contrastive learning (CL) allows us to create meaningful features without any label information. In the first phase, CL approaches learn the features, which are then classified by a linear classifier that has been learned from labeled data. While existing theoretical works have studied the connection between the supervised loss in the second phase and the unsupervised loss in the first phase to explain why the unsupervised loss can support the supervised loss, there has been no theoretical examination of the connection between the unsupervised loss in the first phase and the robust supervised loss in the second phase, which can shed light on how to establish an effective unsupervised loss in the first phase. To fill this gap, our paper develops rigorous theories to identify which components in the supervised loss can aid the robust supervised loss. Finally, we conduct experiments to verify our findings. All code used in this work is available at https://anonymous.4open.science/r/rosa.
arXiv.org Artificial Intelligence
Nov-16-2023