arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2408.09453cs.LGcs.CV

用于长序列建模的重参数化多分辨率卷积

Reparameterized Multi-Resolution Convolutions for Long Sequence Modelling

University College London(伦敦大学学院);Amazon(亚马逊)

Harry Jake Cunningham, Giorgio Giannone, Mingtian Zhang, Marc Peter Deisenroth

AI总结:

针对长卷积训练难、易过拟合的问题,提出融合多分辨率卷积、结构重参数化与可学习核衰减的$\texttt{MRConv}$,在多任务上取得SOTA并提升ImageNet分类性能。

摘要:

全局卷积作为功能强大的通用序列模型,前景日益广阔。然而,训练长卷积颇具挑战,且核参数化方法必须能够在不过拟合的前提下学习长程依赖关系。本研究提出重参数化多分辨率卷积($\texttt{MRConv}$),这是一种为长序列建模对全局卷积核进行参数化的全新方法。通过利用多分辨率卷积、引入结构重参数化以及加入可学习的核衰减机制,$\texttt{MRConv}$ 能够学习到表达力强的长程卷积核,在多种数据模态下均表现出色。我们的实验表明,在卷积模型和线性时间 Transformer 中,$\texttt{MRConv}$ 在 Long Range Arena、Sequential CIFAR 以及 Speech Commands 任务上取得了最优性能。此外,我们通过用 1D $\texttt{MRConv}$ 层替换 2D 卷积,在 ImageNet 分类任务上实现了性能提升。

英文摘要:

Global convolutions have shown increasing promise as powerful general-purpose sequence models. However, training long convolutions is challenging, and kernel parameterizations must be able to learn long-range dependencies without overfitting. This work introduces reparameterized multi-resolution convolutions ($\texttt{MRConv}$), a novel approach to parameterizing global convolutional kernels for long-sequence modelling. By leveraging multi-resolution convolutions, incorporating structural reparameterization and introducing learnable kernel decay, $\texttt{MRConv}$ learns expressive long-range kernels that perform well across various data modalities. Our experiments demonstrate state-of-the-art performance on the Long Range Arena, Sequential CIFAR, and Speech Commands tasks among convolution models and linear-time transformers. Moreover, we report improved performance on ImageNet classification by replacing 2D convolutions with 1D $\texttt{MRConv}$ layers.

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

↑