arXiv:2506.05617cs.LGcs.AI2025-06

提出O(N)复杂度方法,高效计算卷积层奇异值

LFA applied to CNNs: Efficient Singular Value Decomposition of Convolutional Mappings by Local Fourier Analysis

  • 基于局部傅里叶分析与卷积平移不变性,实现快速奇异值分解
  • 相比FFT方法,计算复杂度从O(N log N)降至O(N),可处理高维卷积映射
  • 适合需要精确奇异值的模型压缩、泛化分析等场景

卷积映射的奇异值蕴含重要谱特性,可用于提升CNN的泛化能力与鲁棒性,并支持模型压缩。然而,奇异值计算通常资源消耗巨大:传统方法需将卷积展开为大型稀疏矩阵,受硬件限制难以精确求解,尤其在输入尺寸大或通道数多时更难处理。现有高效方法利用快速傅里叶变换(FFT)将卷积映射转换至频域,可对固定通道数的卷积计算N个奇异值,复杂度为O(N log N)。本文提出一种基于局部傅里叶分析的新方法,利用卷积算子的平移不变性,实现复杂度O(N)的奇异值分解。我们提供了算法运行时间的理论分析,并通过数值实验验证其效率。结果表明,该方法具有良好的可扩展性,能实际计算高维卷积映射的全部奇异值及对应奇异向量。

原文摘要 · Abstract (English)

The singular values of convolutional mappings encode interesting spectral properties, which can be used, e.g., to improve generalization and robustness of convolutional neural networks as well as to facilitate model compression. However, the computation of singular values is typically very resource-intensive. The naive approach involves unrolling the convolutional mapping along the input and channel dimensions into a large and sparse two-dimensional matrix, making the exact calculation of all singular values infeasible due to hardware limitations. In particular, this is true for matrices that represent convolutional mappings with large inputs and a high number of channels. Existing efficient methods leverage the Fast Fourier transformation (FFT) to transform convolutional mappings into the frequency domain, enabling the computation of singular values for matrices representing convolutions with larger input and channel dimensions. For a constant number of channels in a given convolution, an FFT can compute N singular values in O(N log N) complexity. In this work, we propose an approach of complexity O(N) based on local Fourier analysis, which additionally exploits the shift invariance of convolutional operators. We provide a theoretical analysis of our algorithm's runtime and validate its efficiency through numerical experiments. Our results demonstrate that our proposed method is scalable and offers a practical solution to calculate the entire set of singular values - along with the corresponding singular vectors if needed - for high-dimensional convolutional mappings.

奇异值分解卷积网络高效算法傅里叶分析

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。