arXiv:2410.03038cs.LGcs.CV2024-10被引 7

用置信度自适应蒸馏,让轻量模型更好继承复杂特征。

CPFD: Confidence-aware Privileged Feature Distillation for Short Video Classification

  • 根据教师模型置信度动态调整知识蒸馏权重,减少性能波动。
  • 相比纯端到端模型,视频分类F1提升6.76%,性能差距缩小84.6%。
  • 适合工业场景中高效融合历史复杂特征的模型部署需求。

针对短视频分类中定制化密集特征计算开销大、难以在线推理的问题,本文提出置信度感知的特权特征蒸馏方法(CPFD)。该方法在训练阶段自适应地将密集特征(如DF-X-VLM)的知识注入轻量级多模态模型(X-VLM),通过教师模型输出的置信度分数调节蒸馏权重,缓解不同业务场景下的性能不稳问题。在五个多样化任务上,相较于端到端模型X-VLM,CPFD平均提升F1分数6.76%,相比传统蒸馏方法(PFD)提升2.31%;同时性能差距缩小84.6%,结果接近教师模型。在线实验验证了其有效性,已在十余个生产模型中部署。

原文摘要 · Abstract (English)

Dense features, customized for different business scenarios, are essential in short video classification. However, their complexity, specific adaptation requirements, and high computational costs make them resource-intensive and less accessible during online inference. Consequently, these dense features are categorized as `Privileged Dense Features'.Meanwhile, end-to-end multi-modal models have shown promising results in numerous computer vision tasks. In industrial applications, prioritizing end-to-end multi-modal features, can enhance efficiency but often leads to the loss of valuable information from historical privileged dense features. To integrate both features while maintaining efficiency and manageable resource costs, we present Confidence-aware Privileged Feature Distillation (CPFD), which empowers features of an end-to-end multi-modal model by adaptively distilling privileged features during training. Unlike existing privileged feature distillation (PFD) methods, which apply uniform weights to all instances during distillation, potentially causing unstable performance across different business scenarios and a notable performance gap between teacher model (Dense Feature enhanced multimodal-model DF-X-VLM) and student model (multimodal-model only X-VLM), our CPFD leverages confidence scores derived from the teacher model to adaptively mitigate the performance variance with the student model. We conducted extensive offline experiments on five diverse tasks demonstrating that CPFD improves the video classification F1 score by 6.76% compared with end-to-end multimodal-model (X-VLM) and by 2.31% with vanilla PFD on-average. And it reduces the performance gap by 84.6% and achieves results comparable to teacher model DF-X-VLM. The effectiveness of CPFD is further substantiated by online experiments, and our framework has been deployed in production systems for over a dozen models.

视频分类知识蒸馏多模态

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