Transformer 中的黄金时刻:多步任务揭示的 Softmax 引发的优化问题
原文中文,约500字,阅读约需2分钟。
📝
内容提要
该研究发现transformers在多步决策任务中存在学习困难,但通过修复自注意力模块中的Softmax函数,可以提高训练速度和准确度。改进后的模型只需完成训练步骤的20%即可达到基线模型的95%。
🏷️