不动点理论在《A Distributional Perspective on Reinforcement Learning》上的应用

Fixed point理论是 Banach Space 中重要的理论工具。它常被用来讨论某个空间解的存在性，并由此发展出通过迭代的方式进行问题求解的算法。在[1]中，Fixed Piont理论处于整个算法的核心位置，是以分布式Bellman方程代替期望值Bellman方程的理论基础。本文将分成两个部分浅析之：第一部分 Fixed Point的迭代算法介绍；第二部分 Distributional 算法的不动点解析。

一、Fixed Point的迭代算法

首先要回顾以下 Fixed Point 原理，首先要给出的是收缩映射（Contraction）的定义[2]。
Definition 4.13（原书【2】的编号，下同）
A function $f:X\to Y$ between metric spaces is called a contraction if there exists a real number $\alpha$ with $0\le \alpha <1$, such that :
$d_Y(f(x_1),f(x_2))\le \alpha d_X(x_1,x_2)$
[简析]
所谓收缩（Contraction），指的是映射 $f$ 的一个属性，映射前 $x_1,x_2\in X$ 的距离 $d_X(x_1,x_2)$ 大于映射后的距离 $d_Y(f(x_1),f(x_2))$。这里要注意的是，原像空间 $X$ 的测度定义可以与像空间 $Y$ 的测度定义不同。

如上定义，若原像空间 $X$ 与像空间 $Y$ 相同，而且测度定义不变，且 $X$ 是Banach 空间（即完备的赋范空间，Complete Normed Space），则有如下不动点定理：
Theorem 4.8 (Banach Fixed-Point Theorem)
If $X$ is a complete non-empty metric spaces and $f:X\to X$ is a contraction, then $f$ has a unique fixed-point $x_0\in X$,
$f(x_0)=x_0$。
Banach空间是一种特殊的非空、完备的测度空间（metric space），Fixed-Point Theorem 能够保证任意在Banach空间的contraction函数都有且只有一个固定点（a unique fixed-point）。任取一个初始点 $x_1$，经contraction映射后得到 $x_2$，再将 $x_2$ 代入映射得到$x_3$，如此迭代，得到序列 { x n } n ≥ 1 \{x_n\}_{n\ge 1} {xn​}n≥1​，此序列收敛于$x_0$，即$x_n\to x_0$。若求解问题可以转换成在banach空间内不动点问题，就可以通过此迭代方法进行求解。这便是 fixed point 的迭代算法。以下通过一个例子进行说明。

