arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2504.09155cs.CV

自监督学习的进化分层掩码

Evolved Hierarchical Masking for Self-Supervised Learning

School of Computer Science, Peking University(北京大学计算机学院)

Zhanzhou Feng, Shiliang Zhang

AI总结:

针对现有掩码图像建模方法固定掩码模式导致视觉线索建模能力有限的问题,提出进化分层掩码方法,利用训练中的模型生成随训练阶段进化的掩码,提升多下游任务性能。

摘要:

现有的掩码图像建模方法采用固定掩码模式来指导自监督训练。由于这些掩码模式依靠不同标准来描述图像内容,坚持固定模式会导致视觉线索建模能力有限。本文介绍了一种进化分层掩码方法,以在自监督学习中追求通用视觉线索建模。该方法利用正在训练的视觉模型将输入视觉线索解析为层次结构,进而据此生成掩码。层次结构的准确性与正在训练的模型能力相当,导致不同训练阶段的掩码模式不断进化。最初,生成的掩码聚焦于低级视觉线索以掌握基本纹理,然后逐渐进化到描述高级线索以加强对更复杂对象语义和上下文的学习。我们的方法不需要额外的预训练模型或注释,并通过进化训练难度确保训练效率。我们在七个下游任务上进行了广泛实验,包括依赖低级细节的部分重复图像检索,以及需要语义解析能力的图像分类和语义分割。实验结果表明,它在这些任务中显著提升了性能。例如,在相同训练轮次下,它在ImageNet-1K分类中超过了最近的MAE 1.1%,在ADE20K分割中超过了1.4%。我们还将所提出的方法与当前对LLM的研究重点对齐。该方法弥合了对语义要求高的任务与大规模预训练之间的差距,并增强了需要低级特征识别任务中的复杂细节感知。

英文摘要:

Existing Masked Image Modeling methods apply fixed mask patterns to guide the self-supervised training. As those mask patterns resort to different criteria to depict image contents, sticking to a fixed pattern leads to a limited vision cues modeling capability.This paper introduces an evolved hierarchical masking method to pursue general visual cues modeling in self-supervised learning. The proposed method leverages the vision model being trained to parse the input visual cues into a hierarchy structure, which is hence adopted to generate masks accordingly. The accuracy of hierarchy is on par with the capability of the model being trained, leading to evolved mask patterns at different training stages. Initially, generated masks focus on low-level visual cues to grasp basic textures, then gradually evolve to depict higher-level cues to reinforce the learning of more complicated object semantics and contexts. Our method does not require extra pre-trained models or annotations and ensures training efficiency by evolving the training difficulty. We conduct extensive experiments on seven downstream tasks including partial-duplicate image retrieval relying on low-level details, as well as image classification and semantic segmentation that require semantic parsing capability. Experimental results demonstrate that it substantially boosts performance across these tasks. For instance, it surpasses the recent MAE by 1.1\% in imageNet-1K classification and 1.4\% in ADE20K segmentation with the same training epochs. We also align the proposed method with the current research focus on LLMs. The proposed approach bridges the gap with large-scale pre-training on semantic demanding tasks and enhances intricate detail perception in tasks requiring low-level feature recognition.

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

↑