用生成式扩散模型提升短序列分类器的长程泛化能力
Generative Diffusion Prior Distillation for Long-Context Knowledge Transfer
- 将教师模型的完整特征视为被降级的观测,通过扩散模型重建缺失长程信息
- 在多个数据集和架构上,短序列分类准确率提升5.2%~8.7%
- 适合需低延迟、部分输入的工业时间序列分类场景
传统时间序列分类器依赖完整序列输入,但实际应用中受限于延迟与成本,常只能使用部分前缀。由于短序列缺乏类别判别模式,导致分类器泛化能力下降。本文提出生成式扩散先验蒸馏(GDPD),利用知识蒸馏使短序列分类器继承全序列模型的泛化能力。在传统蒸馏中,高容量教师模型为学生提供监督。当泛化差距源于训练数据差异(全序列与短序列)时,教师的全上下文特征对学生的短上下文特征而言可能过于复杂。为此,GDPD将学生特征视为目标教师特征的退化观测,借鉴扩散模型的迭代恢复能力,学习教师特征的生成先验。基于此先验,后验采样出能解释学生特征中缺失长程信息的目标教师表示,并优化学生特征使其尽可能接近这些目标。GDPD为每个学生特征提供一组与任务相关的长程知识分布,显著提升短序列分类性能。在多种早熟设置、数据集和架构上的实验表明,该方法在全序列到短序列蒸馏中具有显著有效性。
原文摘要 · Abstract (English)
While traditional time-series classifiers assume full sequences at inference, practical constraints (latency and cost) often limit inputs to partial prefixes. The absence of class-discriminative patterns in partial data can significantly hinder a classifier's ability to generalize. This work uses knowledge distillation (KD) to equip partial time series classifiers with the generalization ability of their full-sequence counterparts. In KD, high-capacity teacher transfers supervision to aid student learning on the target task. Matching with teacher features has shown promise in closing the generalization gap due to limited parameter capacity. However, when the generalization gap arises from training-data differences (full versus partial), the teacher's full-context features can be an overwhelming target signal for the student's short-context features. To provide progressive, diverse, and collective teacher supervision, we propose Generative Diffusion Prior Distillation (GDPD), a novel KD framework that treats short-context student features as degraded observations of the target full-context features. Inspired by the iterative restoration capability of diffusion models, we learn a diffusion-based generative prior over teacher features. Leveraging this prior, we posterior-sample target teacher representations that could best explain the missing long-range information in the student features and optimize the student features to be minimally degraded relative to these targets. GDPD provides each student feature with a distribution of task-relevant long-context knowledge, which benefits learning on the partial classification task. Extensive experiments across earliness settings, datasets, and architectures demonstrate GDPD's effectiveness for full-to-partial distillation.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。