May, 2024

大矩阵的函数梯度

TL;DR通过导出 Lanczos 和 Arnoldi 迭代的未知共轭系统,该研究利用 JAX 实现了用于科学机器学习模型的数值线性代数效率高的可微分算法,该算法成功地不需要特定问题的代码优化,并在不同领域如选取高斯过程模型和校准贝叶斯神经网络中表现出与其他常见方法相当甚至更好的效果。