用可微分稀疏核表示复杂卷积,提升移动端图像处理效率
DISK: Differentiable Sparse Kernel Complex for Efficient Spatially-Variant Convolution
- 将复杂卷积核分解为可优化的稀疏样本,支持端到端训练
- 在非凸核上实现更高保真度,计算成本低于低秩分解
- 无需重训即可实现空间变化滤波,适合实时渲染应用
图像卷积在摄影、科学成像和动画效果中至关重要,但直接使用密集卷积在资源受限设备上计算开销过大。现有近似方法如模拟退火或低秩分解,要么效率不足,要么无法捕捉非凸卷积核。本文提出一种可微分核分解框架,用一组稀疏核样本表示目标空间变化的密集复杂卷积核。该方法具备:(i) 支持稀疏核的可微优化;(ii) 针对非凸形状的专用初始化策略,避免陷入局部最优;(iii) 核空间插值方案,实现单核滤波向空间变化滤波的扩展,无需重新训练且无额外运行时开销。在高斯核与非凸核上的实验表明,本方法在保真度上优于模拟退火,计算成本显著低于低秩分解。该方法为移动成像和实时渲染提供了实用解决方案,同时保持完全可微性,便于集成进更广泛的学习流程。
原文摘要 · Abstract (English)
Image convolution with complex kernels is a fundamental operation in photography, scientific imaging, and animation effects, yet direct dense convolution is computationally prohibitive on resource-limited devices. Existing approximations, such as simulated annealing or low-rank decompositions, either lack efficiency or fail to capture non-convex kernels. We introduce a differentiable kernel decomposition framework that represents a target spatially-variant, dense, complex kernel using a set of sparse kernel samples. Our approach features (i) a decomposition that enables differentiable optimization of sparse kernels, (ii) a dedicated initialization strategy for non-convex shapes to avoid poor local minima, and (iii) a kernel-space interpolation scheme that extends single-kernel filtering to spatially varying filtering without retraining and additional runtime overhead. Experiments on Gaussian and non-convex kernels show that our method achieves higher fidelity than simulated annealing and significantly lower cost than low-rank decompositions. Our approach provides a practical solution for mobile imaging and real-time rendering, while remaining fully differentiable for integration into broader learning pipelines.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。