arXivDaily arXiv每日学术速递 周一至周五更新
北美退伍军人节(Veterans Day),12 日无更新
arXiv 2402.18508cs.LG
收藏

Orchid: 灵活且数据依赖的卷积用于序列建模

Orchid: Flexible and Data-Dependent Convolution for Sequence Modeling

Google Research(谷歌研究);School of Computer Science University of Waterloo(计算机科学学院)

Mahdi Karami, Ali Ghodsi

AI总结:

Orchid提出了一种灵活且数据依赖的卷积架构,用于提升序列建模的效率和可扩展性。

摘要:

在深度学习迅速发展的领域中,对能够表达性强且计算高效模型的需求从未如此迫切。本文介绍了一种名为Orchid的新架构,旨在解决传统注意力机制的二次复杂性问题,而不牺牲捕捉长距离依赖性和上下文学习的能力。该架构的核心是一种新的数据依赖全局卷积层,通过专用的条件神经网络根据输入序列上下文适配其内核。我们设计了两个简单的条件网络,以保持数据依赖卷积操作中的位移等变性。所提出的动态卷积内核使Orchid具有高表达性,同时保持长序列的近线性可扩展性。我们评估了该模型在多个领域,包括语言建模和图像分类,以突出其性能和通用性。我们的实验表明,这种架构不仅在较小的模型规模下优于传统的注意力架构,如BERT和视觉转换器,而且还能将可行的序列长度扩展到密集注意力层的限制之外。这一成就代表了朝着更高效和可扩展的深度学习模型迈出的重要一步。代码可在https://github.com/Karami-m/orchid获取。

英文摘要:

In the rapidly evolving field of deep learning, the demand for models that are both expressive and computationally efficient has never been more critical. This paper introduces Orchid, a novel architecture designed to address the quadratic complexity of traditional attention mechanisms without compromising the ability to capture long-range dependencies and in-context learning. At the core of this architecture lies a new data-dependent global convolution layer, which contextually adapts its kernel conditioned on input sequence using a dedicated conditioning neural network. We design two simple conditioning networks that maintain shift equivariance in our data-dependent convolution operation. The dynamic nature of the proposed convolution kernel grants Orchid high expressivity while maintaining quasilinear scalability for long sequences. We evaluate the proposed model across multiple domains, including language modeling and image classification, to highlight its performance and generality. Our experiments demonstrate that this architecture not only outperforms traditional attention-based architectures such as BERT and Vision Transformers with smaller model sizes, but also extends the feasible sequence length beyond the limitations of the dense attention layers. This achievement represents a significant step towards more efficient and scalable deep learning models for sequence modeling. The code is available at https://github.com/Karami-m/orchid.

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

↑