消融实验与 baseline:以 RainMamba 为例 · 学习笔记
素材来源:RainMamba: Enhanced Locality Learning with State Space Models for Video Deraining(ACM Multimedia 2024 Oral,arXiv 2407.21773)
主题范畴:深度学习方法论 · 消融实验 / baseline / Mamba-SSM
整理日期:2026-10-02
关联阅读:模块缝合与人物蒸馏笔记(同系列方法论篇)
目录
- 摘要
- 一、术语定位:什么是"消融",什么是"基线"
- 1.1 消融(Ablation)的词源与本义
- 1.2 baseline(基线)及其语境漂移
- 二、背景知识:Mamba 与 SSM
- 2.1 Mamba 不是缩写
- 2.2 Mamba 的核心贡献
- 2.3 SSM 用于视觉任务的共性缺陷
- 三、RainMamba:缝合式方法论的典型样本
- 3.1 任务与命名
- 3.2 三个核心组件
- 3.3 Hilbert 扫描的数学依据
- 四、消融实验数据精读(Table 5,VRDS 数据集)
- 4.1 递进式消融设计
- 4.2 主消融结果
- 4.3 增益瀑布拆解
- 4.4 长序列消融(Table 9,NTURain 数据集)
- 五、方法论:如何用消融实验掂量一篇缝合论文
- 5.1 消融展示的是"缝合增益分解",不是"对骨干的贡献"
- 5.2 判断"可迁移贡献"的标准
- 5.3 读消融表的两个小心眼
- 5.4 读缝合论文的操作流程
- 六、常见误区
- 附录 A:术语速查表
- 附录 B:自查清单(读缝合论文五问)
(待生成)
摘要
消融实验源自医学术语,本质是控制变量法:逐个移除组件,以性能损失量化贡献;基线在消融与对比实验中含义不同。本文以 ACM MM 2024 去雨论文 RainMamba 为样本,解读 Hilbert 扫描与差分对比学习如何修补 SSM 的局部性缺陷,并拆解三组件共 2.69 dB 增益。
一、术语定位:什么是"消融",什么是"基线"
1.1 消融(Ablation)的词源与本义
"消融"原为医学术语(ablation),指通过切除或烧蚀身体某块组织——如心脏消融手术、神经科学中的脑区切除——并观察功能损失,来反推该组织原本承担的职能。深度学习借用了这一思想:
把模型的某个模块"切掉",观察性能下降多少,用性能损失反推该模块的贡献。
其方法论本质即科研中的控制变量法:除被移除组件外,训练数据、超参数、算力配置全部保持一致,每次只动一个零件。
1.2 baseline(基线)及其语境漂移
baseline 指用于比对的参照模型,但在论文不同位置含义会漂移:
| 出现语境 | baseline 通常指 | RainMamba 中的实例 |
|---|---|---|
| 消融实验(Ablation Study) | 拆到最简的裸模型 | M1(去掉全部三个核心组件的基础网络) |
| 对比实验(Comparison) | 领域内已有的他人方法 | ESTINet 等已有视频去雨模型 |
读论文时扫一眼表头即可确认所指。相关高频术语:SOTA(State-of-the-Art,当前最优)、w/o ×××(without,去掉某组件,消融表最常见标记)、upper bound(性能上界)。
二、背景知识:Mamba 与 SSM
2.1 Mamba 不是缩写
Mamba 取自黑曼巴蛇(black mamba),延续 Python 生态的蛇系命名传统,是 2023 年 Albert Gu 与 Tri Dao 提出的架构代号(NeurIPS 2023,Mamba: Linear-Time Sequence Modeling with Selective State Spaces)。真正构成技术缩写的是两个词:
- SSM(State Space Model,状态空间模型):其技术内核;
- S4(Structured State Space Sequence Model):其前身,这个才是缩写。
2.2 Mamba 的核心贡献
在 SSM 基础上引入两项改进:
- 选择性机制(selective SSM):状态参数随输入内容动态变化;
- 硬件感知并行扫描(parallel scan):使一维序列可高效并行训练。
其卖点是线性复杂度完成长序列建模,对标 Transformer 注意力的二次复杂度,序列越长收益越大。
2.3 SSM 用于视觉任务的共性缺陷
SSM 天生只接受一维序列。将视频(三维张量:时间 × 高 × 宽)拉平为一条序列时,相邻像素在序列上被扯散,局部相关性被破坏——这是所有视频类 Mamba 模型的共性痛点,也是 RainMamba 的切入点。
三、RainMamba:缝合式方法论的典型样本
3.1 任务与命名
任务为视频去雨(去除雨条 rain streaks 与雨滴附着物 raindrops)。命名遵循"任务 + 骨架"缝合式起名法:Rain + Mamba。注意:其 selective SSM 算子原封未动,改动集中在算子之外。
3.2 三个核心组件
| 缩写 | 全称 | 机制 | 角色 |
|---|---|---|---|
| GMB | Global Mamba Block | Zigzag(逐行光栅)扫描,全局时空依赖 | 粗到细的"粗" |
| LMB | Local Mamba Block | 3D Hilbert 曲线扫描,序列级局部性 | 粗到细的"细" |
| DCL | Difference-guided dynamic Contrastive Locality learning | 差分引导的动态对比学习,patch 级自相似正则 | 训练期策略 |
3.3 Hilbert 扫描的数学依据
论文以膨胀因子 DF(dilation factor,空间-线性比的上界)刻画扫描方式的局部性保持能力:
- Hilbert 曲线:DF = 6,为常数,与分辨率无关;
- Zigzag 曲线:DF = 4ⁿ − 2ⁿ⁺¹ + 2,随曲线阶数 n 发散至无穷。
即分辨率越高,Hilbert 扫描保持局部性的优势越大。二者分工:Zigzag 偏全局相关性,Hilbert 偏局部相关性,构成互补(分别对应 GMB 与 LMB)。
四、消融实验数据精读(Table 5,VRDS 数据集)
4.1 递进式消融设计
论文采用递进构建(progressive build-up)而非单组件移除:
- M1:移除全部三个核心组件的基础网络(消融基线);
- M2:M1 + GMB;
- M3:M1 + LMB;
- M4:M1 + GMB + LMB(完整粗到细模块);
- Ours:M4 + DCL(完整模型)。
4.2 主消融结果
| 模型 | GMB | LMB | DCL | PSNR↑ | SSIM↑ | LPIPS↓ | GFLOPs | 参数量(M) |
|---|---|---|---|---|---|---|---|---|
| M1 | — | — | — | 29.35 | 0.9118 | 0.0893 | 49.82 | 30.23 |
| M2 | ✓ | — | — | 30.86 | 0.9277 | 0.0774 | 84.40 | 32.49 |
| M3 | — | ✓ | — | 31.04 | 0.9302 | 0.0753 | 84.40 | 32.49 |
| M4 | ✓ | ✓ | — | 31.79 | 0.9361 | 0.0704 | 118.99 | 34.75 |
| Ours | ✓ | ✓ | ✓ | 32.04 | 0.9376 | 0.0684 | 118.99 | 34.75 |
4.3 增益瀑布拆解
从 M1 到完整模型共+2.69 dB,各组件净贡献:
| 步骤 | 增量 | 说明 |
|---|---|---|
| M1 → M2(+GMB) | +1.51 dB | 验证 SSM 捕捉长程依赖的能力 |
| M2 → M4(+LMB) | +0.93 dB | 粗到细设计的互补增益 |
| M4 → Ours(+DCL) | +0.25 dB | 零开销增益:GFLOPs 与参数量完全不变 |
三条关键读表结论:
- M3 > M2 是最干净的对照:同为 84.40 GFLOPs,Hilbert(31.04 dB)胜 Zigzag(30.86 dB)——控制算力、只换扫描方式的单变量实验,直接支撑"Hilbert 更有效"的主张;
- DCL 是性价比最高的零件:纯训练期正则,推理零成本,属于缝合时值得优先积攒的组件类型;
- 扫描方向不敏感(Table 7):time-first / height-first / width-first 三种 Hilbert 方向仅差 0.06 dB,说明增益来自 Hilbert 机制本身而非方向运气。
4.4 长序列消融(Table 9,NTURain 数据集)
| 输入帧数 | 7 | 20 | 50 | 80 | 110 |
|---|---|---|---|---|---|
| PSNR↑ | 37.534 | 37.806 | 37.858 | 37.870 | 37.875 |
| 显存占用 | 5,082M | 9,530M | 20,056M | 32,056M | 41,100M |
帧数增加性能单调提升且边际递减、显存近似线性增长——这正是 SSM 线性复杂度优势的实证:换成 Transformer,110 帧的注意力矩阵开销将不可承受。
五、方法论:如何用消融实验掂量一篇缝合论文
5.1 消融展示的是"缝合增益分解",不是"对骨干的贡献"
RainMamba 未改动 SSM 算子本身,其贡献在于SSM 的视频应用方法论:修补"一维化破坏局部性"这一通用缺陷(Hilbert 扫描)与训练策略(DCL)。措辞上应区分"对 SSM 算子的贡献"与"对 SSM 使用方式的贡献"。
5.2 判断"可迁移贡献"的标准
单一任务消融只能证明"在视频去雨上有用"。要主张通用贡献,需跨任务、跨模型验证(如将 Hilbert 扫描迁移到视频分割、修复或其他视频 Mamba 模型)。可迁移性是缝合零件价值的试金石。
5.3 读消融表的两个小心眼
- 协同效应:单独移除某组件只掉 0.2 dB,不代表它不重要——组件间可能配合工作,单变量消融会低估协同价值;
- 坐标轴截断:论文图表常截断指标轴(如从 88 而非 0 开始),视觉差距被放大,务必看具体数值。
5.4 读缝合论文的操作流程
- 看瀑布图拆解:每个零件各贡献多少 dB / 点数;
- 算力对齐检查:找同 FLOPs 的对照行(如 M3 vs M2);
- 零开销组件优先关注(如 DCL 类训练期策略);
- 问一句:这个零件能否搬到别的任务/模型?
六、常见误区
- 误以为 Mamba 是缩写——它是蛇名,S4 与 SSM 才是缩写;
- 误以为消融证明因果——消融只证明"去掉后变差",模块间的交互与替代路径仍需设计实验排除;
- 误以为 baseline 含义唯一——消融表与对比表中的 baseline 指向不同对象;
- 误以为"贡献于 SSM"——任务级增益与骨干级改进是两个层级的贡献,审稿人会严格区分。
附录 A:术语速查表
| 术语 | 全称 / 含义 |
|---|---|
| Ablation Study | 消融实验,逐个移除组件量化贡献 |
| baseline | 基线,参照模型(语境决定具体所指) |
| SOTA | State-of-the-Art,当前最优 |
| w/o | without,消融表中去掉某组件的标记 |
| SSM | State Space Model,状态空间模型 |
| S4 | Structured State Space Sequence Model |
| selective SSM | 选择性状态空间模型,Mamba 的核心机制 |
| GMB / LMB / DCL | RainMamba 三组件:全局 Mamba 块 / 局部 Mamba 块 / 差分引导对比局部学习 |
| DF | Dilation Factor,膨胀因子,扫描局部性度量 |
| PSNR / SSIM / LPIPS | 去雨/修复任务三大指标:峰值信噪比 / 结构相似度 / 感知距离 |
附录 B:自查清单(读缝合论文五问)
- 消融基线拆到多简?是否含"全无"参照?
- 每个零件的净增益与增量算力分别是多少?
- 是否存在同算力的干净对照?
- 有无零开销的纯训练期组件?
- 核心零件是否做过跨任务 / 跨模型迁移验证?
本笔记由对话蒸馏整理,数据均引自论文原文 Table 5 / 7 / 9。