基于Pytorch的强化学习(DQN)之Q-learning

目录

1. 引言

2. 数学推导

3. 算法


1. 引言

我们上次已经介绍了Saras算法,现在我们来学习一下和Saras算法非常相似的一个算法:

Q-learning算法。

Q-learning是一种用于机器学习的强化学习技术。 Q-learning的目标是学习一种策略,告诉Agent在什么情况下要采取什么行动。 它不需要环境模型,可以处理随机转换和奖励的问题,而无需进行调整。

对于任何有限马尔可夫决策过程(FMDP),Q学习找到一种最优的策略,即从当前状态开始,它在任何和所有后续步骤中最大化总奖励的预期值。在给定无限探索时间和部分随机策略的情况下,Q学习可以为任何给定的 FMDP 确定最佳动作选择策略。“Q”命名函数返回用于提供强化的奖励,并且可以说代表在给定状态下采取的动作的“质量”。

Q-learning算法与Saras算法区别在于Q-learning是学习最优动作价值函数 Q^*,而Saras是学习动作价值函数 Q_{\pi}

2. 数学推导

我们上次推导出了动作价值函数的递推公式Q_{\pi}(s_t,a_t)=E[R_t+\gamma Q_{\pi}(S_{t+1},A_{t+1})],现在我们选取最优策略\pi^*,由最优动作价值函数的定义,我们得到Q^*(s_t,a_t)=E[R_t+\gamma Q^*(S_{t+1},A_{t+1})]

对于含有随机变量的Q^*(S_{t+1},A_{t+1})我们先将其写成最大化的形式,由优化目标我们有:A_{t+1}=\underset{a}{argmax}\,Q^*(S_{t+1},a)

于是得到 

Q^*(S_{t+1},A_{t+1})=\underset{a}{max}\,Q^*(S_{t+1},a)

那么

Q^*(s_t,a_t)=E[R_t+\gamma \underset{a}{max}\,Q^*(S_{t+1},a)]

这个式子中含有随机变量R_t,S_{t+1},我们使用蒙特卡罗算法来近似这个期望:由r_t\approx R_t,s_{t+1}\approx S_{t+1}

我们得到 

Q^*(s_t,a_t)\approx E[r_t+\gamma \underset{a}{max}\,Q^*(s_{t+1},a)]

于是我们将 

r_t+\gamma \underset{a}{max}\,Q^*(s_{t+1},a) 作为TD target 使用TD 算法具体步骤如下:

  1. 观测到state transtition (s_t,a_t,r_t,s_{t+1})
  2. 计算TD target y_t=r_t+\gamma \underset{a}{max}\,Q^*(s_{t+1},a)
  3. 计算TD error \delta_t=Q^*(s_t,a_t)-y_t
  4. 更新最优动作价值函数Q^*(s_t,a_t)\leftarrow Q^*(s_t,a_t)-\alpha \delta_t

上面的 \underset{a}{max}\,Q^*(s_{t+1},a) 可以从Q表中获取,我们只需要从第s_{t+1}行搜索出最大的Q^*(s_{t+1},a)即可

3. 算法

  1.  用神经网络 q(s,a;w) 来近似Q^*(s,a)
  2. 计算TD target: y_t=r_t+\gamma q_(s_{t+1},a_{t+1};w)
  3. 计算TD error: \delta_t=q(s,a;w)-y_t
  4. 损失函数loss:\frac{\delta_t^2}{2}
  5. 梯度下降: w\leftarrow w-\alpha\delta_t\frac{\partial \,q(s,a;w)}{\partial \,w}

伪码如下

  • 3
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
DQN是一种使用深度神经网络来解决强化学习问题的方法。在DQN中,QN指的是Q-Learning的一种变体,而RL是强化学习的缩写。DQN通过将状态和动作映射到Q值,实现了对环境的建模和决策。它具有很多优点,包括能够处理高维状态空间和连续动作空间,以及能够通过经验回放和目标网络来提高训练的稳定性和效果。 在代码中,SARSA是一种基于状态-动作对的强化学习算法。它使用一个Q表来存储状态和动作的Q值,并在每个时间步根据当前状态和选择的动作来更新Q表。在代码中,SARSA函数首先初始化一个Q表,然后根据一定的迭代次数进行训练。在训练过程中,根据当前状态和选择的动作,通过获取奖励和下一个状态来更新Q表的值。最后,函数返回更新后的Q表。 因此,pytorch DQN指的是使用PyTorch框架实现DQN算法。PyTorch是一个流行的深度学习框架,可以用于构建神经网络模型并进行训练。在DQN中,可以使用PyTorch来构建神经网络模型,并使用DQN算法来训练模型以解决强化学习问题。<span class="em">1</span><span class="em">2</span><span class="em">3</span> #### 引用[.reference_title] - *1* *2* *3* [DQN神经网络小结(Pytorch版)](https://blog.csdn.net/FUTEROX/article/details/124615826)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v92^chatsearchT3_1"}}] [.reference_item style="max-width: 100%"] [ .reference_list ]
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值