通过语言引导知识蒸馏进行轻量模型预训练
Lightweight Model Pre-training via Language Guided Knowledge Distillation
基于摘要分析。本文研究面向移动设备的小型视觉模型预训练,提出 Language-Guided Distillation(LGD),利用目标下游任务的类别名称,引导师生蒸馏保留与类别辨别相关的知识。原论文的主要对象是视觉表征及分类、检测、分割任务,并非 EEG 或 BCI。 机制上,LGD 使用预训练文本编码器提取类别名称的语义嵌入,建立 Textual Semantics Bank(TSB);再通过 Language-Guided Knowledge Aggregation(LGKA)模块构建 Visual Semantics Bank(VSB)。学生编码器通过模仿教师在 TSB 和 VSB 上推断的相似度分数分布,学习任务相关知识。其与通用自监督蒸馏的核心区别,是将目标任务的类别语义显式引入知识筛选与传递过程;具体损失形式、语义库构建细节及训练流程尚未验证。 作者报告,相较于通过 ImageNet 预训练或自监督蒸馏得到的小模型,LGD 蒸馏的轻量模型在所评估的分类、检测和分割任务上达到领先表现。摘要未提供具体数据集、指标、数值及比较协议,因此不能据此判断收益幅度或跨任务稳定性。实验协议、消融及统计信息尚未验证。作者提供了代码地址,但代码完整性与复现条件尚未核验。 平台推测(待验证):若 EEG 教师模型的表征与任务类别语义存在可学习对应关系,类别文本约束可能帮助轻量学生保留判别信息。可复用的是语义库与相似度分布蒸馏思路;视觉编码器及 LGKA 的输入适配则需针对 EEG 改造。该假设依赖有意义的类别描述及足够的信号—语义对齐数据,可能受 EEG 低信噪比、跨被试差异、通道配置变化与小样本限制而失败。一个可证伪的小实验是在固定跨被试划分、师生模型与训练预算下,对比普通蒸馏、真实类别文本引导和随机类别文本引导;若真实文本不能稳定优于两项对照,则不支持语义引导的迁移收益。已有 EEG 相关工作尚未检索确认。
值得核对的是 LGD 如何利用目标任务类别名称引导师生蒸馏,以及这种任务相关语义约束相对通用自监督蒸馏的收益;其对 EEG 的适用性尚未验证。
来源:OpenReview · State space models · openreview.net