VAE 的目标函数
拆开 reconstruction 与 KL,并说清每一项约束谁。
为什么需要它
两项失衡会导致只重建或只贴近 prior。
开始前应该已经会回答
- 能解释 approximate posterior
- 会手算 μ=0,logvar=0 的 KL
输入、输出与契约
- 输入
- x, μ, logvar, reconstruction
- 输出
- scalar ELBO loss
- reconstruction 与 KL 分项记录
- batch/reduction convention 一致
- 只报告 total loss
- 把 KL 符号写反
用最小例子检查
单维 μ=0,logvar=0 时手算 KL;再令 μ=1 比较。
如果你看到……
优先检查:公式符号/reduction
优先检查:posterior collapse
预测 → 实验 → 证据
- 先预测
- 增大 KL 权重会更强约束 latent,但可能牺牲 reconstruction。
- 做实验
- 固定模型与 batch,比较两个 KL weight 的分项轨迹。
- 留证据
- recon
- KL
- latent statistics
渐进提示
先独立排查 10–15 分钟,再按需展开。
提示 1
先单维检查
提示 2
分项打印
提示 3
确认 logvar 不是 std
闭卷检查 · VAE 的目标函数
- KL 边界正确
- 分项可复现
- 能解释 trade-off