用教师模型知识生成伪标签,让边缘设备持续学习新数据
Semi-Supervised Online Learning on the Edge by Transforming Knowledge from Teacher Models
- 结合知识蒸馏与主动学习,用教师模型生成学生模型的伪标签
- 稳定教师模型下,学生模型可逼近理论最优性能
- 适合标签难获取或可用预训练模型的边缘场景
边缘机器学习(Edge ML)使模型能直接在分布于网络边缘的数据上训练。然而,现有方法多假设模型在中心端静态训练后部署,难以应对未见过的新数据。为此,本文提出在线边缘学习(Online Edge ML),允许模型在边缘设备上持续更新。核心挑战在于:如何为真正未来、未见的数据点确定标签?我们提出知识转化(KT)方法,融合知识蒸馏、主动学习与因果推理。其核心机制是:利用教师模型的知识生成伪标签,指导学生模型训练。通过两组模拟实验验证:使用不稳定的教师模型和相对稳定的教师模型。结果表明,当教师模型稳定时,学生模型最终可达到预期最大性能。该方法适用于两类场景:(1)教师任务通用,可直接使用预训练模型,无需从头训练;(2)学生任务标签获取困难或成本高。
原文摘要 · Abstract (English)
Edge machine learning (Edge ML) enables training ML models using the vast data distributed across network edges. However, many existing approaches assume static models trained centrally and then deployed, making them ineffective against unseen data. To address this, Online Edge ML allows models to be trained directly on edge devices and updated continuously with new data. This paper explores a key challenge of Online Edge ML: "How to determine labels for truly future, unseen data points". We propose Knowledge Transformation (KT), a hybrid method combining Knowledge Distillation, Active Learning, and causal reasoning. In short, KT acts as the oracle in active learning by transforming knowledge from a teacher model to generate pseudo-labels for training a student model. To verify the validity of the method, we conducted simulation experiments with two setups: (1) using a less stable teacher model and (2) a relatively more stable teacher model. Results indicate that when a stable teacher model is given, the student model can eventually reach its expected maximum performance. KT is potentially beneficial for scenarios that meet the following circumstances: (1) when the teacher's task is generic, which means existing pre-trained models might be adequate for its task, so there will be no need to train the teacher model from scratch; and/or (2) when the label for the student's task is difficult or expensive to acquire.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。