arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2410.01201cs.LGcs.AI
收藏

RNNs曾是我们所需的全部吗?

Were RNNs All We Needed?

Mila – Université de Montréal(米拉-蒙特利尔大学);Borealis AI(博雷亚利斯人工智能公司)

Leo Feng, Frederick Tung, Mohamed Osama Ahmed, Yoshua Bengio, Hossein Hajimirsadeghi

AI总结:

针对Transformer的序列长度可扩展性限制,从历史视角重新研究RNNs(LSTMs和GRUs),通过简化得到参数更少、训练可并行化的minLSTMs和minGRUs,其在多任务上性能媲美Transformer等近期模型。

摘要:

2017年Transformer的引入重塑了深度学习的格局。Transformer最初被提出用于序列建模,此后在多个领域取得了广泛成功。然而,Transformer的可扩展性限制——尤其是在序列长度方面——引发了对新型循环模型的重新关注,这类模型在训练期间可并行化、性能相当且扩展更高效。在这项工作中,我们从历史视角重新审视序列建模,聚焦Transformer兴起前主导该领域二十年的循环神经网络(RNNs),具体研究长短期记忆网络(LSTMs,1997年)和门控循环单元(GRUs,2014年)。我们证明,通过简化这些模型,可得到精简版本(minLSTMs和minGRUs),它们(1)比传统版本参数更少;(2)训练期间完全可并行化;(3)在一系列任务上取得惊人的竞争力,可与包括Transformer在内的近期模型相媲美。

英文摘要:

The introduction of Transformers in 2017 reshaped the landscape of deep learning. Originally proposed for sequence modelling, Transformers have since achieved widespread success across various domains. However, the scalability limitations of Transformers - particularly with respect to sequence length - have sparked renewed interest in novel recurrent models that are parallelizable during training, offer comparable performance, and scale more effectively. In this work, we revisit sequence modelling from a historical perspective, focusing on Recurrent Neural Networks (RNNs), which dominated the field for two decades before the rise of Transformers. Specifically, we examine LSTMs (1997) and GRUs (2014). We demonstrate that by simplifying these models, we can derive minimal versions (minLSTMs and minGRUs) that (1) use fewer parameters than their traditional counterparts, (2) are fully parallelizable during training, and (3) achieve surprisingly competitive performance on a range of tasks, rivalling recent models including Transformers.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