IT数码 购物 网址 头条 软件 日历 阅读 图书馆
TxT小说阅读器
↓语音阅读,小说下载,古典文学↓
图片批量下载器
↓批量下载图片,美女图库↓
图片自动播放器
↓图片自动播放器↓
一键清除垃圾
↓轻轻一点,清除系统垃圾↓
开发: C++知识库 Java知识库 JavaScript Python PHP知识库 人工智能 区块链 大数据 移动开发 嵌入式 开发工具 数据结构与算法 开发测试 游戏开发 网络协议 系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程
数码: 电脑 笔记本 显卡 显示器 固态硬盘 硬盘 耳机 手机 iphone vivo oppo 小米 华为 单反 装机 图拉丁
 
   -> 人工智能 -> 强化学习算法TRPO之共轭梯度优化 -> 正文阅读

[人工智能]强化学习算法TRPO之共轭梯度优化

TRPO是OpenAI提出的一种策略单调提升的算法,关于其论文以及解读见我的另一篇论文笔记之TRPO这篇文论文解读将TRPO的重点以及细节都指明了,但是关于目标函数的优化部分由于篇幅原因只是简单说明了以下,具体细节将放在这里进行说明。

参考目录:
①优化理论相关书籍
TRPO论文
TRPO论文推导
TRPO的补充
TRPO讲解

TRPO论文优化部分理解
看过TRPO论文的都知道作者采用共轭梯度+线性搜索来迭代求出关于目标函数的最优解 θ ? \theta^* θ?
因此接下来我将详细分析作者是如何利用CG和一维搜索来应用于RL算法的。
Note:

  1. 关于共轭梯度下降和线性搜索的基础知识就不讲解了,这部分内容不清楚的最好去参考优化理论的书籍,做个详细的了解。

首先给出整个优化过程的伪代码,有2个版本,一个是简化版,一个是基于共轭梯度下降的版本:

  1. 简化版:在这里插入图片描述
  2. 基于CG版:在这里插入图片描述

为了便于分析,接下来基于CG版本进行讲解。2个红框中标出的是TRPO论文算法比标准CG算法多出来的地方,分别是对学习率 α k \alpha_k αk?进行修正且进行一维搜索得到最佳学习率;另一部分是利用Hessian-Free技术去避免求解Hessian矩阵的问题。
面对以上伪代码,应该有四个问题,接下来将一一解答:

  1. Q 1 Q1 Q1:如何将RL目标函数转成共轭梯度算法需要的二次型问题。
  2. Q 2 Q2 Q2:为何要修正 α \alpha α?如何去修正呢?
  3. Q 3 Q3 Q3:如何进行线性搜索呢?
  4. Q 4 Q4 Q4:如何使用Hessian-Free技术。

首先给出我们要优化的目标,也就是TRPO论文中的公式(14):
m a x i m i z e θ E s ~ ρ θ o l d , a ~ q [ π θ ( a ∣ s ) q ( a ∣ s ) Q θ o l d ( s , a ) ] s . t . ??? E s ~ ρ θ o l d [ D K L ( π θ o l d ( ? ∣ s ) ∣ ∣ π θ ( ? ∣ s ) ) ] ≤ δ . \mathop{maximize}\limits_\theta\mathbb{E}_{s\sim\rho_{\theta_{old}},a\sim q}[\frac{\pi_\theta(a|s)}{q(a|s)}Q_{\theta_{old}}(s,a)]\\ s.t.\,\,\,\mathbb{E}_{s\sim\rho_{\theta_{old}}}[D_{KL}(\pi_{\theta_{old}}(\cdot|s)||\pi_\theta(\cdot|s))]\leq\delta. θmaximize?Esρθold??,aq?[q(as)πθ?(as)?Qθold??(s,a)]s.t.Esρθold???[DKL?(πθold??(?s)πθ?(?s))]δ.我们将其简化成:
m a x i m i z e θ f ( θ ) s . t . ??? ? k l ( θ ) ≥ ? δ . \mathop{maximize}\limits_\theta f(\theta)\\ s.t.\,\,\,-kl(\theta)\ge-\delta. θmaximize?f(θ)s.t.?kl(θ)?δ.

二次型转换

Q 1 Q1 Q1:如何将RL目标函数转成共轭梯度算法需要的二次型问题?
共轭梯度下降算法的目标函数是二次型函数,但对于一般非线性函数,我们是通过泰勒定理将其转为近似二次型。在这里我将分两步走来解决这个问题。

