MENU

寻求一个光滑的最大值函数

July 5, 2021 • Read: 2956 • 数学阅读设置

这篇文章的目的是推导最大值函数 $\max (x,y)$ 的一个光滑可导函数,并且该函数具有多阶可导性。实际上这和深度学习的关系并不是特别大,只有极少数情况会用到

在数学分析中,当 $x\ge0,y\ge0$ 时,我们有

$$ \max(x,y) = \frac{1}{2}(|x+y|+|x-y|)\tag{1} $$

那么,为了寻求一个最大值的光滑函数,我们首先考虑寻找一个能够近似表示绝对值 $|x|$ 的函数。那么,哪些函数可以使用呢?

直接观察挺难发现哪个函数可以使用的,我们将问题逐步向简单推进。对 $f (x)=|x|$ 求导,除了 $x=0$ 这一点外,其他都可以顺利求导

$$ f'(x) = \begin{cases}1,\quad &x>0\\-1, \quad &x < 0\end{cases} \tag{2} $$

到此为止,我们要做的是利用一个函数 $g (x)$,将其仿射变换为 $f'(x)$。而关于函数 $g (x)$ 的选择,应该至少满足以下两点:

  1. 应该是个分段函数
  2. 求不定积分简单,或者说至少能查到 $\int g (x) dx$ 的结果

实际上 $g (x)$ 也确实有许多满足上述两点的选择,以下就分别给出选择 $\text {sign}(x)$ 和单位阶跃函数 $\theta (x)$ 的推导

$\text{sign}(x)$

首先回顾 $\text {sign}(x)$ 的形式:

$$ \text{sign}(x)=\begin{cases}1,\quad &x>0\\-1, \quad &x < 0\end{cases} \tag{3} $$

于是就有 $f'(x)=\text {sign}(x)$,下面只需要寻求 $\text {sign}(x)$ 的近似函数,很容易想到

$$ \text{sign}(x)=\lim_{k\to +\infty} \tanh(kx)\tag{4} $$

其中,$\tanh (x)=\frac {e^x-e^{-x}}{e^x+e^{-x}}$。由于 $f'(x)=\tanh (kx)$,积分得

$$ \begin{aligned} f(x) &= \frac{1}{k}\ln(\cosh(kx))\\ &=\frac{1}{k}\ln(\frac{e^{kx}+e^{-kx}}{2})\\ &=\frac{1}{k}[\ln(e^{kx}+e^{-kx})-\ln2] \end{aligned}\tag{5} $$

不难发现,(5) 式中的对数部分,在 $k$ 足够大的时候,常数 $\ln2$ 的影响微乎其微,把它去掉之后,我们有一个比较简单的绝对值函数:

$$ |x|=\lim_{k\to +\infty} \frac{1}{k}\ln(e^{kx}+e^{-kx})\tag{6} $$

$\theta(x)$

首先回顾单位阶跃函数 $\theta (x)$ 的形式:

$$ \theta(x)=\begin{cases}1,\quad &x>0\\0, \quad &x < 0\end{cases} \tag{7} $$

那么

$$ f'(x) = 2\theta(x) - 1 \tag{8} $$

下面只需要寻求 $\theta (x)$ 的近似函数,物理学家已经提供现成的函数给我们了,一个比较简单的形式是

$$ \theta(x)=\lim_{k\to +\infty} \sigma(kx)=\lim_{k\to +\infty} \frac{1}{1+e^{-k x}}\tag{9} $$

那么我们就可以取 $\frac {1}{1+e^{-kx}}$ 作为近似函数了,带入 (8) 式得到 $\frac {2}{1+e^{-kx}} - 1$,积分得

$$ \begin{aligned}f(x)&=\frac{2}{k}\ln(1+e^{kx})-x\\ &= \frac{1}{k}\ln(1+e^{kx}) +\frac{1}{k}[\ln(1+e^{kx}) -\ln e^{kx}]\\ &=\frac{1}{k}\left[\ln(1+e^{kx})+\ln(1+e^{-kx})\right]\\ &=\frac{1}{k}\ln(2+e^{kx}+e^{-kx})\end{aligned}\tag{10} $$

