arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2306.16699cs.CVcs.AIcs.ARcs.LG

Rapid-INR:使用隐式神经表示的存储高效无CPU深度神经网络训练

Rapid-INR: Storage Efficient CPU-free DNN Training Using Implicit Neural Representation

School of Electrical and Computer Engineering, Georgia Institute of Technology(佐治亚理工学院电气与计算机工程学院)

Hanqiu Chen, Hang Yang, Stephen Fitzmeyer, Cong Hao

AI总结:

本文提出Rapid-INR,利用隐式神经表示将数据集直接存储于GPU以消除CPU通信开销,结合动态剪枝与逐层量化,在保持精度的同时大幅降低内存并提升训练速度。

摘要:

隐式神经表示(INR)是一种创新方法,用于表示复杂形状或物体,而无需显式定义其几何或表面结构。相反,INR将物体表示为连续函数。先前的研究已经证明了使用神经网络作为INR进行图像压缩的有效性,展示了与JPEG等传统方法相当的性能。然而,INR在图像压缩之外还具有各种应用的潜力。本文介绍了Rapid-INR,这是一种新颖的方法,利用INR对图像进行编码和压缩,从而加速计算机视觉任务中的神经网络训练。我们的方法涉及将整个数据集直接以INR格式存储在GPU上,从而减轻训练期间CPU和GPU之间显著的数据通信开销。此外,从INR到RGB格式的解码过程高度并行化并即时执行。为了进一步增强压缩,我们在先前工作的基础上提出了迭代动态剪枝以及逐层量化。我们在图像分类任务上评估了我们的框架,利用ResNet-18骨干网络和三个具有不同图像大小的常用数据集。Rapid-INR将内存消耗减少到仅约为RGB格式原始数据集大小的5%,并在准确率仅有微小下降的情况下,相比PyTorch训练流程实现了最大6倍加速,相比DALI训练流程实现了最大1.2倍加速。重要的是,Rapid-INR可以轻松应用于其他计算机视觉任务和骨干网络,只需合理的工程努力。我们的实现代码已在https://github.com/sharc-lab/Rapid-INR公开发布。

英文摘要:

Implicit Neural Representation (INR) is an innovative approach for representing complex shapes or objects without explicitly defining their geometry or surface structure. Instead, INR represents objects as continuous functions. Previous research has demonstrated the effectiveness of using neural networks as INR for image compression, showcasing comparable performance to traditional methods such as JPEG. However, INR holds potential for various applications beyond image compression. This paper introduces Rapid-INR, a novel approach that utilizes INR for encoding and compressing images, thereby accelerating neural network training in computer vision tasks. Our methodology involves storing the whole dataset directly in INR format on a GPU, mitigating the significant data communication overhead between the CPU and GPU during training. Additionally, the decoding process from INR to RGB format is highly parallelized and executed on-the-fly. To further enhance compression, we propose iterative and dynamic pruning, as well as layer-wise quantization, building upon previous work. We evaluate our framework on the image classification task, utilizing the ResNet-18 backbone network and three commonly used datasets with varying image sizes. Rapid-INR reduces memory consumption to only about 5% of the original dataset size in RGB format and achieves a maximum 6$\times$ speedup over the PyTorch training pipeline, as well as a maximum 1.2x speedup over the DALI training pipeline, with only a marginal decrease in accuracy. Importantly, Rapid-INR can be readily applied to other computer vision tasks and backbone networks with reasonable engineering efforts. Our implementation code is publicly available at https://github.com/sharc-lab/Rapid-INR.

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

↑