轨迹 Transformer
此模型仅处于维护模式,因此我们不会接受任何更改其代码的新 PR。
如果您在运行此模型时遇到任何问题,请重新安装支持此模型的最后一个版本:v4.30.0。您可以通过运行以下命令来执行此操作:pip install -U transformers==4.30.0
。
概述
轨迹 Transformer 模型由 Michael Janner、Qiyang Li 和 Sergey Levine 在 离线强化学习作为一项大型序列建模问题 中提出。
论文中的摘要如下
强化学习(RL)通常关注估计平稳策略或单步模型,利用马尔可夫性质将问题在时间上分解。但是,我们也可以将 RL 视为一个通用的序列建模问题,其目标是产生一系列导致一系列高奖励的动作。从这个角度来看,人们不禁要考虑在自然语言处理等其他领域表现良好的高容量序列预测模型是否也能为 RL 问题提供有效的解决方案。为此,我们探讨了如何利用序列建模工具来解决 RL,使用 Transformer 架构对轨迹上的分布进行建模,并将束搜索重新用作规划算法。将 RL 作为序列建模问题简化了一系列设计决策,使我们能够摒弃离线 RL 算法中常见的许多组件。我们展示了这种方法在长周期动力学预测、模仿学习、目标条件 RL 和离线 RL 中的灵活性。此外,我们还表明,这种方法可以与现有的无模型算法相结合,在稀疏奖励、长周期任务中产生最先进的规划器。
此模型由 CarlCochet 贡献。原始代码可以在这里找到 这里.
使用技巧
此 Transformer 用于深度强化学习。要使用它,您需要从所有先前时间步的动作、状态和奖励创建序列。此模型将把所有这些元素一起视为一个大的序列(轨迹)。
TrajectoryTransformerConfig
class transformers.TrajectoryTransformerConfig
< source >( vocab_size = 100 action_weight = 5 reward_weight = 1 value_weight = 1 block_size = 249 action_dim = 6 observation_dim = 17 transition_dim = 25 n_layer = 4 n_head = 4 n_embd = 128 embd_pdrop = 0.1 attn_pdrop = 0.1 resid_pdrop = 0.1 learning_rate = 0.0006 max_position_embeddings = 512 initializer_range = 0.02 layer_norm_eps = 1e-12 kaiming_initializer_range = 1 use_cache = True pad_token_id = 1 bos_token_id = 50256 eos_token_id = 50256 **kwargs )
参数
- vocab_size (
int
, 可选, 默认值为 100) — TrajectoryTransformer 模型的词汇量。定义了调用 TrajectoryTransformerModel 时传递的trajectories
可以表示的不同标记的数量 - action_weight (
int
, 可选, 默认值为 5) — 损失函数中动作的权重 - reward_weight (
int
, 可选, 默认值为 1) — 损失函数中奖励的权重 - value_weight (
int
, 可选, 默认值为 1) — 损失函数中值的权重 - block_size (
int
, 可选, 默认值为 249) — 轨迹转换器中块的大小。 - action_dim (
int
, 可选, 默认值为 6) — 动作空间的维度。 - observation_dim (
int
, 可选, 默认值为 17) — 观测空间的维度。 - transition_dim (
int
, 可选, 默认值为 25) — 转移空间的维度。 - n_layer (
int
, 可选, 默认值为 4) — Transformer 编码器中隐藏层的数量。 - n_head (
int
, 可选, 默认值为 4) — Transformer 编码器中每个注意力层的注意力头的数量。 - n_embd (
int
, 可选, 默认值为 128) — 嵌入和隐藏状态的维度。 - resid_pdrop (
float
, 可选, 默认值为 0.1) — 嵌入、编码器和池化器中所有全连接层的 dropout 概率。 - embd_pdrop (
int
, 可选, 默认值为 0.1) — 嵌入的 dropout 比例。 - attn_pdrop (
float
, 可选, 默认值为 0.1) — 注意力的 dropout 比例。 - hidden_act (
str
或function
, 可选, 默认值为"gelu"
) — 编码器和池化器中的非线性激活函数(函数或字符串)。 如果是字符串,则支持"gelu"
、"relu"
、"selu"
和"gelu_new"
。 - initializer_range (
float
, 可选, 默认为 0.02) — 初始化所有权重矩阵的截断正态初始化的标准差。 - layer_norm_eps (
float
, 可选, 默认为 1e-12) — 层归一化层使用的epsilon。 - kaiming_initializer_range (`float, 可选, 默认为 1) — 用于 EinLinear 层的 kaiming 初始化整流器的负斜率的系数缩放。
- use_cache (
bool
, 可选, 默认为True
) — 模型是否应返回最后一个键/值注意力(并非所有模型都使用)。 仅当config.is_decoder=True
时才相关。 示例 —
这是用于存储 TrajectoryTransformerModel 配置的配置类。 它用于根据指定的参数实例化 TrajectoryTransformer 模型,定义模型架构。 使用默认值实例化配置将产生与 TrajectoryTransformer CarlCochet/trajectory-transformer-halfcheetah-medium-v2 架构类似的配置。
配置对象继承自 PretrainedConfig 并且可以用于控制模型输出。 阅读 PretrainedConfig 的文档以获取更多信息。
>>> from transformers import TrajectoryTransformerConfig, TrajectoryTransformerModel
>>> # Initializing a TrajectoryTransformer CarlCochet/trajectory-transformer-halfcheetah-medium-v2 style configuration
>>> configuration = TrajectoryTransformerConfig()
>>> # Initializing a model (with random weights) from the CarlCochet/trajectory-transformer-halfcheetah-medium-v2 style configuration
>>> model = TrajectoryTransformerModel(configuration)
>>> # Accessing the model configuration
>>> configuration = model.config
TrajectoryTransformerModel
class transformers.TrajectoryTransformerModel
< 源代码 >( config )
参数
- config (TrajectoryTransformerConfig) — 包含模型所有参数的模型配置类。 使用配置文件初始化不会加载与模型关联的权重,而只会加载配置。 查看 from_pretrained() 方法以加载模型权重。
输出原始隐藏状态的裸 TrajectoryTransformer 模型转换器,没有任何特定的头部。 此模型是 PyTorch torch.nn.Module 子类。 将其用作常规 PyTorch 模块,并参考 PyTorch 文档了解有关一般使用和行为的所有事宜。
完整的 GPT 语言模型,上下文大小为 block_size
forward
< 源代码 >( trajectories: Optional = None past_key_values: Optional = None targets: Optional = None attention_mask: Optional = None use_cache: Optional = None output_attentions: Optional = None output_hidden_states: Optional = None return_dict: Optional = None ) → transformers.models.deprecated.trajectory_transformer.modeling_trajectory_transformer.TrajectoryTransformerOutput
或 tuple(torch.FloatTensor)
参数
- trajectories (
torch.LongTensor
形状为(batch_size, sequence_length)
) — 轨迹批次,其中轨迹是状态、动作和奖励的序列。 - past_key_values (
Tuple[Tuple[torch.Tensor]]
长度为config.n_layers
,可选) — 包含模型计算的预先计算的隐藏状态(注意块中的键和值)(见下面的past_key_values
输出)。可用于加快顺序解码。传递给此模型的input_ids
应该不作为input_ids
传递,因为它们已经计算过了。 - targets (
torch.LongTensor
形状为(batch_size, sequence_length)
,可选) — 用于计算损失的目标。 - attention_mask (
torch.FloatTensor
形状为(batch_size, sequence_length)
,可选) — 掩码,以避免对填充令牌索引执行注意力。掩码值选择在[0, 1]
中:- 1 表示未掩码的令牌,
- 0 表示掩码的令牌。
- use_cache (
bool
,可选) — 如果设置为True
,则返回past_key_values
键值状态,可用于加快解码(见past_key_values
)。 - output_attentions (
bool
,可选) — 是否返回所有注意力层的注意力张量。有关详细信息,请参阅返回张量中的attentions
。 - output_hidden_states (
bool
,可选) — 是否返回所有层的隐藏状态。有关详细信息,请参阅返回张量中的hidden_states
。 - return_dict (
bool
,可选) — 是否返回 ModelOutput 而不是普通元组。
返回
transformers.models.deprecated.trajectory_transformer.modeling_trajectory_transformer.TrajectoryTransformerOutput
或 tuple(torch.FloatTensor)
transformers.models.deprecated.trajectory_transformer.modeling_trajectory_transformer.TrajectoryTransformerOutput
或 torch.FloatTensor
的元组(如果传递了 return_dict=False
或当 config.return_dict=False
时),包含根据配置 (TrajectoryTransformerConfig) 和输入的不同元素。
- loss (
torch.FloatTensor
形状为(1,)
,可选,在提供labels
时返回) — 语言建模损失。 - logits (
torch.FloatTensor
形状为(batch_size, sequence_length, config.vocab_size)
) — 语言建模头的预测分数(SoftMax 之前的每个词汇表令牌的分数)。 - past_key_values (
Tuple[Tuple[torch.Tensor]]
,可选,在传递use_cache=True
或config.use_cache=True
时返回) — 长度为config.n_layers
的元组,包含形状为(batch_size, num_heads, sequence_length, embed_size_per_head)
的张量元组。包含预先计算的隐藏状态(注意块中的键和值),可用于(见past_key_values
输入)加快顺序解码。 - hidden_states (
tuple(torch.FloatTensor)
,可选,在传递output_hidden_states=True
或config.output_hidden_states=True
时返回) —torch.FloatTensor
的元组(一个用于嵌入的输出,一个用于每一层的输出),形状为(batch_size, sequence_length, hidden_size)
。每一层的模型输出处的隐藏状态加上初始嵌入输出。 - attentions (
tuple(torch.FloatTensor)
,可选,在传递output_attentions=True
或config.output_attentions=True
时返回) —torch.FloatTensor
的元组(每一层一个),形状为(batch_size, num_heads, sequence_length, sequence_length)
。注意力 Softmax 后的 GPT2Attentions 权重,用于计算自注意力头中的加权平均值。
TrajectoryTransformerModel 正向方法,覆盖 __call__
特殊方法。
虽然正向传递的配方需要在此函数中定义,但应该在之后调用 Module
实例,而不是此函数,因为前者负责运行预处理和后处理步骤,而后者则默默忽略它们。
例子
>>> from transformers import TrajectoryTransformerModel
>>> import torch
>>> model = TrajectoryTransformerModel.from_pretrained(
... "CarlCochet/trajectory-transformer-halfcheetah-medium-v2"
... )
>>> model.to(device)
>>> model.eval()
>>> observations_dim, action_dim, batch_size = 17, 6, 256
>>> seq_length = observations_dim + action_dim + 1
>>> trajectories = torch.LongTensor([np.random.permutation(self.seq_length) for _ in range(batch_size)]).to(
... device
... )
>>> targets = torch.LongTensor([np.random.permutation(self.seq_length) for _ in range(batch_size)]).to(device)
>>> outputs = model(
... trajectories,
... targets=targets,
... use_cache=True,
... output_attentions=True,
... output_hidden_states=True,
... return_dict=True,
... )