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