arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2007.09070cs.LGcs.CVstat.ML
收藏

Hybrid Discriminative-Generative Training via Contrastive Learning

University of California, Berkeley(加州大学伯克利分校)

Hao Liu, Pieter Abbeel

英文摘要:

Contrastive learning and supervised learning have both seen significant progress and success. However, thus far they have largely been treated as two separate objectives, brought together only by having a shared neural network. In this paper we show that through the perspective of hybrid discriminative-generative training of energy-based models we can make a direct connection between contrastive learning and supervised learning. Beyond presenting this unified view, we show our specific choice of approximation of the energy-based loss outperforms the existing practice in terms of classification accuracy of WideResNet on CIFAR-10 and CIFAR-100. It also leads to improved performance on robustness, out-of-distribution detection, and calibration.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