泰勒展开

首先我们要做的就是在 θ o l d \theta_{old} θold?处对 f ( θ ) 、 k l ( θ ) f(\theta)、kl(\theta) f(θ)kl(θ)泰勒展开
对于目标函数,省去二阶项以及后面的高阶项:
f ( θ ) ≈ f ( θ o l d ) + g T ? ( θ ? θ o l d ) f(\theta)\approx f(\theta_{old})+g^T\cdot(\theta-\theta_{old}) f(θ)f(θold?)+gT?(θ?θold?)对于约束条件,同样省去常数项、一阶项以及后面的高阶项:
k l ( θ ) ≈ 1 2 ( θ ? θ o l d ) T F ( θ o l d ) ( θ ? θ o l d ) kl(\theta)\approx\frac{1}{2}(\theta-\theta_{old})^TF(\theta_{old})(\theta-\theta_{old}) kl(θ)21?(θ?θold?)TF(θold?)(θ?θold?)Note:

  1. g = ( ? f ? θ ) T ∣ θ = θ o l d g=(\frac{\partial{f}}{\partial{\theta}})^T|_{\theta=\theta_{old}} g=(?θ?f?)Tθ=θold??
  2. 第一项为常数项,我们将其忽略,故 f ( θ ) ≈ g T ? ( θ ? θ o l d ) f(\theta)\approx g^T\cdot(\theta-\theta_{old}) f(θ)gT?(θ?θold?)
  3. F F FHessian矩阵,展开为 ? 2 k l ( θ ) ? θ 2 ∣ θ = θ o l d \frac{\partial^2{kl(\theta)}}{\partial{\theta^2}}|_{\theta=\theta_{old}} ?θ2?2kl(θ)?θ=θold??。黑塞矩阵在二次项函数中是常数矩阵,但是在非线性函数中对于每一次迭代都需要计算一次,计算量和存储量都很大,因此后续我们才要去消除它的计算。

因此我们的优化目标就变成了:
m a x i m i z e θ g T ? ( θ ? θ o l d ) s . t . ??? 1 2 ( θ ? θ o l d ) T F ( θ o l d ) ( θ ? θ o l d ) ≤ δ \mathop{maximize}\limits_\theta g^T\cdot(\theta-\theta_{old})\\ s.t.\,\,\,\frac{1}{2}(\theta-\theta_{old})^TF(\theta_{old})(\theta-\theta_{old})\le\delta θmaximize?gT?(θ?θold?)s.t.21?(θ?θold?)TF(θold?)(θ?θold?)δ

无约束的转换

