arXiv stat.ML· Giordano Vitale, Giacomo Meanti, Marco Rando, Blanca Ayuso de Dios, Lorenzo Rosasco·· 2 小时前AI 评分20
BLOKS:用草图块预条件扩展核求解器
BLOKS: Scaling Kernel Solvers with Sketched Block Preconditioning
AI 导读
BLOKS 是一种基于共轭梯度的 Nyström 草图核岭回归(KRR)求解器,其新型预条件器将块 Jacobi 近似与草图结合,内存开销与数据集规模和参数数量均无关。该实现经 GPU 加速,在核求解器基准测试的标准大规模数据集上取得与 SOTA 方法相当或更优的表现。
来源:arXiv stat.ML · arxiv.org