wuff-mann/CNN-WaveAttention-v2

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 对齐微调。