首页 > AI前沿 > FlashDiffusion: Fused Tiled Kernel Spectral Decomposition

FlashDiffusion: Fused Tiled Kernel Spectral Decomposition

arXiv机器学习 2026-10-01 12:00 7 阅读 查看原文

Diffusion maps, and kernel methods more generally, provide an interpretable nonlinear spectral representation basis for geometric learning.

In the geometric limit, small bandwidth, these matrices tend to be high rank and thus require materializing dense Gaussian kernels requires $O(N^2)$ memory.

We introduce FlashDiffusion, a matrix-free method that evaluates dense Gaussian kernel blocks in fused GPU tiles and couples the eigensolver to an empirical $\beta$-flow that selects the finite-sample resolution scale.

A continuation over sample size and bandwidth warm-starts increasingly expensive spectral solves from coarser resolutions.