跳到正文
原文
arXiv cs.LG· Xiaojun Wu, Cehao Yang, Xuhui Jiang, Chengjin Xu, Jia Li, Jian Guo·· 16 小时前AI 评分35

FARO:基于跨块等风险贡献预算分配的函数空间自适应优化器

FARO: Function-Space Adaptive Return-to-Risk Optimization via Cross-Block Equal-Risk-Contribution Budget Allocation

AI 导读

新研究提出函数空间自适应优化器 FARO,以参数块为单元,利用等风险贡献(ERC)原则自适应分配更新预算。在 FineWeb 训练的 GPT 风格模型上,FARO-Adaptive 取得 3.2780 nats,持平手工设计的 NorMuonAndAdam(3.2794 nats);FARO-ERC 在 6 个种子下平均进一步降低 0.0136 nats 损失且不增加额外显存与耗时,代码已开源。

来源:arXiv cs.LG · arxiv.org