跳到正文
原文
arXiv · Statistics Machine Learning· Giordano Vitale, Giacomo Meanti, Marco Rando, Blanca Ayuso de Dios, Lorenzo Rosasco·· 3 小时前AI 评分24

BLOKS:用草图化块预条件子扩展核求解器

BLOKS: Scaling Kernel Solvers with Sketched Block Preconditioning

AI 导读

研究者提出 BLOKS,一种面向 Nyström 草图化核岭回归(KRR)的共轭梯度求解器,其预条件子结合块 Jacobi 近似与草图化。该预条件子的内存开销与数据集大小和参数数量均无关,并配有内存高效、GPU 加速的实现。在标准大规模核求解器基准数据集上,BLOKS 取得与最先进方法相当或更优的性能,同时保持有利的时间/内存复杂度。

来源:arXiv · Statistics Machine Learning · arxiv.org