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

wanda_row_pruning

wanda_row_pruning

【效率与协作】Compute row-wise Wanda pruning scores, masks, and pruned weights without any weight update.

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

技能说明

GitHub

name: wanda_row_pruning description: Compute row-wise Wanda pruning scores, masks, and pruned weights without any weight update.

Wanda Row Pruning

Use this skill to prune a linear layer with the Wanda rule: score each weight by abs(W_ij) * activation_norm_j, compare scores only within the same output row, and zero the lowest-scoring weights. Do not use it when the experiment requires global layer-wise magnitude thresholds.

Inputs

  • Weight matrix shaped [output_channels][input_channels].
  • Activation norm vector with one entry per input channel.
  • Unstructured sparsity ratio in [0, 1).

Outputs

  • Boolean prune mask where true means the weight is removed.
  • Pruned weight matrix with masked entries set to zero.
  • Score matrix and exact sparsity metadata.

Workflow

  1. Validate the activation vector length against the matrix width.
  2. Compute scores by multiplying every row of abs(W) by the activation norms.
  3. For each output row, deterministically select the lowest floor(width * sparsity_ratio) scores.
  4. Set only masked weights to zero; never alter unmasked values.

Validation

The tests include an outlier-channel case where Wanda keeps a low-magnitude but high-activation weight that magnitude pruning removes.

Limitations

This skill handles one dense matrix. Model traversal and calibration capture are separate modules.

我的收藏