例：线性方程组求解的迭代算法
考虑线性方程组：
$$\sum _{i=1}^n{a}_{ki}{x}_i=b_k(k=1,2,\cdots ,n)(1.1)$$
写成矩阵形式为 $\mathbf{A}\mathbf{x}=\mathbf{b}$，其中 $\mathbf{A}$ 是 $n\times n$ 矩阵，$\mathbf{x}=(x_1,x_2,\cdots ,x_n{)}^T$，$\mathbf{b}=(b_1,b_2,\cdots ,b_n{)}^T$，于是有：
$$\mathbf{A}\mathbf{x}=\mathbf{b}\Rightarrow \mathbf{x}=({\mathbf{I}}_n-\mathbf{A})\mathbf{x}+\mathbf{b}(1.2)$$
其中，${\mathbf{I}}_n$ 是 $n\times n$ 单位矩阵。考虑算子（operator） $F={\mathbf{I}}_n-\mathbf{A}$，有
$\mathbf{x}=F\mathbf{x}+\mathbf{b}$
算子 $F_b\mathbf{x}=F\mathbf{x}+\mathbf{b}$，则有 $\mathbf{x}=F_b\mathbf{x}$，这便是一个不动点问题。其中，$F$ 是线性算子，而 $F_b$ 是非线性算子，而 Banach Fixed Point 并不要求必须是线性算子，若 $F$ 是收缩算子，则 $F_b$ 也是收缩算子，这是因为：
$$d(F_b\mathbf{x},F_b\mathbf{y})=\Vert F_b\mathbf{x}-F_b\mathbf{y}\Vert =\Vert F\mathbf{x}-F\mathbf{y}\Vert =d(F\mathbf{x},F\mathbf{y})(1.3)$$
由$\mathbf{x}=F_b\mathbf{x}$，又 ${\mathbb{R}}^n$ 是一个Banach Space，若$F$ 是收缩算子，则符合Banach Fixed-Point Theorem的要求，则在${\mathbb{R}}^n$ 中存在唯一一个不动点。现在的焦点集中在：$F$ 是否收缩算子？这由 Banach Space 的范数（Norm）定义决定。考虑两种情况：
Case 1:
在 ${\mathbb{R}}^n$ 上取 $l_{\infty }$ 为范数，即：
$$\Vert \mathbf{x}\Vert =\underset{1\le k\le n}{\max }|\mathbf{x}|$$
于是，对于任意 $n\times n$矩阵 $C$ 与 $n\times 1$ 矢量 $\mathbf{x},\mathbf{y}$ 有：
$$\begin{gathered} \Vert C\mathbf{x}-C\mathbf{y}\Vert =\underset{1\le k\le n}{\max }\left|\sum _{i=1}^n{c}_{k,i}(x_i-y_i)\right|\le \underset{1\le k\le n}{\max }\left[\sum _{i=1}^n|c_{k,i}|\cdot |x_i-y_i|\right] \\ \le \underset{1\le k\le n}{\max }\left[\sum _{i=1}^n|c_{k,i}|\right]\cdot \underset{1\le k\le n}{\max }|x_i-y_i|=\underset{1\le k\le n}{\max }\left[\sum _{i=1}^n|c_{k,i}|\right]\Vert \mathbf{x}-\mathbf{y}\Vert \end{gathered}$$
因此，如果以下不等式成立：
$$\underset{1\le k\le n}{\max }\left[\sum _{i=1}^n|c_{k,i}|\right]<1(1.4)$$
则 $C$ 就是一个收缩算子，回到 $F={\mathbf{I}}_n-\mathbf{A}$ 算子，要使 $F$ 为收缩，则必须：
$$\underset{1\le k\le n}{\max }\left[\sum _{i=1}^n|-a_{k,i}+{\delta }_{k,i}|\right]<1(1.5)$$
其中，$a_{k,i}$ 是矩阵 $A$ 的k行，i列元素，${\delta }_{k,i}$ 是矩阵 $I_n$ 的k行，i列元素。满足这个要求的 $A\mathbf{x}=\mathbf{b}$，可用迭代的方法求解。

Case 2:
取 $l_2$ 范数，有
$$\Vert \mathbf{x}\Vert =\sum _{i=1}^n{x}_k^2$$
对于任意 $n\times n$矩阵 $C$ 与 $n\times 1$ 矢量 $\mathbf{x}$ 积的范数有,
$$\Vert C\mathbf{x}{\Vert }^2=\sum _{i=1}^n({\mathbf{c}}_k\mathbf{x}{)}^2(1.6)$$
其中 ${\mathbf{c}}_k$ 是矩阵 $C$ 的第k行，则 ${\mathbf{c}}_k\mathbf{x}$ 是两个矢量的标准内积，于是有：$({\mathbf{c}}_k\mathbf{x}{)}^2=\Vert {\mathbf{c}}_k\mathbf{x}{\Vert }^2\le \Vert {\mathbf{c}}_k{\Vert }^2\cdot \Vert \mathbf{x}{\Vert }^2$，因此
$$\begin{gathered} \Vert C\mathbf{x}{\Vert }^2=\sum _{i=1}^n({\mathbf{c}}_k\mathbf{x}{)}^2\le \sum _{i=1}^n\Vert {\mathbf{c}}_k{\Vert }^2\cdot \Vert \mathbf{x}{\Vert }^2 \\ \Vert C\mathbf{x}\Vert \le \left[\sqrt{\sum _{i=1}^n\Vert {\mathbf{c}}_k{\Vert }^2}\right]\Vert \mathbf{x}\Vert (1.7) \end{gathered}$$
由上我们得知，若要使 $F$(或$F_b$)收缩，在$l_2$ 范数下，必须：
$$\sum _{i=1}^n(-a_{k,i}+{\delta }_{k,i}{)}^2<1(1.8)$$

小结，上述两种情况（Case 1和Case 2）范数定义不同，得到的对算子$F$ 的要求不一样，但迭代过程是相同的，其实在迭代过程中，若满足要求，则不需要理会范数的定义。以下给出一个例子（二元一次方程组）：
$$\begin{gathered} \left[\begin{array}{cc} \\ 0.5& 0.2\\ 0.3& 0.5\end{array}\right]\left[\begin{array}{c} \\ {x}_1\\ x_2\end{array}\right]=\left[\begin{array}{c} \\ 0.1\\ 0.7\end{array}\right] \end{gathered}$$
求解过程用numpy实现如下：