共轭梯度算法是一种无约束优化算法,因此有必要将上述约束型转为无约束,这里我们使用拉格朗日乘子法
添加拉格朗日乘子 λ \lambda λ,则上式转为:
L ( θ , λ ) = g T ( θ ? θ o l d ) ? λ 2 [ ( θ ? θ o l d ) T F ( θ ? θ o l d ) ? δ ] K K T 条 件 : { 1 2 ( θ ? θ o l d ) T F ( θ o l d ) ( θ ? θ o l d ) ≤ δ λ ≥ 0 λ ( 1 2 ( θ ? θ o l d ) T F ( θ o l d ) ( θ ? θ o l d ) ? δ ) = 0 L(\theta,\lambda)=g^T(\theta-\theta_{old})-\frac{\lambda}{2}[(\theta-\theta_{old})^TF(\theta-\theta_{old})-\delta]\\ KKT条件:\begin{cases} \frac{1}{2}(\theta-\theta_{old})^TF(\theta_{old})(\theta-\theta_{old})\le\delta\\ \lambda\ge0\\ \lambda(\frac{1}{2}(\theta-\theta_{old})^TF(\theta_{old})(\theta-\theta_{old})-\delta)=0 \end{cases} L(θ,λ)=gT(θ?θold?)?2λ?[(θ?θold?)TF(θ?θold?)?δ]KKT??????21?(θ?θold?)TF(θold?)(θ?θold?)δλ0λ(21?(θ?θold?)TF(θold?)(θ?θold?)?δ)=0?这样一来约束问题就转为了无约束问题。但在正式运用之前,还需要将目标函数变个型,令 s = θ ? θ o l d s=\theta-\theta_{old} s=θ?θold?,则
L ( θ , λ ) = ? λ 2 s T F s + g T s + λ 2 δ L(\theta,\lambda)=-\frac{\lambda}{2}s^TFs+g^Ts+\frac{\lambda}{2}\delta L(θ,λ)=?2λ?sTFs+gTs+2λ?δ这就是转化之后的二次型了!

学习率修正

Q 2 Q2 Q2:为何要修正 α \alpha α?如何去修正呢?
为何要修正 α \alpha α呢,因为我们在转无约束的时候,多了一个 K K T KKT KKT条件: λ ( 1 2 ( θ ? θ o l d ) T F ( θ o l d ) ( θ ? θ o l d ) ? δ ) = 0 \lambda(\frac{1}{2}(\theta-\theta_{old})^TF(\theta_{old})(\theta-\theta_{old})-\delta)=0 λ(21?(θ?θold?)TF(θold?)(θ?θold?)?δ)=0,我们接下来对这个式子进行处理:
因为根据共轭梯度算法的流程,有 θ = θ o l d + α ? d ( k ) , α ∈ R \theta=\theta_{old}+\alpha\cdot d^{(k)},\alpha\in\mathbb{R} θ=θold?+α?d(k),αR,所以
1 2 ( θ ? θ o l d ) T F ( θ o l d ) ( θ ? θ o l d ) ? δ = 1 2 ( α d ) T F ( α d ) ? δ = 1 2 α 2 ( d T F d ) ? δ \frac{1}{2}(\theta-\theta_{old})^TF(\theta_{old})(\theta-\theta_{old})-\delta\\ =\frac{1}{2}(\alpha d)^TF(\alpha d)-\delta\\ =\frac{1}{2}\alpha^2(d^TFd)-\delta 21?(θ?θold?)TF(θold?)(θ?θold?)?δ=21?(αd)TF(αd)?δ=21?α2(dTFd)?δ又因为理论上最好的优化区域并不是在置信区域内,而是在置信边界上,即 λ > 0 \lambda>0 λ>0,这样一来 1 2 ( θ ? θ o l d ) T F ( θ o l d ) ( θ ? θ o l d ) ? δ ≡ 0 \frac{1}{2}(\theta-\theta_{old})^TF(\theta_{old})(\theta-\theta_{old})-\delta\equiv0 21?(θ?θold?)TF(θold?)(θ?θold?)?δ0,故可以得出以下结论:
α = 2 δ d T F d \alpha=\sqrt{\frac{2\delta}{d^TFd}} α=dTFd2δ? ?这样一来就解决了如何去修正 α \alpha α对的问题,也就是说当我们在TRPO算法中使用共轭梯度下降的时候,就不能用标准的 α \alpha α公式了,而是要用上式这个经过 K K T KKT KKT修正过的

线性搜索

Q 3 Q3 Q3:如何进行线性搜索呢?
根据标准的共轭梯度算法,其会在 n n n步迭代就收敛,其 α \alpha α其实就是用一维搜索方法进行优化的。但是 Q 2 Q2 Q2中已经将算法进行了改变,故收敛性无法得到保证,那么如何挽回呢?也就是说现在的问题就是如何获得一个更好的学习率 α ? \alpha^* α?,使得这一步的 α \alpha α是一个不错的学习率,这又是一个优化问题:输入是学习率 α ∈ R \alpha\in\mathbb{R} αR,输出目标函数 L ∈ R L\in\mathbb{R} LR R → R \mathbb{R}\to\mathbb{R} RR的问题我们采用线性搜索(即一维搜索)来做,比如常见的黄金分割、斐波那契、二分法、牛顿法、割线法等等。作者在这里简单的采用了指数式下降来搜寻最佳的学习率 α \alpha α,类似于Armijo划界法,也就是说从一开始最大的学习率 α ( 0 ) \alpha^{(0)} α(0)开始,接下来每一个学习率都在上一个备选值的基础上乘以缩减因子 τ = 0.5 \tau=0.5 τ=0.5,这样就构成 a k = τ m a ( 0 ) a_k=\tau^ma^{(0)} ak?=τma(0),相当于构成了一个序列,我们要在这个序列中找出使得目标函数提升的最大学习率 a ? a^* a?

Hessian-Free

Q 4 Q4 Q4:如何使用Hessian-Free技术。
我们在伪代码中看出我们需要计算一个Hessian矩阵,黑塞矩阵在二次项函数中是常数矩阵,但是在非线性函数中对于每一次迭代都需要计算一次,计算量和存储量都很大,因此我们必须要消除它。理论上的消除办法有3种,这里我介绍一种,即论文中作者采用的Hestenes-Stiefel公式,它用来消除 F F F的原理是什么呢?
我们发现共轭梯度的算法流程中,Hessian矩阵总以matrix-vector products的形式出现。因此我们可以直接带走他们。
具体来说:
对于二次型: f = 1 2 θ T F θ ? b T θ f=\frac{1}{2}\theta^TF\theta-b^T\theta f=21?θTFθ?bTθ来说:
已知 θ k + 1 = θ k + α k d k \theta^{k+1}=\theta^{k}+\alpha_kd^{k} θk+1=θk+αk?dk,等式两侧同时左乘Hessian矩阵 F F F并减去常数向量 b b b F ? θ k + 1 ? b = F ? θ k + α k F d k ? b F\cdot\theta^{k+1}-b=F\cdot\theta^{k}+\alpha_kFd^{k}-b F?θk+1?b=F?θk+αk?Fdk?b,又因为 g k = ? f ? θ k = F ? θ k ? b g^k=\frac{\partial{f}}{\partial{\theta^k}}=F\cdot\theta^k-b gk=?θk?f?=F?θk?b,带入之后: g k + 1 = g k + α k F d k g^{k+1}=g^k+\alpha_kFd^k gk+1=gk+αk?Fdk,所以可得以下结论:
F d k = g k + 1 ? g k α k Fd^k=\frac{g^{k+1}-g^k}{\alpha_k} Fdk=αk?gk+1?gk?从这个式子可以看出,有关要计算 F d k Fd^k Fdk的地方我们以后都只需要计算2个梯度值就行了,而这两个梯度值在共轭梯度算法里本来就需要算的,因此并没有增加额外的算法负担
所以 β \beta β就变成了:
β k = ( g k + 1 ) T ? [ g k + 1 ? g k ] ( d k ) T [ g k + 1 ? g k ] \beta_k=\frac{(g^{k+1})^T\cdot[g^{k+1}-g^k]}{(d^k)^T[g^{k+1}-g^k]} βk?=(dk)T[gk+1?gk](gk+1)T?[gk+1?gk]?

总结

  1. 接下来只要按照伪代码的流程,不断迭代,就可以优化出参数 θ ? \theta^* θ?,共轭梯度算法最著名的地方就在于他可以一边迭代,一边利用当前点梯度和上一个搜索方向的线性结合创造出和之前搜索方向均共轭的下一个搜索方向 d k + 1 d^{k+1} dk+1出来。
  2. 总的来说,TRPO的优化采用修正的共轭梯度算法+一维搜索算法。我们通过泰勒定理以及拉格朗日乘子法将非线性的约束目标转为了近似二次型函数的无约束目标,并利用 K K T KKT KKT条件对学习率 α \alpha α进行修正
  3. 可以看出虽然我们消除了Hessian矩阵的计算,但是共轭梯度算法还是有一定复杂度的,这也是TRPO算法不如PPO实用的一个原因,而PPO采用的是随机梯度优化算法,每一次迭代都基于采样的mini-batch个样本。
  人工智能 最新文章
2022吴恩达机器学习课程——第二课(神经网
第十五章 规则学习
FixMatch: Simplifying Semi-Supervised Le
数据挖掘Java——Kmeans算法的实现
大脑皮层的分割方法
【翻译】GPT-3是如何工作的
论文笔记:TEACHTEXT: CrossModal Generaliz
python从零学(六)
详解Python 3.x 导入(import)
【答读者问27】backtrader不支持最新版本的
上一篇文章      下一篇文章      查看所有文章
加:2021-08-22 13:32:41  更:2021-08-22 13:33:20 
 
开发: C++知识库 Java知识库 JavaScript Python PHP知识库 人工智能 区块链 大数据 移动开发 嵌入式 开发工具 数据结构与算法 开发测试 游戏开发 网络协议 系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程
数码: 电脑 笔记本 显卡 显示器 固态硬盘 硬盘 耳机 手机 iphone vivo oppo 小米 华为 单反 装机 图拉丁

360图书馆 购物 三丰科技 阅读网 日历 万年历 2024年11日历 -2024/11/27 18:41:46-

图片自动播放器
↓图片自动播放器↓
TxT小说阅读器
↓语音阅读,小说下载,古典文学↓
一键清除垃圾
↓轻轻一点,清除系统垃圾↓
图片批量下载器
↓批量下载图片,美女图库↓
  网站联系: qq:121756557 email:121756557@qq.com  IT数码