Zhang, Y., Yin, Y., Zhang, Y., & Zimmermann, R. (2022). Mix-Up Self-Supervised Learning for Contrast-Agnostic Applications. 2022 IEEE International Conference on Multimedia and Expo (ICME). https://doi.org/10.1109/icme52920.2022.9859753
Abstract:
Contrastive self-supervised learning has attracted significant research attention recently. It learns effective visual representations from unlabeled data by embedding augmented views of the same image close to each other while pushing away embeddings of different images. Despite its great success on ImageNet classification, COCO object detection, etc., its performance degrades on contrast-agnostic applications, e.g., medical image classification, where all images are visually similar to each other. This creates difficulties in optimizing the embedding space as the distance between images is rather small. To solve this issue, we present the first mix-up self-supervised learning framework for contrast-agnostic applications. We address the low variance across images based on cross-domain mix-up and build the pretext task based on two
synergistic objectives: image reconstruction and transparency prediction. Experimental results on two benchmark datasets validate the effectiveness of our method, where an improvement of 2.5% - 7.4% in top-1 accuracy was obtained compared to existing self-supervised learning methods.
License type:
Publisher Copyright
Funding Info:
This research / project is supported by the Ministry of Education - Academic Research Fund Tier 1
Grant Reference no. : T1 251RES2029