【问题标题】:Can we use GPT-2 sentence embedding for classification tasks?我们可以将 GPT-2 句子嵌入用于分类任务吗?
【发布时间】:2020-06-19 18:46:19
【问题描述】:

我正在尝试在句子分类任务中使用转换器嵌入没有微调它们。我使用过 BERT 嵌入,这些实验给了我非常好的结果。现在我想使用 GPT-2 嵌入(无需微调)。所以我有两个问题,

  1. 我可以使用这样的 GPT-2 嵌入吗(因为我知道 Gpt-2 是 从左到右训练)
  2. GPT-2 在 生成任务以外的分类任务?
  3. 如果我可以使用 GPT-2embeddings,我应该怎么做?

【问题讨论】:

  • 其实我用的是罗伯塔。我为句子情感分类任务测试了冻结的 Roberta 嵌入,即使没有微调也能正常工作。
  • 我正在使用 Roberta 嵌入并通过分类头(全连接层)发送所有这些嵌入的平均值。
  • 如上所示,请分享代码。
  • 您在这方面取得了进展吗?我也想知道是否可以使用 GPT-3 进行类似的文本分类。
  • @ShamaneSiriwardhana 你的问题解决了吗?

标签: nlp huggingface-transformers gpt-2


【解决方案1】:

我基本上解决了这个问题。这里我使用了从 GPT-2 中提取的嵌入。

  1. 所以是的,我们可以使用 GPT-2 嵌入序列的最终标记作为类标记。由于从左到右的自注意​​力机制,最终的token可以表示顺序信息。

  2. 请查看以下 GitHub 问题,了解使用 GPT-2 嵌入的实现。 github issue

  3. 我进行了比较 GPT-2 嵌入和 RoBERTa 嵌入的实验。我只使用 RoBERTa 嵌入而不是 GPT-2 获得了更好的结果。

【讨论】:

    猜你喜欢
    • 2019-12-06
    • 2019-07-28
    • 2022-12-09
    • 1970-01-01
    • 2021-12-06
    • 1970-01-01
    • 1970-01-01
    • 1970-01-01
    • 2023-02-18
    相关资源
    最近更新 更多