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