用动态对齐学习让小模型学大模型的人类视觉能力
Dynamic Pattern Alignment Learning for Pretraining Lightweight Human-Centric Vision Models
- 设计动态模式解码器,按需提取身份、形状、互动三类视觉模式
- 在图像、像素、实例关系三层实现对齐,缩小轻量模型与大模型差距
- 500万参数小模型性能媲美3070万参数大模型,适合移动端部署
人中心视觉模型(HVMs)因大规模人物图像预训练而具备强大泛化能力,但其依赖大型神经网络架构且预训练数据获取受限,制约了实际应用。为此,本文提出动态模式对齐学习(DPAL),一种基于知识蒸馏的预训练框架,可高效训练轻量化HVMs以从大型HVMs中习得强泛化能力。人中心视觉感知高度依赖全局身份、局部形状和多人互动三类典型视觉模式。为此,我们设计动态模式解码器(D-PaDe),作为条件式专家混合(MoE)模型,根据输入图像和模式查询自适应提取三类模式。同时提出三层对齐目标:图像级、像素级和实例关系级,旨在最小化轻量模型与大型模型间的泛化差距。大量实验在15个挑战性数据集上验证有效性。当以PATH-B为教师模型时,DPAL-ViT/Ti(500万参数)的泛化性能接近现有大型HVMs如PATH-B(8400万参数)和Sapiens-L(3.07亿参数),显著优于Proteus-ViT/Ti(500万参数)和TinyMiM-ViT/Ti(500万参数)等先前蒸馏方法。
原文摘要 · Abstract (English)
Human-centric vision models (HVMs) have achieved remarkable generalization due to large-scale pretraining on massive person images. However, their dependence on large neural architectures and the restricted accessibility of pretraining data significantly limits their practicality in real-world applications. To address this limitation, we propose Dynamic Pattern Alignment Learning (DPAL), a novel distillation-based pretraining framework that efficiently trains lightweight HVMs to acquire strong generalization from large HVMs. In particular, human-centric visual perception are highly dependent on three typical visual patterns, including global identity pattern, local shape pattern and multi-person interaction pattern. To achieve generalizable lightweight HVMs, we firstly design a dynamic pattern decoder (D-PaDe), acting as a dynamic Mixture of Expert (MoE) model. It incorporates three specialized experts dedicated to adaptively extract typical visual patterns, conditioned on both input image and pattern queries. And then, we present three levels of alignment objectives, which aims to minimize generalization gap between lightweight HVMs and large HVMs at global image level, local pixel level, and instance relation level. With these two deliberate designs, the DPAL effectively guides lightweight model to learn all typical human visual patterns from large HVMs, which can generalize to various human-centric vision tasks. Extensive experiments conducted on 15 challenging datasets demonstrate the effectiveness of the DPAL. Remarkably, when employing PATH-B as the teacher, DPAL-ViT/Ti (5M parameters) achieves surprising generalizability similar to existing large HVMs such as PATH-B (84M) and Sapiens-L (307M), and outperforms previous distillation-based pretraining methods including Proteus-ViT/Ti (5M) and TinyMiM-ViT/Ti (5M) by a large margin.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。