Pytorch 交叉熵输入维度

Question

我正在尝试使用 Huggingface 的 BertModel 和 Pytorch 开发二元分类器。分类器模块是这样的：

class SSTClassifierModel(nn.Module):

  def __init__(self, num_classes = 2, hidden_size = 768):
    super(SSTClassifierModel, self).__init__()
    self.number_of_classes = num_classes
    self.dropout = nn.Dropout(0.01)
    self.hidden_size = hidden_size
    self.bert = BertModel.from_pretrained('bert-base-uncased')
    self.classifier = nn.Linear(hidden_size, num_classes)

  def forward(self, input_ids, att_masks,token_type_ids,  labels):
    _, embedding = self.bert(input_ids, token_type_ids, att_masks)
    output = self.classifier(self.dropout(embedding))
    return output

我训练模型的方式如下：

loss_function = BCELoss()
model.train()
for epoch in range(NO_OF_EPOCHS):
  for step, batch in enumerate(train_dataloader):
        input_ids = batch[0].to(device)
        input_mask = batch[1].to(device)
        token_type_ids = batch[2].to(device)
        labels = batch[3].to(device)
        # assuming batch size = 3, labels is something like:
        # tensor([[0],[1],[1]])
        model.zero_grad()        
        model_output = model(input_ids,  
                             input_mask, 
                             token_type_ids,
                             labels)
        # model output is something like: (with batch size = 3) 
        # tensor([[ 0.3566, -0.0333],
                 #[ 0.1154,  0.2842],
                 #[-0.0016,  0.3767]], grad_fn=<AddmmBackward>)

        loss = loss_function(model_output.view(-1,2) , labels.view(-1))

我正在做 .view()s 因为 Huggingface 的 BertForSequenceClassification here 源代码使用完全相同的方法来计算损失。但是我得到这个错误：

/usr/local/lib/python3.6/dist-packages/torch/nn/functional.py in binary_cross_entropy(input, target, weight, size_average, reduce, reduction)
   2068     if input.numel() != target.numel():
   2069         raise ValueError("Target and input must have the same number of elements. target nelement ({}) "
-> 2070                          "!= input nelement ({})".format(target.numel(), input.numel()))
   2071 
   2072     if weight is not None:

ValueError: Target and input must have the same number of elements. target nelement (3) != input nelement (6)

我的标签有问题吗？或者我模型的输出？我真的被困在这里了。 Pytorch 的 BCELoss 文档说：

输入：(N,*) 其中*表示任意数量的附加维度
目标：(N,*)，与输入相同的形状

我应该如何使我的标签与模型输出的形状相同？我觉得我错过了一些巨大的东西，但我找不到它。

Answer 1

观察很少：

您引用的代码使用 CrossEntropyLoss，但您使用的是 BCELoss。
CrossEntropyLoss 采用预测对数（大小：(N,D)）和目标标签（大小：(N,)），而 BCELoss 采用 p(y=1|x)（大小: (N,)) 和目标标签 (size: (N,)) 作为 p(y=0|x) 可以从 p(y=1|x)
CrossEntropyLoss 期望对数，即 BCELoss 期望概率值

解决方案：

因为你传了一个(N,2)张量，所以报错。你只需要传递 p(y=1|x)，因此你可以做

loss = loss_function(model_output.view(-1,2)[:,1] , labels.view(-1))

上面我假设第二个值是p(y=1|x)。

一种更简洁的方法是让模型只输出一个值，即 p(y=1|x) 并将其传递给损失函数。从代码看来，您传递的是 logit 值而不是概率值，因此如果您想使用 BCELoss 或者您可以使用 BCEWithLogitsLoss，您可能还需要计算 sigmoid (model_output)。

另一种方法是将损失更改为 CrossEntropyLoss，这应该也适用，因为它也适用于二进制标签。

Pytorch 交叉熵输入维度

Pytorch cross entropy input dimensions

python

pytorch

python-3.7

cross-entropy

huggingface-transformers