半猎豹

../../../_images/half_cheetah.gif

此环境是 Mujoco 环境的一部分,其中包含有关该环境的通用信息。

动作空间

Box(-1.0, 1.0, (6,), float32)

观测空间

Box(-inf, inf, (17,), float64)

导入

gymnasium.make("HalfCheetah-v5")

描述

此环境基于 P. Wawrzyński 在“类猫机器人实时学习奔跑”中的工作。半猎豹是一个二维机器人,由9个身体部件和连接它们的8个关节(包括两只爪子)组成。目标是对关节施加扭矩,使猎豹尽可能快地向前(向右)奔跑,根据向前移动的距离获得正奖励,向后移动则获得负奖励。猎豹的躯干和头部是固定的,扭矩只能施加到其他6个关节上,包括前后大腿(连接到躯干)、小腿(连接到大腿)和脚(连接到小腿)。

动作空间

../../../_images/half_cheetah.png

动作空间是一个 Box(-1, 1, (6,), float32)。一个动作代表施加在铰链关节上的扭矩。

编号

动作

控制最小值

控制最大值

名称(在对应 XML 文件中)

关节

类型(单位)

0

施加在后大腿转子上的扭矩

-1

1

bthigh

铰链

扭矩 (N m)

1

施加在后小腿转子上的扭矩

-1

1

bshin

铰链

扭矩 (N m)

2

施加在后脚转子上的扭矩

-1

1

bfoot

铰链

扭矩 (N m)

3

施加在前大腿转子上的扭矩

-1

1

fthigh

铰链

扭矩 (N m)

4

施加在前小腿转子上的扭矩

-1

1

fshin

铰链

扭矩 (N m)

5

施加在前脚转子上的扭矩

-1

1

ffoot

铰链

扭矩 (N m)

观测空间

观测空间按顺序包含以下部分:

  • qpos (默认为 8 个元素): 机器人身体部位的位置值。

  • qvel (9 个元素): 这些单个身体部位的速度(它们的导数)。

默认情况下,观测不包含机器人的 x 坐标 (rootx)。在构建时可以通过传递 exclude_current_positions_from_observation=False 来包含它。在这种情况下,观测空间将是 Box(-Inf, Inf, (18,), float64),其中第一个观测元素是机器人的 x 坐标。无论 exclude_current_positions_from_observation 设置为 True 还是 False,x 和 y 坐标都会在 info 中以键 "x_position""y_position" 返回。

然而,默认情况下,观测空间是一个 Box(-Inf, Inf, (17,), float64),其元素如下:

编号

观测

最小值

最大值

名称(在对应 XML 文件中)

关节

类型(单位)

0

前端尖端的 z 坐标

-Inf

Inf

rootz

滑动

位置 (m)

1

前端尖端的角度

-Inf

Inf

rooty

铰链

角度 (rad)

2

后大腿的角度

-Inf

Inf

bthigh

铰链

角度 (rad)

3

后小腿的角度

-Inf

Inf

bshin

铰链

角度 (rad)

4

后脚的角度

-Inf

Inf

bfoot

铰链

角度 (rad)

5

前大腿的角度

-Inf

Inf

fthigh

铰链

角度 (rad)

6

前小腿的角度

-Inf

Inf

fshin

铰链

角度 (rad)

7

前脚的角度

-Inf

Inf

ffoot

铰链

角度 (rad)

8

前端尖端的 x 坐标速度

-Inf

Inf

rootx

滑动

速度 (m/s)

9

前端尖端的 z 坐标速度

-Inf

Inf

rootz

滑动

速度 (m/s)

10

前端尖端的角速度

-Inf

Inf

rooty

铰链

角速度 (rad/s)

11

后大腿的角速度

-Inf

Inf

bthigh

铰链

角速度 (rad/s)

12

后小腿的角速度

-Inf

Inf

bshin

铰链

角速度 (rad/s)

13

后脚的角速度

-Inf

Inf

bfoot

铰链

角速度 (rad/s)

14

前大腿的角速度

-Inf

Inf

fthigh

铰链

角速度 (rad/s)

15

前小腿的角速度

-Inf

Inf

fshin

铰链