同理,当 $k$ 足够大的时候,常数 2 的影响微乎其微,把它去掉之后,我们同样得到一个比较简单的绝对值函数

$$ |x| = \lim\limits_{k \to +\infty}\frac{1}{k} \ln(e^{kx} + e^{-kx}) \tag{11} $$

对比 (6) 和 (11) 可知,无论选用 $\text {sign}(x)$ 还是 $\theta (x)$,最终 $|x|$ 的近似函数都是一样的,这也从侧面证明了这个式子的准确性

将式 (11) 带入到式 (1) 得

$$ \begin{aligned} \max(x,y) &= \frac{1}{2}(|x+y|+|x-y|)\\ &= \lim\limits_{k \to +\infty}\frac{1}{2k}\{\ln[e^{k(x+y)} + e^{-k(x+y)}]+\ln[e^{k(x-y)} + e^{-k(x-y)}]\}\\ &= \lim\limits_{k \to +\infty}\frac{1}{2k}\ln(e^{2kx}+e^{-2kx}+e^{2ky}+e^{-2ky}) \end{aligned}\tag{12} $$

由于式 (1) 是在 $x\ge 0, y\ge 0$ 时成立的,所以式 (12) 中的 $e^{-2kx},e^{-2ky}$ 都不重要了,我们也把它去掉,进一步得到

$$ \max(x,y) = \lim\limits_{k \to +\infty}\frac{1}{2k}\ln(e^{2kx}+e^{2ky}) \tag{13} $$

或者写成

$$ \max(x,y) = \lim\limits_{k \to +\infty}\frac{1}{k}\ln(e^{kx}+e^{ky}) \tag{14} $$

(14) 式正是我们希望得到的理想的最大值函数。虽然我们的推导基于 $x\ge0,y\ge0$,但是不难发现,对于 $x,y$ 中出现负数时,上述公式仍然成立!它甚至还可以推广到多个变量的最大值函数:

$$ \max(x,y,z,\dots)=\lim_{k\to +\infty} \frac{1}{k}\ln(e^{kx}+e^{ky}+e^{kz}+\dots)\tag{15} $$

补充

对于式 (14) 我们不妨设 $k=1$,令 $x=3,y=8$,则 $\ln (e^3 + e^8)\approx8.00672$ 。由于 $x$ 和 $y$ 都在指数位置上,因此它们的差距将会放得很大。虽然 3 和 8 这两个数虽然相隔不远,但 $e^3\approx20.0855,e^8\approx2980.96$,两个幂差了几个数量级。因此,把 $e^3$ 加到 $e^8$ 上,几乎不会改变 $e^8$ 的大小。对 $e^3+e^8$ 取对数的结果和直接对 $e^8$ 取对数的结果相差不多,并且这个函数可以通过控制 $k$ 的大小实现精度控制,$k$ 越大则结果越近似

这实际上是做了这样的一个事情:找一个在整个实数域上都单调递增的函数,而且增长速度要快于线性增长,然后求和,最后取逆函数。因此,不难构造出类似的函数:我们选 $f (x)=x^{2k+1}$,那么得到

$$ \max(x,y)=\lim_{k\to+\infty} \sqrt[2k+1]{x^{2k+1}+y^{2k+1}}\tag{16} $$

当然,(16) 的精度(或者说收敛速度)远没有 (14) 那么好,要提高精度也不难,比如

$$ \max(x,y)=\lim_{k\to +\infty} \frac{1}{k}\ln\ln \left(e^{e^{kx}}+e^{e^{ky}}\right)\tag{17} $$

References

Last Modified: July 19, 2021
Archives Tip
QR Code for this page
Tipping QR Code