跳到正文
OpenReview · State space models· Jonathan Kahana; Eliahu Horwitz; Imri Shuval; Yedid Hoshen·· 2024-01-01精选AI 评分74

用于权重空间学习的深层线性探针生成器

Deep Linear Probe Generators for Weight Space Learning

AI 导读

基于摘要分析。本文研究权重空间学习,即从神经网络中提取其训练数据集、泛化误差等信息,提出 Deep Linear Probe Generators(ProbeGen),通过共享的深层线性生成器构造探针,以改善探针学习并降低过拟合。 直接对模型权重建模面临高维参数及神经元置换对称性等挑战。另一条路径是 Probing:将一组学习得到的输入送入待分析模型,以对应输出表示该模型,再训练预测器提取目标信息。作者报告,初步实验中的普通 Probing 基线表现较好,但现有探针学习策略效果不佳。ProbeGen 在这一流程中加入共享生成器,以深层线性架构提供偏向结构化探针的归纳偏置;摘要未说明生成器结构、探针数量、损失函数及具体训练流程。 作者报告,ProbeGen 的表现显著优于其比较的当前最优方法,且相对其他领先方法所需 FLOPs 少 30 至 1000 倍。摘要未提供这些比较对应的任务、模型集合、数据划分、基线名称、性能指标与 FLOPs 核算范围,因此不能据此判断优势适用于哪些设置。实验协议、消融及统计信息尚未验证。 平台推测(待验证):该方法可能用于分析已训练 EEG 解码模型的泛化误差,而非直接解码 EEG。其对应瓶颈是模型质量评估;可复用探针—模型输出—预测器流程,但需将探针适配为模型可接受的通道与时间输入,并建立带有真实泛化误差标签的模型集合。假设该表示能捕捉泛化相关差异,而非仅识别架构或训练配置,可在固定架构的 EEG 模型集合上比较普通 Probing 与 ProbeGen 对跨被试泛化误差的预测能力,并按被试来源隔离模型集合。低信噪比、通道差异及可用模型数量不足可能使该假设失效;已有 EEG 相关工作尚未检索确认。

阅读价值

值得核对 ProbeGen 如何以深层线性生成器约束探针学习,以及其相对直接权重建模的性能与计算优势是否在可比协议下成立;对 EEG 模型分析的适用性尚未验证。

来源:OpenReview · State space models · openreview.net