| # CNN-WaveAttention v2 | |
| 基于海洋学波系分割理论 + β-VAE 自剪枝的对象中心海浪方向谱分解/重构模型。 | |
| **v2 相对 v1 的改动(修复 slot 坍缩)**: | |
| - 解码改为能量域非负相加 `E=Σ softplus(content)·mask`(物理可加),堵掉单 slot+tanh 饱和坍缩捷径 | |
| - 新增 batch 级占用均衡正则(load_balance),惩罚"所有像素堆到一个 slot"的坍缩 | |
| - β_max 降一档(3e-4)+ warmup 拉长,让分割先成型再加 KL 压力 | |
| - v1(旧坍缩版)保留在 `wuff-mann/CNN-WaveAttention` 作为 baseline 对照,本仓库与之隔离 | |
| - 架构: CNN 编码器 + Slot Attention(自适应波系分割) + per-slot 变分隐向量(β-VAE 自剪枝) + 逐波系能量域组合解码 | |
| - 输入/输出: E(f,θ) 47×72 | |
| - 训练数据: `wuff-mann/ATLAS-WDS` | |
| - 目录: `checkpoints/`(best/last/编号), `logs/`(history.json + 逐维KL快照), `figures/`(含 slot_usage 坍缩监控), `config.json` | |
| 阶段一:完整模型(能量域组合式解码器)。阶段二:冻结编码器,解码器替换为 cLDM 对齐微调。 | |