跳到正文
OpenReview · State space models· Jiaqi Han; Jingwen Ye; Shunyu Liu; Haofei Zhang; Jie Song; Zunlei Feng; Mingli Song·· 2025-12-31精选AI 评分72

强化学习驱动的模型合并

Reinforced Model Merging

AI 导读

基于摘要分析。本文研究视觉与 NLP 模型的参数空间合并,针对统一处理所有参数可能导致性能退化、搜索式合并效率不足的问题,提出 Reinforced Model Merging(RMM),通过智能体与合并环境交互,搜索逐层合并方案。 机制上,RMM 将合并过程组织为逐层动作,并通过评估反馈引导搜索。作者说明,该过程不对原始模型进行梯度计算,并据此提出其在边缘设备上运行的可行性;这不等于搜索无需计算或评估数据,实际资源需求尚未验证。方法还使用数据子集进行评估,以缩短奖励反馈耗时。具体动作空间、奖励定义、搜索算法、子集采样方式及模型兼容条件,摘要未披露。 作者报告,RMM 在多种视觉与 NLP 数据集上达到当时最优表现,并在采用评估子集的设置下实现最高 100 倍加速;摘要未给出数据集名称、性能指标、对照配置、计时范围及硬件背景,不能据此判断普遍加速幅度或边缘部署成本。实验协议、消融及统计信息尚未验证。摘要提供了代码仓库地址,但代码内容与复现条件尚未核对。 平台推测(待验证):若多个 EEG 模型具备可兼容的参数结构,逐层合并搜索可能用于探索跨被试或跨会话模型的知识整合,以缓解统一合并权重忽略层间差异的问题。可考虑复用合并环境与搜索流程,但需改造任务评估、奖励及数据子集选择;低信噪比、被试差异、通道不一致和小数据可能使奖励不稳定,或使参数合并失效。一个可检验的小实验是,在同结构、同通道配置的 EEG 模型上,用独立验证集比较 RMM 与统一权重合并,并在未参与搜索的测试集上同时测量任务表现和总搜索耗时,检验子集奖励能否保留方案排序。已有 EEG 相关工作尚未检索确认。

阅读价值

值得核对 RMM 如何以逐层强化学习搜索替代统一参数合并,以及评估子集如何影响搜索效率与合并质量;其在 EEG 模型合并中的价值尚未验证。

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