import numpy as np
import matplotlib.pyplot as plt

#  直接通过求逆矩阵的方法求解
A = np.array([[0.5, 0.2],[0.3, 0.5]])
A_inv = np.linalg.inv(A)
b = [[0.1],[0.7]]
x = np.matmul(A_inv, b)   # x=array([[-0.47368421], [ 1.68421053]]) 

# 用迭代的方法求解
k = 30
x = np.array([[0.3],[0.1]])
x_arr = []
x_arr.append(x)
for i in range(k):
    C = np.identity(2)-A
    x = np.matmul(C,x)+b
    x_arr.append(x)

for p in x_arr:
    plt.scatter(p[0,0], p[1,0])
print(x_arr[-1]) 
plt.text(x_arr[-1][0,0]+0.03,x_arr[-1][1,0],'[%2.2f, %2.2f]'%(x_arr[-1][0,0],x_arr[-1][1,0]))
plt.show()

图1 用迭代法求解方程
上述二元一次方程组满足case1和case2对矩阵A的要求，用迭代的方法求出的数值与矩阵求逆的方法的相同。当矩阵A的维数大时，迭代方法要比求逆方法有优势，它不仅计算量小，速度快，而且有很强的鲁棒性。

二、Distributional Bellman算法的不动点解析

在增强学习（Reinforcement Learning）中，Bellman Equation占着极其重要的地位，传统 Bellman Equation 可以写成如下形式：
$$Q(x,a)=\mathbb{E}R(x,a)+\gamma \mathbb{E}Q(x^{\prime },a^{\prime })(2.1)$$
式中，$Q(x,a)$ 是状态 $x$ 下，采取行动 $a$ 的总收益（价值函数），$R(x,a)$ 是当前回报，$\gamma$ 是折扣，$\mathbb{E}$ 表示取随机量的期望。该方程反映的是总收益与随机量期望间的关系，但有时候均值并不能很好地反映实际分布，例如：具有两个模式的双峰分布，均值就在两个波谷之间。若能直接估计出总收益、当前回报等随机变量的分布，则岂不更好？《A Distributional Perspective on Reinforcement Learning》¹所阐述的思想就是基于此出发点的。[1]将公式（2.1）的 $\mathbb{E}$ 拿掉，变成了：
$$Q(x,a)\stackrel{D}{=}R(x,a)+\gamma Q(x^{\prime },a^{\prime })(2.2)$$
式中，$\stackrel{D}{=}$ 表示随机变量分布的相等，(2.2)表示当下“状态-行为”总收益的分布$Q(x,a)$ 和当前回报 $R(x,a)$ 与考虑折扣的下一“状态-行为”的总收益$\gamma Q(x^{\prime },a^{\prime })$ 的和相同。公式(2.1)与公式(2.2)的形式很相似，但意义却有巨大不同，公式(2.1)反映的是值（value）之间的关系，而公式(2.2)反映的是随机变量分布（distribution）之间的关系。[1]指出Q分布构成一个Banach 空间$\mathbb{Q}$，在$\mathbb{Q}$中定义Wasserstein metric 以衡量各分布( $U,V\in \mathbb{Q}$ )之间距离：
$$d_p(F,G):=\underset{U,V}{\inf }\Vert U-V{\Vert }_p(2.3)$$
式中，F、G 分别是随机变量 U、V 的 c.d.f(cumulative distribution function, 累积分布函数)。再取 $d_p$ 的上界，得到 ${\bar{d}}_p$:
$${\bar{d}}_p(Q_1,Q_2):=\sup d_p(Q_1(x,a),Q_2(x,a))(2.4)$$
以此 ${\bar{d}}_p$ 为空间 $\mathbb{Q}$ 的范数，[1]证明了算子 ${\mathcal{T}}^{\pi }:\mathbb{Q}\to \mathbb{Q}$ 是收缩的：
$${\mathcal{T}}^{\pi }Q(x,a)\stackrel{D}{=}R(x,a)+\gamma P^{\pi }Q(x,a)(2.5)$$
其中 $P^{\pi }$ 是状态转换算子，它表示现状态-行为总收益到NEXT状态-行为总收益的转换概率：
$$\begin{gathered} P^{\pi }Q(x,a)\stackrel{D}{=}Q(x^{\prime },a^{\prime }) \\ {x}^{\prime }\sim P(\cdot |x,a),a^{\prime }\sim \pi (\cdot |x^{\prime })(2.6) \end{gathered}$$
式中，$x^{\prime }$ 表示Next状态，它服从给定(x,a)下的P分布，$a^{\prime }$服从在$x^{\prime }$下采取行为的概率分布$\pi$。
综上，考虑迭代过程：$Q_{n+1}={\mathcal{T}}^{\pi }{Q}_n,\forall Q_0\in \mathbb{Q}$ ，它必收敛于算子 ${\mathcal{T}}^{\pi }$ 的不动点，该不动点即是所需搜索的最优 $Q^{\ast }(x,a)$ 分布。

