[论文解读] A Robust Approximation to a Lambert-Type Function

K. Roberts|arXiv (Cornell University)|Apr 8, 2015

Sports Dynamics and Biomechanics参考文献 3被引用 8

一句话总结

本文提出了一种鲁棒的、与语言无关的算法，用于计算函数 $ y = \log(W(e^x)) $，其中 $ W $ 为Lambert W函数，采用自适应初始估计的Halley方法。该方法避免了数值下溢/上溢问题，在所有可表示的实数 $ x $ 上均能可靠运行，仅需几次迭代即可实现高精度。

ABSTRACT

The function $y = g(x) = \mathrm{log}\big(W(e^x)\big)$, where $W()$ denotes the Lambert W function, is the solution to the equation $y + e^y = x$. It appears in various problem situations, for instance the calculation of current-voltage curves for solar cells. A direct calculation of $g(x)$ may be inaccurate because of arithmetic underflow or overflow. We present a simple algorithm for calculating $g(x)$ that is robust, in that it will work for almost all $x$ values which are representable in the arithmetic of one's chosen computer language. The algorithm does not assume that the chosen computer language implements the Lambert W function.

研究动机与目标

• 解决在浮点数运算中因下溢或上溢导致计算 $ y = \log(W(e^x)) $ 时出现的数值不稳定性问题。
• 为任意缺乏原生Lambert W函数实现的编程语言提供一种可靠的该函数求值方法。
• 确保在整个实数域 $ x $ 上具有高精度和高稳定性，尤其适用于极端值情况。
• 开发一种计算高效且数值鲁棒的算法，适用于太阳能电池建模等应用场景。

提出的方法

• 将函数 $ y = \log(W(e^x)) $ 重述为求解方程 $ y + e^y = x $，并通过Halley方法迭代求解。
• 初始估计值选择为：当 $ x \leq -e $ 时取 $ y_0 = x $，当 $ x \geq e $ 时取 $ y_0 = \log(x) $，当 $ -e < x < e $ 时采用线性插值。
• 使用导数 $ h'(y) = 1 + e^y $ 和 $ h''(y) = e^y $ 应用Halley方法，其迭代公式由通用Halley更新公式推导得出。
• 每轮迭代仅计算一次 $ e^y $，并通过收敛准则或固定迭代次数确保精度。
• 该方法通过使更新公式分母中第一项占主导地位，有效减少取消误差，从而提升稳定性。
• 即使对于 $ x $ 值高达 $ \pm 10^6 $ 的情况，该方法仍表现稳健，四次迭代后误差低于 $ 10^{-80} $。

实验结果

研究问题

• RQ1 当直接求值可能导致下溢或上溢时，如何鲁棒地计算函数 $ y = \log(W(e^x)) $？
• RQ2 何种初始估计策略可确保在实数轴全范围内快速收敛于该函数的迭代求解器？
• RQ3 Halley方法能否被有效应用于求解 $ y + e^y = x $，以实现高精度且计算成本最低？
• RQ4 如何在任意编程语言中实现该算法，而无需依赖内置的Lambert W函数？

主要发现

• 该算法在仅使用Halley方法四次迭代的情况下，对 $ x \in [-10^6, 10^6] $ 的区间内实现了绝对误差低于 $ 10^{-80} $。
• 函数 $ g(x) = \log(W(e^x)) $ 严格递增且为凹函数，当 $ x $ 取绝对值较大的负值时其行为渐近趋近于 $ x $，当 $ x $ 取绝对值较大的正值时渐近趋近于 $ \log(x) $。
• 由于 $ g(x) $ 的收缩性质，即 $ |g'(x)| \leq 1 $，该方法具有数值稳定性，对输入噪声不敏感。
• 采用Halley方法可实现三阶收敛，从而能从粗略的初始估计快速实现高精度逼近。
• 由于表达式中第一项占主导地位，该方法在Halley更新公式的分母中避免了取消误差问题。
• 该方法适用于实际问题，如太阳能电池的电流-电压建模，其中高精度计算 $ \log(W(e^x)) $ 至关重要。