论文标题

信息引导的像素的像素范围的对比度学习

Information-guided pixel augmentation for pixel-wise contrastive learning

论文作者

Quan, Quan, Yao, Qingsong, Li, Jun, Zhou, S. kevin

论文摘要

对比学习(CL)是一种自我监督学习的一种形式,已被广泛用于各种任务。与经过广泛研究的实例级对比度学习不同,像素对比度学习主要有助于按像素的任务,例如医疗地标检测。实例级别CL中与实例的对应物是像素的相邻上下文。为了构建更好的功能表示,有关于设计实例级别CL的实例增强策略的大量文献;但是,对于像素粒度的像素CL的像素增强功能几乎没有类似的工作。在本文中,我们试图弥合这一差距。我们首先根据像素所包含的信息数量将像素分为三类,即低,中和高信息。然后,我们受到``感染力''原则的启发,然后在增强强度和采样率方面为每个类别设计单独的增强策略。广泛的实验验证了我们的信息引导的像素增强策略成功地编码了更具歧视性的代表性,并超越了一个不可识别的本地效果。据我们所知,我们是第一个使用像素粒度来增强无监督像素对比度学习的像素增强方法的人。

Contrastive learning (CL) is a form of self-supervised learning and has been widely used for various tasks. Different from widely studied instance-level contrastive learning, pixel-wise contrastive learning mainly helps with pixel-wise tasks such as medical landmark detection. The counterpart to an instance in instance-level CL is a pixel, along with its neighboring context, in pixel-wise CL. Aiming to build better feature representation, there is a vast literature about designing instance augmentation strategies for instance-level CL; but there is little similar work on pixel augmentation for pixel-wise CL with a pixel granularity. In this paper, we attempt to bridge this gap. We first classify a pixel into three categories, namely low-, medium-, and high-informative, based on the information quantity the pixel contains. Inspired by the ``InfoMin" principle, we then design separate augmentation strategies for each category in terms of augmentation intensity and sampling ratio. Extensive experiments validate that our information-guided pixel augmentation strategy succeeds in encoding more discriminative representations and surpassing other competitive approaches in unsupervised local feature matching. Furthermore, our pretrained model improves the performance of both one-shot and fully supervised models. To the best of our knowledge, we are the first to propose a pixel augmentation method with a pixel granularity for enhancing unsupervised pixel-wise contrastive learning.

扫码加入交流群

加入微信交流群

微信交流群二维码

扫码加入学术交流群,获取更多资源