自监督与表征学习 · CVPR2026 ·2026.06.03

渐进式掩码蒸馏用于自监督视频表征

Progressive Mask Distillation for Self-supervised Video Representation

Kewei Wu、Chong Liang、Zhao Xie、Dan Guo

与你有关的部分

把「由易到难」的课程式蒸馏引入掩码视频建模:用低掩码率学生模型的知识去指导高掩码率学生模型,并配合区域级损失加权与跨层特征对齐,可直接迁移到任何 MAE 类视频/图像预训练框架中提升高掩码率下的细节重建。

原文依据 we propose Progressive Mask Distillation (PMD), which utilizes dynamic mask ratios to facilitate progressive semantic learning from easy to hard
可迁移到 视频动作识别自监督视频预训练掩码图像建模

摘要译文

由模型翻译 · 术语统一 · 请对照原文核对关键结论

掩码视觉建模是一种不使用视觉标注的自监督学习任务,旨在通过掩码重建任务学习判别性表征。重建中单一的掩码比例可能无法捕捉复杂语义,这促使了动态掩码策略的出现。在本工作中,我们提出渐进式掩码蒸馏(PMD),利用动态掩码比例促进由易到难的渐进式语义学习。PMD 集成了三个关键组件:渐进式学生蒸馏器、难度感知区域增强器以及跨层特征对齐器。首先,为捕捉动态视觉语义,我们设计了一个渐进式学生蒸馏器,以逐步递增的掩码比例训练多个学生模型。早期阶段的学生(低掩码比例)从更多可见 token 中学习简单的低层语义,所学知识随后指导下一阶段的学生(更高掩码比例)从更少可见 token 中捕捉困难的高层语义。这种渐进式蒸馏机制增强了高掩码比例下的细节重建。其次,为缓解语义区域学习不足的问题,我们设计了难度感知区域增强器。我们首先对区域重建损失进行平滑,以减少训练轮次间的大幅波动;随后用平滑后的损失学习区域级权重,优先准确学习重建损失较大的区域。第三,为进一步弥合网络层间的语义鸿沟,我们设计了跨层特征对齐。该模块对齐浅层、中层和深层编码器的特征,确保浅层特征融入来自更深层的语义信息。大量实验表明,我们的方法在 Something-Something V2、Kinetics-400、UCF-101 和 HMDB-51 数据集上取得了最优性能。

英文原文摘要 1897 字符

Masked visual modeling is a self-supervised learning task that does not use visual annotations. It aims to learn discriminative representations via a mask-reconstruction task. A single mask ratio in reconstruction may fail to capture complex semantics, which motivates dynamic masking strategies. In this work, we propose Progressive Mask Distillation (PMD), which utilizes dynamic mask ratios to facilitate progressive semantic learning from easy to hard. PMD integrates three key components: a progressive student distiller, a difficulty-aware region enhancer, and a cross-layer feature aligner. First, to capture dynamic visual semantics, we design a progressive student distiller that trains multiple student models with progressively increasing mask ratios. The early-phase student (with a low mask ratio) learns easy, low-level semantics from more visible tokens. This learned knowledge then guides the next-phase student (with a higher mask ratio) to capture hard, high-level semantics from fewer visible tokens. This progressive distillation mechanism enhances detail reconstruction at a high mask ratio. Second, to alleviate insufficient learning of semantic regions, we design a difficulty-aware region enhancer. We first smooth the region reconstruction loss to reduce large fluctuations across training epochs. The smoothed loss is then used to learn region-level weights, prioritizing accurate learning of regions with large reconstruction losses. Third, to further bridge the semantic gap across network layers, we design cross-layer feature alignment. This module aligns features across shallow, middle, and deep encoder layers, ensuring that shallow-layer features incorporate semantic information from deeper layers. Extensive experiments demonstrate that our method achieves state-of-the-art performance on the Something-Something V2, Kinetics-400, UCF-101, and HMDB-51 datasets.

标签

会议 CVPR2026
页码 41677-41687
Bibkey
来源 CVF Open Access