arXiv:2409.10473cs.CVcs.AI2024-09ECCV被引 22

用扩散模型统一学习人体骨骼表示,提升下游任务表现。

MacDiff: Unified Skeleton Modeling with Masked Conditional Diffusion

论文配图:MacDiff: Unified Skeleton Modeling with Masked Conditional Diffusion
图 1 · 摘自论文原文
  • 设计掩码条件扩散框架,通过遮蔽输入引导语义编码器学习
  • 在多个基准上达到最优表示性能,且保持生成能力
  • 适用于标签数据少的场景,增强微调效果

自监督学习在基于骨骼的人体动作理解中已证明有效。然而,以往方法或依赖对比学习(存在误负样本问题),或基于重建(学习过多无关低层线索),导致下游任务表征能力受限。近年来生成学习取得显著进展,其作为建模数据分布的预训练任务具有天然挑战性与意义。但生成模型在稀疏空间、冗余时间的骨骼数据上的表示学习潜力仍待挖掘。为此,我们提出掩码条件扩散(MacDiff)统一框架,首次将扩散模型用于骨骼表征学习。具体地,训练一个以语义编码器输出为条件的扩散解码器,对编码器输入随机掩码以引入信息瓶颈并消除冗余。理论上,我们的生成目标包含对比学习目标,能对齐遮蔽与噪声视图;同时强制表示补全噪声视图,提升泛化能力。MacDiff在表征学习基准上达到当前最佳性能,并保持生成能力。此外,利用扩散模型进行数据增强,在标签稀缺场景下显著提升微调表现。项目地址:https://lehongwu.github.io/ECCV24MacDiff/

原文摘要 · Abstract (English)

Self-supervised learning has proved effective for skeleton-based human action understanding. However, previous works either rely on contrastive learning that suffers false negative problems or are based on reconstruction that learns too much unessential low-level clues, leading to limited representations for downstream tasks. Recently, great advances have been made in generative learning, which is naturally a challenging yet meaningful pretext task to model the general underlying data distributions. However, the representation learning capacity of generative models is under-explored, especially for the skeletons with spacial sparsity and temporal redundancy. To this end, we propose Masked Conditional Diffusion (MacDiff) as a unified framework for human skeleton modeling. For the first time, we leverage diffusion models as effective skeleton representation learners. Specifically, we train a diffusion decoder conditioned on the representations extracted by a semantic encoder. Random masking is applied to encoder inputs to introduce a information bottleneck and remove redundancy of skeletons. Furthermore, we theoretically demonstrate that our generative objective involves the contrastive learning objective which aligns the masked and noisy views. Meanwhile, it also enforces the representation to complement for the noisy view, leading to better generalization performance. MacDiff achieves state-of-the-art performance on representation learning benchmarks while maintaining the competence for generative tasks. Moreover, we leverage the diffusion model for data augmentation, significantly enhancing the fine-tuning performance in scenarios with scarce labeled data. Our project is available at https://lehongwu.github.io/ECCV24MacDiff/.

骨骼表示扩散模型自监督学习数据增强

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。