跳到正文
OpenReview · State space models· Almog Gueta; Elad Venezian; Colin Raffel; Noam Slonim; Yoav Katz; Leshem Choshen·· 2023-10-08精选AI 评分78

微调语言模型的知识是权重空间中的一个区域

Knowledge is a Region in Weight Space for Fine-tuned Language Models

AI 导读

基于摘要分析。本文研究微调语言模型之间的权重空间关系及其损失景观,核心贡献是提出并检验:高性能微调模型并非孤立分布,而是位于具有一定共同性能特征的权重区域,并利用区域中心选择后续微调的初始化模型。 作者报告,在同一数据集上微调的语言模型形成较紧密的权重空间簇;在同一底层任务的不同数据集上微调的模型形成较松散的簇。作者进一步报告,在模型之间的区域内移动可得到性能相当或更好的模型,且这一现象涉及原模型未针对其微调的任务。摘要将这些结果解释为:位于两个相似模型之间的模型可能兼具两者的知识;这一解释的适用范围仍需结合正文中的任务、参数操作与评估协议核对。 在后续微调实验中,作者报告,以区域中心而非预训练模型作为初始化,在所评估的12个数据集中有11个达到相当或更好的效果,平均 Accuracy 改善为3.06。摘要未明确该数值的单位、各数据集结果及平均方式,因此不能将其直接写成百分比或百分点。模型名称与规模、区域中心的构造、权重距离度量、训练设置、数据划分、消融及统计信息尚未验证。 平台推测(待验证):若多个 EEG 模型具有可对齐的参数结构及兼容的训练起点,权重区域中心可能成为跨被试或跨会话微调的候选初始化;可复用的是参数空间聚合与初始化选择思路,需改造的是 EEG 输入编码、通道对齐和任务输出。假设的主要风险包括低信噪比、小数据及被试差异使权重邻近不对应功能邻近。可用同一初始化训练若干源被试模型,构造中心后,在固定目标被试数据划分下与原始初始化比较,并核对目标测试数据未参与中心选择。原语言模型结果不构成 EEG/BCI 有效性证据,已有 EEG 相关工作尚未检索确认。

阅读价值

值得核对其微调模型的权重空间聚集现象及区域中心初始化策略,尤其是参数区域如何定义、模型如何对齐,以及初始化收益是否依赖任务与数据集选择。

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