界面已本地化;技能介绍与说明文档按所选语言由 AI 自动翻译,代码、命令、路径和专业术语保留原文,以原作者文档为准。
探索技能
效率与协作 · SKILL.md

pplm_attribute_models

pplm_attribute_models

【效率与协作】Use when a recovery or implementation needs differentiable PPLM attribute scores for BoW or simple discriminator-style control.

独立 GitHub 快照 · 自动分类 · 以原作者文档为准

技能说明

GitHub

name: pplm_attribute_models description: Use when a recovery or implementation needs differentiable PPLM attribute scores for BoW or simple discriminator-style control.

PPLM Attribute Models

Use this skill to build the plug-in attribute model p(a|x) used by PPLM. Do not use it to modify base LM weights.

Inputs

  • Vocabulary tokens.
  • Topic BoW terms or linear classifier weights.
  • Candidate token probabilities or a hidden vector.

Outputs

  • Numeric attribute score.
  • Attribute loss suitable for gradient ascent/descent in a controller.
  • Diagnostics for missing BoW terms.

Workflow

  1. Normalize vocabulary and target words.
  2. For BoW control, score the probability mass assigned to target words using -log(sum p(target)) as a minimization loss.
  3. For discriminator-style control, apply a linear score or softmax outside the base LM.
  4. Return diagnostics; never update the base LM.

Validation

Run python tests/test_attribute_models.py.

Limitations

Reduced proxy implementations must declare that they are not full GPT-2 345M reproduction.

我的收藏