角速度 (rad/s)

16

前脚的角速度

-Inf

Inf

ffoot

铰链

角速度 (rad/s)

已排除

前端尖端的 x 坐标

-Inf

Inf

rootx

滑动

位置 (m)

奖励

总奖励为:reward = forward_reward - ctrl_cost

  • forward_reward:向前移动的奖励,如果半猎豹向前(沿正 \(x\) 方向/向右)移动,则此奖励为正。\(w_{forward} \times \frac{dx}{dt}\),其中 \(dx\) 是“尖端”的位移(\(x_{after-action} - x_{before-action}\)),\(dt\) 是动作之间的时间,这取决于 frame_skip 参数(默认为 \(5\)),以及 frametime(为 \(0.01\))——因此默认情况下 \(dt = 5 \times 0.01 = 0.05\)\(w_{forward}\)forward_reward_weight(默认为 \(1\))。

  • ctrl_cost:一种负奖励,用于惩罚半猎豹采取过大的动作。\(w_{control} \times \|action\|_2^2\),其中 \(w_{control}\)ctrl_cost_weight(默认为 \(0.1\))。

info 包含各个奖励项。

起始状态

初始位置状态是 \(\mathcal{U}_{[-reset\_noise\_scale \times I_{9}, reset\_noise\_scale \times I_{9}]}\)。初始速度状态是 \(\mathcal{N}(0_{9}, reset\_noise\_scale^2 \times I_{9})\)

其中 \(\mathcal{N}\) 是多元正态分布,\(\mathcal{U}\) 是多元均匀连续分布。

回合结束

终止

半猎豹从不终止。

截断

一个回合的默认持续时间是 1000 个时间步。

参数

HalfCheetah 提供了一系列参数,用于修改观测空间、奖励函数、初始状态和终止条件。这些参数可以在 gymnasium.make 期间按以下方式应用:

import gymnasium as gym
env = gym.make('HalfCheetah-v5', ctrl_cost_weight=0.1, ....)

参数

类型

默认值

描述

xml_file

str

"half_cheetah.xml"

MuJoCo 模型的路径

forward_reward_weight

float

1

forward_reward 项的权重(参见奖励部分)

ctrl_cost_weight

float

0.1

ctrl_cost 权重的权重(参见奖励部分)

reset_noise_scale

float

0.1

初始位置和速度随机扰动的尺度(参见起始状态部分)

exclude_current_positions_from_observation

bool

True

是否从观测中省略 x 坐标。排除位置可以作为一种归纳偏置,以在策略中引入位置无关的行为(参见观测状态部分)

版本历史

  • v5

    • 最低 mujoco 版本现在是 2.3.3。

    • 添加了对使用 xml_file 参数的完全自定义/第三方 mujoco 模型的支持(以前只能对现有模型进行少量更改)。

    • 添加了 default_camera_config 参数,这是一个用于设置 mj_camera 属性的字典,主要用于自定义环境。

    • 添加了 env.observation_structure,这是一个用于指定观测空间组成(例如 qpos, qvel)的字典,对构建 MuJoCo 环境的工具和封装器很有用。

    • 使用 reset() 返回非空的 info,以前返回的是空字典,新键与 step() 的状态信息相同。

    • 添加了 frame_skip 参数,用于配置 dtstep() 的持续时间),默认值因环境而异,请查阅环境文档页面。

    • 恢复了 xml_file 参数(在 v4 中已移除)。

    • info["reward_run"] 重命名为 info["reward_forward"] 以与其他环境保持一致。

  • v4:所有 MuJoCo 环境现在都使用 mujoco >= 2.1.3 中的 MuJoCo 绑定。

  • v3:支持 gymnasium.make 的 kwargs,例如 xml_filectrl_cost_weightreset_noise_scale 等。RGB 渲染来自跟踪摄像机(因此智能体不会跑出屏幕)。已移至 gymnasium-robotics 仓库

  • v2:所有连续控制环境现在都使用 mujoco-py >= 1.50。已移至 gymnasium-robotics 仓库

  • v1:基于机器人的任务的最大时间步数提高到 1000。向环境中添加了 reward_threshold。

  • v0:初始版本发布。