跳到正文
OpenReview · State space models· Zixuan Hu; Xiaotong Li; Shixiang Tang; Jun Liu; Yichun Hu; Ling-Yu Duan·· 2024-01-01精选AI 评分77

LEAD:探索用于模型选择的 Logit 空间演化

LEAD: Exploring Logit Space Evolution for Model Selection

AI 导读

基于摘要分析。本文研究视觉任务中如何高效选择适合下游任务的预训练模型,提出 LEAD:通过建模网络输出 logit 在微调过程中的非线性演化,预测模型迁移性,避免对候选模型逐一执行完整微调。主要贡献是将模型选择依据与微调优化目标对齐,并引入类别感知的演化分解。 机制方面,作者指出,已有方法常在特征空间中以线性变换近似微调动力学,可能难以充分对应微调目标及优化过程的非线性。LEAD 转向 logit 空间,建立优化过程的理论框架,并推导常微分方程(ODE)描述向最终 logit 状态的演化;类别感知分解用于考虑不同类别的演化差异。作者称该方法可用单步近似跨越优化差距,从而绕过耗时的完整微调。具体微分方程、近似求解方式、监督信息需求及计算复杂度尚未验证。 实验方面,作者报告在 24 个监督与自监督预训练模型、10 个下游数据集上进行了评估,并称方法表现良好,在低数据情境中仍具有适应性。摘要未提供具体数据集、评价指标、数值结果、划分协议或对照方法,故尚不能量化其相对优势;实验协议、消融及统计信息尚未验证。 平台推测(待验证):若 EEG 下游任务具备多个输出类别相容的候选预训练模型及少量标注数据,logit 演化近似可能用于降低模型筛选成本,对应小数据条件下反复微调代价较高的问题。可复用的是候选模型迁移性评估思路,需改造或核对 EEG 输入适配、类别映射及微调设置。低信噪比、跨被试分布差异、通道不一致和小样本估计不稳定,都可能使单步近似失效。一个可检验的小实验是在固定 Cross-subject 划分与标注预算下,比较 LEAD 给出的模型排序与统一协议下实际微调所得排序的一致性,并记录筛选成本。以上并非已证实的 EEG/BCI 效果,已有 EEG 相关工作尚未检索确认。

阅读价值

LEAD 将预训练模型选择与 logit 空间中的非线性微调动力学联系起来,值得核对其迁移性预测与实际微调表现的一致性,以及单步近似的计算成本和适用边界。

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