AREAL-DTA: Dynamic Tree Attention for Efficient Reinforcement Learning of Large Language Models
AREAL-DTA:用于大语言模型高效强化学习的动态树注意力机制
机构 * University of Science and Technology of China(中国科学技术大学)
AI总结 针对RL后训练中rollout序列共享前缀导致计算冗余的问题,提出基于深度优先搜索的动态树注意力机制,结合负载均衡分布式批处理,实现高效前缀共享,训练吞吐量提升最高8.31倍。
Comments Accepted at ICML 2026. Camera-ready version. Code: https://github.com/areal-project/AReaL/tree/feat/dta