跳到正文
原文
arXiv 机器学习· Chuansai Zhou, Yanlin Liu, Bingning Huang, Shidi Xi, Chao Huang, Boxing Chen, Mingming Zhu, Guanghuan Fang, Ben Feng, Zongzuo Wang·· 5 小时前AI 评分39

用于昇腾 Triton 算子生成的压缩感知强化学习 ATRL

Compaction-Aware Reinforcement Learning For Ascend Triton Kernel Generation

AI 导读

研究人员提出用于长流程昇腾 Triton 算子生成的后训练方法 ATRL,将 GRPO 优化为压缩感知的片段级目标,以解决多轮交互上下文超出有效窗口以及终端奖励稀疏的问题。

来源:arXiv 机器学习 · arxiv.org