arXiv:2601.04264cs.LGcs.CE2026-01被引 1

针对时间序列模型设计新型知识蒸馏方法,让小模型高效模仿大模型。

MemKD: Memory-Discrepancy Knowledge Distillation for Efficient Time Series Classification

  • 基于时序数据的内存差异设计新损失函数,捕捉师生模型记忆差异。
  • 参数量和内存使用减少约500倍,性能接近教师模型。
  • 适合可穿戴设备等资源受限场景部署,提升实时分析能力。

深度学习模型,尤其是循环神经网络及其变体(如长短期记忆网络),显著推进了时间序列数据分析。这些模型能捕捉时间序列中的复杂序列模式,实现实时评估。然而,其高计算复杂度和大模型尺寸在资源受限环境(如可穿戴设备和边缘计算平台)中部署面临挑战。知识蒸馏(KD)通过将大型复杂模型(教师)的知识迁移至小型高效模型(学生),在保持高性能的同时降低计算需求。现有KD方法最初为计算机视觉任务设计,忽视了时间序列模型独特的时序依赖性和记忆保留特性。为此,我们提出一种名为记忆差异知识蒸馏(MemKD)的新框架。MemKD利用专门设计的损失函数,捕捉教师与学生模型在时间序列子序列内的记忆保留差异,确保学生模型有效模仿教师模型的行为。该方法有助于开发紧凑且高性能的循环神经网络,适用于实时时间序列分析任务。大量实验表明,MemKD显著优于现有先进KD方法,在参数量和内存使用上减少约500倍的同时,性能与教师模型相当。

原文摘要 · Abstract (English)

Deep learning models, particularly recurrent neural networks and their variants, such as long short-term memory, have significantly advanced time series data analysis. These models capture complex, sequential patterns in time series, enabling real-time assessments. However, their high computational complexity and large model sizes pose challenges for deployment in resource-constrained environments, such as wearable devices and edge computing platforms. Knowledge Distillation (KD) offers a solution by transferring knowledge from a large, complex model (teacher) to a smaller, more efficient model (student), thereby retaining high performance while reducing computational demands. Current KD methods, originally designed for computer vision tasks, neglect the unique temporal dependencies and memory retention characteristics of time series models. To this end, we propose a novel KD framework termed Memory-Discrepancy Knowledge Distillation (MemKD). MemKD leverages a specialized loss function to capture memory retention discrepancies between the teacher and student models across subsequences within time series data, ensuring that the student model effectively mimics the teacher model's behaviour. This approach facilitates the development of compact, high-performing recurrent neural networks suitable for real-time, time series analysis tasks. Our extensive experiments demonstrate that MemKD significantly outperforms state-of-the-art KD methods. It reduces parameter size and memory usage by approximately 500 times while maintaining comparable performance to the teacher model.

知识蒸馏时间序列RNN边缘计算

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