Patch Notes — 2026-08-17 — 探针修复 + 消融开关 + 推理偏置降级
背景
监控发现每个 worker 日志都打印 [CATTN] fwd=0 ⚠ 概念注意力未参与前向,
但实际训练 forward 走的是 attention_forward_concept_ctx(长上下文记忆版),
不是 attention_forward_concept(简单版)。旧探针只统计后者,所以报假警报。
代码审查同时发现:
- C3 loss / 概念图 / 概念注意力三套机制从未做过消融对比,无法量化真实贡献
- 推理时
graph_concept_bias强度硬编码 1.0,训练早期 wte 余弦基本是噪声, 过强偏置可能加剧生成漂移(README 自己承认 step 1400+ 仍乱码)
本次补丁解决这三个问题,不改默认行为(除推理偏置强度从 1.0 → 0.2)。
改动
1. 修探针(runtime/lal_runtime.c)
ConceptAttnStats增加字段:forwards_ctx、ctx_memory_slots_used、ctx_total_attendattention_forward_concept_ctx函数开头增加计数器累加(含等效全注意力对数)concept_attn_probe_print新格式:[CATTN] fwd_simple=X fwd_ctx=Y ...(同时显示两个版本)[CATTN-CTX] 平均候选/前向=... (含概念槽=..., sink+window=...) 槽命中率=...(ctx 专属)fwd=0警报只在两个计数都为 0 时触发,并多打cctx_enable和cg状态便于诊断
2. 加消融开关(models/ste_train.c)
三个环境变量,启动时读取一次:
| 变量 | 作用 |
|---|---|
LAL_NO_C3=1 |
禁用 C3 训练辅助 loss(保留概念图与概念注意力) |
LAL_NO_CGRAPH=1 |
禁用概念图构建 + 长上下文记忆(训练时不构建图,推理时也不构建) |
LAL_CONCEPT_ATTN=0 |
禁用概念注意力(已有开关,回退标准 sliding attention) |
启动时会打印 [ABLATION] 块列出所有生效的消融开关。
3. 降推理偏置强度(models/ste_train.c)
graph_concept_bias 强度从硬编码 1.0 改为环境变量 LAL_GRAPH_BIAS 控制,默认 0.2。
| 取值 | 含义 |
|---|---|
LAL_GRAPH_BIAS=0.0 |
完全关闭偏置(纯自回归 baseline) |
LAL_GRAPH_BIAS=0.2 |
默认(温和偏置,自回归主信号为主) |
LAL_GRAPH_BIAS=1.0 |
旧行为(激进,用于复现历史实验) |
启动时打印 [*] graph_concept_bias strength = 0.200 (env LAL_GRAPH_BIAS, default 0.2)
默认行为变化
- 训练:默认行为不变(C3 + 概念图 + 概念注意力 全开)
- 推理:
graph_concept_bias强度 1.0 → 0.2(更温和,预期改善生成多样性)
推荐实验
A. 概念图机制总贡献(消融实验)
跑两个 worker,其他配置完全相同,对比同 step 的 CE/boundary/core_diff:
# 实验组 (关 C3 + 关概念图 + 关概念注意力): 纯 sliding attention baseline
LAL_NO_C3=1 LAL_NO_CGRAPH=1 LAL_CONCEPT_ATTN=0 \
OMP_NUM_THREADS=4 LAL_WORKER_ID=0 ./ste_train > worker_ablate.log 2>&1 &
# 对照组 (默认全开)
OMP_NUM_THREADS=4 LAL_WORKER_ID=1 ./ste_train > worker_full.log 2>&1 &
同 step 100/200/500 对比 loss=, avg=, boundary=, core_diff=,量化概念图机制净贡献。
B. 推理偏置强度扫描
固定一个 ckpt,扫描不同强度看生成质量:
for s in 0.0 0.1 0.2 0.5 1.0; do
echo "=== strength=$s ==="
LAL_GRAPH_BIAS=$s ./ste_train --diagnose-only --resume model_mp_0.ste \
--prompt "什么是火?" --max-gen 40
done
人眼对比不同 strength 下的生成连贯度,找最佳点。
C. 探针验证
默认全开模式下跑 10 步训练,应看到:
fwd_simple=0(训练时不用简单版)fwd_ctx > 0(训练时实际走的是 ctx 版)- 不再有
fwd=0 ⚠假警报
如果 fwd_simple=0 && fwd_ctx=0 才是真没在跑概念注意力,那才是 bug。
已知不变
C3_LOSS_WEIGHT = 0.02(未改)logic_lr = 0.001(未改)- 概念图 K=5、cache 文件名
concept_graph_cache_w{id}.bin(未改) - 滑动窗口 attn_window=1024 / attn_sink=64(未改)
- 三值量化 {-1,0,+1}(未改)
部署步骤
# 1. 拉取
cd /work/home/scntp6qxqe/lalmodel && git pull
# 2. 重新编译
make
# 3. 不重启训练的情况下验证编译产物
./ste_train --help | head -5
# 4. (可选) 重启 1 个 worker 验证新探针输出
# kill 当前 worker 0, 启动新版, 跑 20 步, 看 [CATTN] 行不再报 fwd=0
回滚
git revert <commit-hash> 即可。三个改动都是独立的,逻辑上不互相依赖。