arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2107.06200math.NAcs.NA
收藏

多级混合精度迭代精化

Multistage Mixed Precision Iterative Refinement

Eda Oktay, Erin Carson

摘要:

低精度运算,特别是半精度浮点运算,现在在商业硬件中是可用的。使用较低的精度可以显著节省计算和通信成本,并按比例节省能源。受此启发,人们对求解线性方程组的混合精度迭代加细$Ax=b$重新产生了兴趣,并开发了基于GMRES的迭代加细的新变体。具有给定精度组合的每个特定变体导致用于后向误差和前向误差收敛的不同的基于条件数的约束,并且每个变体具有不同的性能成本。在文献中给出的收敛约束,作为分析的人工制品,在实践中往往过于严格,因此可能导致用户选择更昂贵的变体时,一个较便宜的就足够了。在这项工作中,我们开发了一个多级混合精度迭代精化求解器,其目的是结合现有的混合精度的方法,以平衡性能和精度,提高可用性。对于精度的初始组合,该算法从最便宜的方法开始,并通过迭代过程中产生的数量的廉价计算来监测收敛。如果使用特定的停止标准检测到缓慢收敛或发散,则算法切换到使用更昂贵但更可靠的变体。我们的方法的一个新颖的方面是,与现有的实现不同,我们的算法首先尝试使用“更强”的解决方案更新之前,诉诸于提高精度(S)。在某些情况下,这可以避免以更高精度重构矩阵的需要。我们进行了广泛的随机密集的问题和问题,从实际应用中证实的好处多级方法的数值实验。

英文摘要:

Low precision arithmetic, in particular half precision floating point arithmetic, is now available in commercial hardware. Using lower precision can offer significant savings in computation and communication costs with proportional savings in energy. Motivated by this, there has been a renewed interest in mixed precision iterative refinement for solving linear systems $Ax=b$, and new variants of GMRES-based iterative refinement have been developed. Each particular variant with a given combination of precisions leads to different condition number-based constraints for convergence of the backward and forward errors, and each has different performance costs. The constraints for convergence given in the literature are, as an artifact of the analyses, often overly strict in practice, and thus could lead a user to select a more expensive variant when a less expensive one would have sufficed. In this work, we develop a multistage mixed precision iterative refinement solver which aims to combine existing mixed precision approaches to balance performance and accuracy and improve usability. For an initial combination of precisions, the algorithm begins with the least expensive approach and convergence is monitored via inexpensive computations with quantities produced during the iteration. If slow convergence or divergence is detected using particular stopping criteria, the algorithm switches to use a more expensive, but more reliable variant. A novel aspect of our approach is that, unlike existing implementations, our algorithm first attempts to use ``stronger'' solvers for the solution update before resorting to increasing the precision(s). In some scenarios, this can avoid the need to refactorize the matrix in higher precision. We perform extensive numerical experiments on random dense problems and problems from real applications which confirm the benefits of the multistage approach.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