Qwen2.5-7B NLA Affect AV

Experimental activation verbalizer (AV): a full-parameter SFT of Qwen2.5-7B-Instruct that maps a 3,584-dimensional layer-20 activation to a structured natural-language explanation with context, affect, and local-token sections.

This repository contains only the latest Exp19 checkpoint: one SFT pass over 5,000 examples, then a fresh-optimizer continuation over 2,400 additional examples (7,400 unique AV examples total).

Important usage note

This is not a normal text-generation checkpoint. The activation vector must be injected at token id 149705 with scale 150 using the forward hook in the accompanying training or evaluation code. Exact metadata and prompt templates are in nla_meta.yaml.

Evaluation

On 200 fresh Ultra-FineWeb prefixes with zero exact overlap with the 10k training set:

  • Valid <explanation> format: 200/200.
  • Full AV -> AR cycle FVE: 0.03765, bootstrap 95% CI [-0.00674, 0.07194].
  • Full-cycle mean cosine: 0.65645.
  • Shuffled full-cycle FVE: -0.74131.

The model is fluent and activation-conditioned but the full-cycle FVE is low; it should be treated as an SFT warm start for joint NLA RL, not as a finished interpretability tool.

Training

  • Target layer: raw model.layers[20], no final RMSNorm.
  • Dense full-parameter transformer-block SFT, not LoRA.
  • FP16, PyTorch SDPA, gradient checkpointing.
  • Effective global batch: 33 on three V100-32GB GPUs.
  • Exp17: 152 optimizer steps, peak LR 7e-6.
  • Exp19: 73 optimizer steps, peak LR 3.5e-6, fresh optimizer.

See MaxChess/nla-affect-10k for exact splits and source code.

Downloads last month
11
Safetensors
Model size
8B params
Tensor type
F32
·
F16
·
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support

Model tree for MaxChess/qwen2.5-7b-nla-affect-av

Base model

Qwen/Qwen2.5-7B
Finetuned
(2983)
this model

Dataset used to train MaxChess/qwen2.5-7b-nla-affect-av