卷积神经网络(CNN)已经在涉及二维(2D)数据的任务中发现了许多应用,诸如图像分类和图像处理。因此,2D卷积层在CPU和GPU上进行了大量优化。然而,在许多应用中-例如基因组学和语音识别,数据可以是一维(1D)的。这些应用可以从优化的1D卷积层中受益。在这项工作中,我们介绍了一个通用的1D卷积层,涵盖了广泛的参数,我们的有效实现。它针对x86 CPU架构进行了优化,特别是包含Intel AVX-512和AVX-512 BFloat 16指令的架构。我们使用LIBXSMM库的批量减少通用矩阵乘法(BRGEMM)内核来实现FP 32和BFloat 16精度。我们证明了我们的实施可以在英特尔至强Cascade Lake和Cooper Lake CPU上实现高达80%的效率。此外,我们展示了我们的BRGEMM为基础的方法的泛化能力,通过在一系列参数实现高效率。对于不同的输入张量宽度、滤波器宽度、通道数、滤波器和膨胀参数,我们始终实现比使用英特尔oneDNN库后端的1D卷积层更高的效率。最后,我们通过将其用于具有真实基因组数据集的端到端神经网络训练来展示我们优化的1D卷积层的性能,并在Cascade Lake CPU上实现了高达6.86倍的基于oneDNN库的加速比。我们还演示了16个插槽的Cascade/Cooper Lake CPU的扩展,并使用类似的功率包络在8个V100 GPU上实现了显著的加速。在端到端训练中,我们在Cascade Lake上使用FP 32获得了1.41倍的加速比,在Cooper Lake上使用FP 32获得了1.57倍的加速比,在Cooper Lake上使用BFloat 16获得了2.27倍的加速比。
英文摘要:
Convolutional neural networks (CNNs) have found many applications in tasks involving two-dimensional (2D) data, such as image classification and image processing. Therefore, 2D convolution layers have been heavily optimized on CPUs and GPUs. However, in many applications - for example genomics and speech recognition, the data can be one-dimensional (1D). Such applications can benefit from optimized 1D convolution layers. In this work, we introduce our efficient implementation of a generic 1D convolution layer covering a wide range of parameters. It is optimized for x86 CPU architectures, in particular, for architectures containing Intel AVX-512 and AVX-512 BFloat16 instructions. We use the LIBXSMM library's batch-reduce General Matrix Multiplication (BRGEMM) kernel for FP32 and BFloat16 precision. We demonstrate that our implementation can achieve up to 80% efficiency on Intel Xeon Cascade Lake and Cooper Lake CPUs. Additionally, we show the generalization capability of our BRGEMM based approach by achieving high efficiency across a range of parameters. We consistently achieve higher efficiency than the 1D convolution layer with Intel oneDNN library backend for varying input tensor widths, filter widths, number of channels, filters, and dilation parameters. Finally, we demonstrate the performance of our optimized 1D convolution layer by utilizing it in the end-to-end neural network training with real genomics datasets and achieve up to 6.86x speedup over the oneDNN library-based implementation on Cascade Lake CPUs. We also demonstrate the scaling with 16 sockets of Cascade/Cooper Lake CPUs and achieve significant speedup over eight V100 GPUs using a similar power envelop. In the end-to-end training, we get a speedup of 1.41x on Cascade Lake with FP32, 1.57x on Cooper Lake with FP32, and 2.27x on Cooper Lake with BFloat16 over eight V100 GPUs with FP32.
Learning Delays in Spiking Neural Networks using Dilated Convolutions with Learnable Spacings
CerCo UMR 5549(塞尔科UMR 5549机构);CNRS(法国国家科学研究中心);Université de Toulouse(图卢兹大学);Artificial and Natural Intelligence Toulouse Institute (ANITI)(图卢兹人工与自然智能研究所)
Enhancing Satellite Object Localization with Dilated Convolutions and Attention-aided Spatial Pooling
Department of Information Systems, University of Maryland, Baltimore County(信息系统系,马里兰大学巴尔的摩分校);Department of Physics, Catholic University of America(物理系,天主教美国大学);NASA Goddard Space Flight Center(NASA戈达德空间飞行中心)
Multi-Scale Dilated Convolution Network for Long-Term Time Series Forecasting
National Key Laboratory for Novel Software Technology(现代软件工程国家重点实验室);Nanjing University(南京大学);School of Artificial Intelligence(人工智能学院);School of Electronic Science and Engineering(电子科学与工程学院)
An Order-Invariant and Interpretable Hierarchical Dilated Convolution Neural Network for Chemical Fault Detection and Diagnosis
Zhejiang University(浙江大学);University of Illinois Urbana–Champaign Institute(伊利诺伊大学厄巴纳-香槟分校学院);University of Electronic Science and Technology of China(电子科技大学)
DeepVol: Volatility Forecasting from High-Frequency Data with Dilated Causal Convolutions
Oxford-Man Institute of Quantitative Finance, University of Oxford(牛津大学牛津-曼量化金融研究院);Universidad Carlos III de Madrid(马德里卡洛斯三世大学);University of Oxford(牛津大学)
Hybrid Data Augmentation and Deep Attention-based Dilated Convolutional-Recurrent Neural Networks for Speech Emotion Recognition
Institute for Computational Science, Ton Duc Thang University(孙德胜大学计算科学研究所);Faculty of Electrical and Electronics Engineering, Ton Duc Thang University(孙德胜大学电气与电子工程学院);School of Graduate Studies, Ton Duc Thang University(孙德胜大学研究生院)
Boosting Dilated Convolutional Networks with Mixed Tensor Decompositions
Institute for Advanced Study(高等研究院);The Hebrew University of Jerusalem(希伯来大学耶路撒冷分校);cs.huji.ac.il(胡西学院计算机科学系(注:此处按要求不应提取此类非机构性内容,实际正确机构应为前三个,但原文本中cs.huji.ac.il为作者邮箱后缀,非机构,故有效机构仅前三者中重复的前两个,经核对原文HTML中的\textbf{coltauthor}后跟随的机构为Institute for Advanced Study,随后重复出现The Hebrew University of Jerusalem两次,而作者的Affiliation行中未明确列出具体机构名,仅邮箱后缀,故根据提取范围,仅提取明确出现的机构,即前两个重复的The Hebrew University of Jerusalem和Institute for Advanced Study,但严格按原文HTML开头文本中的机构描述,coltauthor部分的机构应为:Institute for Advanced Study,The Hebrew University of Jerusalem,The Hebrew University of Jerusalem,后续作者的Affiliation字段未提供机构名,故最终有效机构为:高等研究院、希伯来大学耶路撒冷分校(重复两次合并为一个)。但根据输出要求,需严格按原文明确出现的机构,原文中coltauthor后有三个机构:Institute for Advanced Study,The Hebrew University of Jerusalem,The Hebrew University of Jerusalem,因此正确输出应为:{)
Scale-Equivariant Generative Forecasting: Weight-Tied Dilated Convolutions, Wavelet Scattering Inputs, and Spectral-Consistency Training for Self-Similar Time Series
Attentive Dilated Convolution for Automatic Sleep Staging using Force-directed Layout
Department of Electrical and Electronic Engineering, BRAC University(BRAC大学电气与电子工程系);Department of Biomedical Engineering, Linköping University(林雪平大学生物医学工程系);Department of Electrical and Computer Engineering, Texas A&M University(德克萨斯农工大学电气与计算机工程系)