[3]给出了一个pytorch实现例子，大致流程如下（还可参考我的另一篇博客[4]）：

图2 distributional dqn 处理流程
该方案由两个网络构成：Net 和 Target-Net，这两个网络分别代表迭代过程（$Q_{n+1}={\mathcal{T}}^{\pi }{Q}_n,\forall Q_0\in \mathbb{Q}$）中的$Q_{n+1}$ 和 $Q_n$，图2的处理步骤如下：

1. Net 接受Transition 采样中的当前状态 x。其中，一个Transition包含(x,a,r,x’)
   ，分别表示当前状态x、采取行动a，当前回报r，Next状态x’；
2. Net 输出的Q分布——$Z(x,a)$，它由N个$[V_{min},V_{max}]$ 离散分布构成，一般在$[V_{min},V_{max}]$上可以分为51个间隔，称为C51。N对应actions的数量，也即一个action对应一条分布，即图中一条$a_i$ 分布；
3. 从输出中选取Transition中a对应的分布，作为后续交叉熵计算所需分布 $Z(x,a)$;
4. 将 Transition 中Next状态 x’ 作为Target-Net的输入，得到下一状态价值分布 $Z(x^{\prime },a^{\prime })$，其组成也如2步骤的输出一样，因为Net的网络结构与Target-Net的结构是一样的；
5. 从$Z(x^{\prime },a^{\prime })$中选取期望最大的那一个$a^{\ast }$对应的分布——$Z(x^{\prime },a^{\ast })$;
6. 计算 $r+\gamma Z(x^{\prime },a^{\ast })$ 分布函数在 $[V_{min},V_{max}]$ 上的投影$Z^{\prime }(x,a)=Pro{j}_{[V_{min},V_{max}]}(r+\gamma Z(x^{\prime },a^{\ast }))$，其中 r 是Transition中的当前回报，$Z^{\prime }(x,a)$便是不动点迭代 $Q_{n+1}={\mathcal{T}}^{\pi }{Q}_n$ 算法中的 $Q_{n+1}$；
7. 用交叉熵定义$Z^{\prime }(x,a)$与$Z(x,a)$的距离，通过SGD（统计梯度下降）优化Net的参数，使$Z(x,a)$逼近$Z^{\prime }(x,a)$，上述的pytorch实现中通过 target_net_sync=1000，设置梯度学习循环次数；
8. 用学习后的Net 同步 Target-Net，实现不动点算法的迭代，即：$Q_{n+1}={\mathcal{T}}^{\pi }{Q}_n,\text{such that }{Q}_0\to Q_1\to \cdots \to Q_n\to Q_{n+1}\cdots$
9. 最后收敛，得到$Q^{\ast }(x,a)$分布。

以上方法，用多层神经网络作为分布函数的逼近工具，通过SGD方法优化网络，是典型的deep learning方法，另外，distributional Bellman算法还根据不动点理论，构建迭代学习框架，让deep learning成为整个框架的一部分，有机地将基础算法与deep learning融合起来，既扩展了deep learning使用的范围，也为未来搭建更复杂的系统提供了思路。

参考文献：
1、A Distributional Perspective on Reinforcement Learning，https://arxiv.org/abs/1707.06887
2、A Primer on Hilbert Space Theory：Linear Spaces, Topological Spaces, Metric Spaces, Normed Spaces, and Topological Groups，Carlo Alabiso，Springer International Publishing Switzerland 2015
3、《Deep Reinforcement Learning Hands-On》https://github.com/PacktPublishing/Deep-Reinforcement-Learning-Hands-On/blob/master/Chapter07/07_dqn_distrib.py
4、《A Distributional Perspective on Reinforcement Learning》的理解：https://blog.csdn.net/StreamRock/article/details/86605272

---

1. 1 ↩︎