Masking Improves Contrastive Self-Supervised Learning for ConvNets, and Saliency Tells You Where

Zhi Yi Chin*, Chieh Ming Jiang, Ching Chun Huang, Pin Yu Chen, Wei Chen Chiu

*此作品的通信作者

研究成果: Conference contribution同行評審

摘要

While image data starts to enjoy the simple-but-effective self-supervised learning scheme built upon masking and self-reconstruction objective thanks to the introduction of tokenization procedure and vision transformer backbone, convolutional neural networks as another important and widely-adopted architecture for image data, though having contrastive-learning techniques to drive the self-supervised learning, still face the difficulty of leveraging such straightforward and general masking operation to benefit their learning process significantly. In this work, we aim to alleviate the burden of including masking operation into the contrastive-learning framework for convolutional neural networks as an extra augmentation method. In addition to the additive but unwanted edges (between masked and unmasked regions) as well as other adverse effects caused by the masking operations for ConvNets, which have been discussed by prior works, we particularly identify the potential problem where for one view in a contrastive sample-pair the randomly-sampled masking regions could be overly concentrated on important/salient objects thus resulting in misleading contrastiveness to the other view. To this end, we propose to explicitly take the saliency constraint into consideration in which the masked regions are more evenly distributed among the foreground and background for realizing the masking-based augmentation. Moreover, we introduce hard negative samples by masking larger regions of salient patches in an input image. Extensive experiments conducted on various datasets, contrastive learning mechanisms, and downstream tasks well verify the efficacy as well as the superior performance of our proposed method with respect to several state-of-the-art baselines. Our code is publicly available at: https://github.com/joycenerd/Saliency-Guided-Masking-for-ConvNets

原文English
主出版物標題Proceedings - 2024 IEEE Winter Conference on Applications of Computer Vision, WACV 2024
發行者Institute of Electrical and Electronics Engineers Inc.
頁面2749-2758
頁數10
ISBN(電子)9798350318920
DOIs
出版狀態Published - 3 1月 2024
事件2024 IEEE Winter Conference on Applications of Computer Vision, WACV 2024 - Waikoloa, United States
持續時間: 4 1月 20248 1月 2024

出版系列

名字Proceedings - 2024 IEEE Winter Conference on Applications of Computer Vision, WACV 2024

Conference

Conference2024 IEEE Winter Conference on Applications of Computer Vision, WACV 2024
國家/地區United States
城市Waikoloa
期間4/01/248/01/24

指紋

深入研究「Masking Improves Contrastive Self-Supervised Learning for ConvNets, and Saliency Tells You Where」主題。共同形成了獨特的指紋。

引用此