跳到正文
OpenReview · State space models· Logan Engstrom; Andrew Ilyas; Benjamin Chen; Axel Feldmann; William S. Moses; Aleksander Madry·· 2025-09-19精选AI 评分75

机器学习模型训练的元优化

Meta-Optimizing ML Model Training

AI 导读

基于摘要分析。本文研究大规模机器学习模型的训练配置优化:如何在庞大的训练设计空间中找到更好的训练方案。核心贡献是提出可在大规模场景下高效计算 metagradients(贯穿模型训练过程的梯度)的算法,并结合“smooth model training”框架,使训练过程本身能够接受梯度优化。 方法上,作者使用 metagradient descent(MGD)优化训练设置,而不只是优化模型参数。摘要明确区分了两项工作:高效计算贯穿训练过程的梯度,以及通过平滑训练框架使这些梯度能够有效用于优化。但具体可优化变量、平滑机制、梯度计算方式、训练与元优化目标,以及时间和内存成本均尚未验证,不能据此推断其使用特定网络结构或近似算法。 作者报告,MGD 在数据集选择任务上显著优于已有方法,并在降低准确率的数据投毒攻击方面较已有攻击方法取得一个数量级的优势。摘要未提供对应数据集、模型、数据划分、对照方法或量化指标,因此该数量级表述不能解读为准确率变化,也不能直接用于比较不同实验协议;实验协议、消融及统计信息尚未验证。 平台推测(待验证):若 EEG 训练中的样本权重等选择能够连续参数化,且训练过程允许稳定计算 metagradients,该机制可能用于探索训练数据选择,以应对跨被试数据质量和分布差异。可复用的是训练过程求梯度与元优化思路,需要改造的是 EEG 数据选择变量、元目标及被试隔离的验证协议。低信噪比、小样本和被试差异可能使元梯度不稳定,或使配置过度适应少数验证被试。一个可检验的小实验是在固定 EEG 模型与跨被试划分下,以训练被试内独立验证集优化连续样本权重,与均匀权重及相同计算预算的配置搜索比较,再在未参与元优化的被试上评估。该建议不是论文已验证的结果,已有 EEG 相关工作尚未检索确认。

阅读价值

值得阅读的具体原因是其将训练配置搜索转化为可通过训练过程求梯度的优化问题,并以数据集选择与数据投毒展示用途,但计算开销、优化稳定性及 EEG 适用性尚未验证。

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