跳转至

LEMMA: Laplacian Pyramids for Efficient Marine Semantic Segmentation

会议: CVPR 2026
arXiv: 2603.25689
代码: 无
领域: 语义分割
关键词: 轻量化语义分割, 拉普拉斯金字塔, 海洋语义分割, 边缘检测, 无人水面艇

一句话总结

提出LEMMA,一种基于拉普拉斯金字塔的轻量级海洋语义分割模型,通过金字塔分解提取边缘信息来替代深层特征计算,在参数量减少71倍的条件下实现了SOTA级别的分割精度(MaSTr1325上98.97% mIoU)。

研究背景与动机

海洋场景的语义分割对于无人水面艇(USV)自主导航和沿海地球观测(如油污检测)至关重要。然而,现有的语义分割方法(如WaSR-T、DeepLabv3等)通常依赖深层CNN或Transformer架构,拥有数千万甚至上亿的参数量和极高的计算开销,难以在无人机、USV等资源受限的边缘设备上实时运行。

核心矛盾在于:海洋场景需要高精度分割(水面反射、薄油膜等低对比度区域),但部署平台(无人机/USV)的算力极为有限。现有方法在精度和效率之间无法兼顾——WaSR-T虽然达到99.80% mIoU,但需要71.4M参数和133.8 GFLOPs。

本文的切入角度是利用拉普拉斯金字塔分解天然提供的边缘信息。金字塔的各层包含了不同分辨率下的边缘细节,这些信息可以在特征提取的早期阶段就被注入,从而避免在深层网络中进行昂贵的特征图计算。核心idea:用拉普拉斯金字塔的边缘先验替代深层特征提取,实现轻量化与高精度的兼得。

方法详解

整体框架

LEMMA 要解决的问题是:海洋分割既要看清水面反射、薄油膜这类低对比度的边缘,又得跑在无人机/USV 这种几乎没算力的平台上。它的思路是把"提边缘"这件事从网络深处搬到输入端——先用拉普拉斯金字塔把图像拆成不同分辨率的高频细节,再让网络只负责精炼和融合这些现成的边缘,而不是从头学。

具体地,输入图像被分解成深度为 3 的拉普拉斯金字塔 \(L_1\)\(L_2\)\(L_3\)\(L_3\) 分辨率最低、\(L_1\) 最高),三个分支由低到高接力处理:Low-level Feature Branch (LFB) 先吃下最粗的 \(L_3\);Middle-level Feature Branch (MFB) 把 \(L_2\) 和 LFB 的输出拼起来继续精炼;High-level Feature Branch (HFB) 再把 \(L_1\) 连同前两支的特征一起融合,在最高分辨率上直接生成分割掩码。分支之间靠级联拼接(concat)把粗尺度的语义和细尺度的边缘对齐,靠转置卷积把低分辨率特征上采样回去——整个网络没有深层 backbone,参数量被压到 1M 级别。

%%{init: {'flowchart': {'rankSpacing': 24, 'nodeSpacing': 28, 'padding': 6, 'wrappingWidth': 400}}}%%
flowchart TD
    IN["输入图像"] --> LP["拉普拉斯金字塔分解(深度 3)<br/>L₃ 最粗 / L₂ / L₁ 最细<br/>各层 = 该尺度的边缘先验"]
    LP -->|L₃| LFB["三分支由粗到细:LFB 低层分支<br/>64 通道 · 可配置残差块 ×NRBL"]
    LFB -->|转置卷积上采样 + concat L₂| MFB["MFB 中层分支<br/>可配置残差块 ×NRBM"]
    MFB -->|上采样 + concat L₁| HFB["HFB 高层分支(最高分辨率)<br/>仅 16 通道 · 可配置残差块 ×NRBH"]
    HFB --> OUT["分割掩码"]

关键设计

1. 拉普拉斯金字塔分解:把边缘先验直接喂给网络,省掉深层学边缘的开销

海洋场景里区分水面、障碍物、油污,靠的几乎全是边缘和高频纹理;而常规做法是堆一个几千万参数的 backbone,让网络在很深的层里慢慢学出这些边缘,代价是巨大的特征图计算。LEMMA 反过来用拉普拉斯金字塔在输入端一次性把多尺度边缘拿出来——金字塔每一层本身就是"原图减去高斯模糊后上采样"的残差,天然存的就是该分辨率下的高频细节,相当于免费的边缘先验。网络拿到的不再是原始 RGB,而是已经分好尺度的边缘图,于是不必再用深度去换边缘表征,这是它能做到 1M 参数还保住精度的根本原因。一个额外的好处是,金字塔做的是高频提取,太阳眩光、水面反射这类缓慢变化的低频光照漂移会被隐式压掉,对海面这种强反光场景尤其有用。

2. 三分支由粗到细接力 (LFB/MFB/HFB):把通道预算花在刀刃上

