arXiv · 2207.13367
Optimizing transformations for contrastive learning in a differentiable framework
Abstract
Current contrastive learning methods use random transformations sampled from a large list of transformations, with fixed hyperparameters, to learn invariance from an unannotated database. Following previous works that introduce a small amount of supervision, we propose a framework to find optimal transformations for contrastive learning using a differentiable transformation network. Our method increases performances at low annotated data regime both in supervision accuracy and in convergence speed. In contrast to previous work, no generative model is needed for transformation optimization. Transformed images keep relevant information to solve the supervised task, here classification. Experiments were performed on 34000 2D slices of brain Magnetic Resonance Images and 11200 chest X-ray images. On both datasets, with 10% of labeled data, our model achieves better performances than a fully supervised model with 100% labels.
Explore related subjects
Keep this discovery
Camille Ruppli, Pietro Gori, Roberto Ardon, Isabelle Bloch. 2022-07-27. Optimizing transformations for contrastive learning in a differentiable framework. https://doi.org/10.1007/978-3-031-16760-7_10
Cite the original work for its findings. Save a collection to share your selection of sources.