光有金字塔还不够,关键是怎么把三层边缘融合成一张干净的掩码,同时不让高分辨率层的计算量爆掉。LEMMA 让三个分支按分辨率从低到高接力:LFB 在最粗的 \(L_3\) 上用较宽的 64 通道做重处理(这里分辨率低,宽一点也不贵),MFB 承接中间尺度,到了分辨率最高、最吃算力的 HFB,通道数反而砍到只有 16。因为高分辨率特征图上的 GFLOPs 和通道数成正比,把宽度留在小图、把窄通道用在大图,正好避开了最贵的那块开销;而实验也证实 16 通道已经足够在最高分辨率上重建出掩码。分支之间用 concat 而不是相加来融合,是为了原样保留每一层的边缘信息、不被求和稀释掉,再交给后续卷积去取舍。

3. 可配置的残差块链:用一个旋钮按数据集调精度/参数的平衡

不同视角的海洋数据对"网络该多深"的需求并不一样——地面 USV 视角和无人机航拍视角的难点分布不同。LEMMA 没有写死深度,而是在每个分支里放可配置数量(NRBL / NRBM / NRBH)的残差块,每个块是 conv–LeakyReLU–conv 再加一条残差连接的标准结构。通过消融可以为每个数据集挑出最优配置:MaSTr1325 是 7/7/1,Oil Spill 是 6/7/4。值得注意的是 MaSTr1325 上 HFB 只配 1 个块——再往 HFB 加块反而掉点,印证了高分辨率层"轻处理就够"的判断;调深度的红利主要在中低分辨率分支上。

损失函数 / 训练策略

  • 使用Focal Loss作为损失函数,在两个数据集上均优于Dice Loss和CE+Dice组合
  • 使用Adam优化器,batch size为8,训练300个epoch
  • 在NVIDIA TESLA P100上训练,推理使用NVIDIA 2080和Intel 4-core XEON CPU

实验关键数据

主实验

数据集 指标 本文(LEMMA) 之前SOTA 提升
MaSTr1325 mIoU 98.97% 99.91% (BEMRF-Net) -0.94%(但参数少71x)
MaSTr1325 参数量 1.07M 71.4M (WaSR-T) 减少66.7x
MaSTr1325 GFLOPs 17.83 156.0 (BEMRF-Net) 减少88.5%
MaSTr1325 推理时间 7.3ms 47.55ms (DeepLabv3) 减少84.65%
Oil Spill mIoU 93.42% 92.66% (R-GSSNet) +0.76%
Oil Spill 参数量 1.01M 62.6M (R-Segformer) 减少62x

消融实验

配置 关键指标 说明
残差块 7/7/1 (MaSTr1325) mIoU 98.96% 最优配置,增加HFB块数反而降低性能
残差块 6/7/4 (Oil Spill) mIoU 93.42% 最优配置
Focal Loss vs Dice Loss 98.97% vs 98.72% Focal Loss在两个数据集上均最优
Focal Loss vs CE+Dice 98.97% vs 98.86% 验证Focal Loss的优势

关键发现

  • LEMMA在参数量仅1M左右的情况下,可以与拥有数千万参数的模型(如WaSR-T的71.4M)性能相当
  • 模型在USV地面视角(MaSTr1325)和无人机航拍视角(Oil Spill)两种截然不同的视角下均表现优异,展示了跨平台鲁棒性
  • HFB使用16个通道就足够完成高分辨率掩码重建,这是降低计算量的关键设计
  • 拉普拉斯金字塔能隐式抑制低频光照漂移(如太阳眩光、水面反射)

亮点与洞察

  • 将传统图像处理技术(拉普拉斯金字塔)与深度学习残差网络巧妙结合,用物理先验减少学习负担
  • 极致的轻量化:1M参数即可达到接近SOTA的精度,适合在无人机/USV等资源受限设备上实时部署
  • 跨平台通用性好:同一个架构既适用于地面USV障碍物检测,也适用于航拍油污分割
  • 不需要ImageNet预训练,从头训练即可达到高性能

局限与展望

  • 反射/波浪/眩光等环境因素会影响拉普拉斯金字塔的质量,导致失败(论文展示了反射导致的失败案例)
  • 当前使用固定金字塔层数和静态残差块配置,未来可探索自适应金字塔深度分配
  • 数据集规模有限(MaSTr1325仅1325张,Oil Spill仅847张),难以验证在大规模场景下的泛化能力
  • 与WaSR-T等最强模型在精度上仍有约1%的差距

相关工作与启发

  • vs WaSR-T: WaSR-T使用Transformer达到99.80% mIoU,但需要71.4M参数;LEMMA在1.07M参数下达98.97%,效率提升数十倍
  • vs DeepLabv3: DeepLabv3达97.67% mIoU需48M参数和123 GFLOPs;LEMMA以1/45参数量超越其性能
  • vs LETNet: 同为轻量模型,LETNet 83.18% mIoU,LEMMA用相近参数量(1.07M vs 0.94M)提升了近16个百分点
  • 启发: 传统CV技术(金字塔、边缘检测)与深度学习结合可以在特定领域实现极致轻量化

评分

  • 新颖性: ⭐⭐⭐ 拉普拉斯金字塔用于分割不算全新,但在海洋场景的落地和三分支设计有新意
  • 实验充分度: ⭐⭐⭐⭐ 两个数据集、大量baselines对比、详细消融实验
  • 写作质量: ⭐⭐⭐⭐ 结构清晰,动机明确,实验分析充分
  • 价值: ⭐⭐⭐⭐ 对边缘设备部署的海洋分割有直接实用价值