← 返回 2026-03-09

SLER-IR:用于一体化图像复原的球面层级专家路由框架 SLER-IR: Spherical Layer-wise Expert Routing for All-in-One Image Restoration

Peng Shurui, Xin Lin, Shi Luo, Jincen Ou, Dizhe Zhang, Lu Qi, Truong Nguyen, Chao Ren 📅 2026-03-06 👍 1 2026-07-13 08:35
Mixture-of-Experts 专家混合 图像复原 对比学习 层级路由 球面嵌入

用球面退化嵌入 + 层级专家路由做一体化图像复原

前置知识

All-in-One 图像复原

指用一个统一模型同时处理多种退化类型(如去噪、去雾、去雨、去模糊、低光照增强)的图像复原范式,目标是摆脱单任务专用网络对每种退化分别训练的限制。代表性方法如 AirNet、PromptIR、InstructIR、DA-CLIP、MoCE-IR 等。

这是本文要解决的核心问题域,必须先理解为什么要做 all-in-one、难点在哪里(任务冲突、特征干扰、退化表征偏差),才能看懂 SLER-IR 的设计动机。

Mixture-of-Experts (MoE)

把网络某一层的参数拆成多个并行的「专家」子网络,由路由器(gating)根据输入动态选择激活哪些专家,从而在不显著增加推理成本的前提下获得更大的模型容量与灵活性。本文每层有 $C=3$ 个候选专家。

SLER-IR 的整个架构核心是层级 MoE(每层独立选专家,8 层共 $3^8 = 6561$ 条推理路径),如果不懂 MoE 就无法理解 routing vector、expert center、softmax gating 等概念。

超球面(Hypersphere)嵌入与 L2 归一化

将特征向量做 $\hat{\mathbf{f}} = \mathbf{f}/\|\mathbf{f}\|_2$ 的 L2 归一化,使其落在单位超球面上;此时向量间相似度用余弦相似度衡量,即 $\cos(\theta) = \hat{\mathbf{f}}^\top \hat{\mathbf{c}}$,只与方向有关,与模长无关。

这是 SLER-IR 区别于其他线性空间退化表征方法的关键。没有这一归一化,路由器会受 embedding 几何偏差(class-distance bias)污染,导致相似度门控选错专家。

对比学习与三元组损失(Triplet-Constrained Contrastive Loss)

通过构造正/负样本对,最小化同类样本距离同时最大化异类样本距离。文中损失形式为 $\mathcal{L}_{HC,l} = \max\left(0, \mathbb{E}_N[S_{i,j,l}] - \mathbb{E}_P[S_{i,j,l}] + \lambda\right)$,其中 $S_{i,j,l}=\hat{\mathbf{f}}_{i,l}^\top \hat{\mathbf{f}}_{j,l}$ 是归一化后样本对相似度,$\lambda$ 为 margin。

是 hyperspherical degradation representation 的训练目标,正是它把同类退化在球面上拉紧、异类推开,从而为路由提供几何一致的相似度信号。

CLIP-based 退化/内容编码器

借鉴 CLIP 的双塔结构,分别用 Content CLIP 提取语义内容表征、用 Degraded CLIP 提取退化线索。本文中从 Degraded CLIP 取出 CLS token 作为退化特征向量 $\mathbf{f}$,再送入 MLP 路由头。

理解输入怎么变成路由向量,必须先知道 CLIP CLS token 是怎么来的,以及为什么用 L2 归一化之后可以做余弦门控。

两阶段训练(Stage 1 概率路由 + Stage 2 确定性路由)

第一阶段(15 epoch)用 softmax 得到每个专家的选中概率(软路由),使所有专家都能收到梯度、稳定学习;第二阶段(80 epoch)冻结路由器,按 argmax 硬选专家并只训练复原网络,让专家彻底专精化。

这是 SLER-IR 训练流程的关键节奏。如果直接用硬路由会让多数专家得不到更新、导致退化解锁不稳定;概率路由先打基础再硬切换,是它能在五任务上稳定领先的原因之一。

研究动机

现有的 all-in-one 图像复原方法主要分两类范式,且都存在明显缺陷。第一类是特征调制类(AirNet、PromptIR、InstructIR、DA-CLIP 等),通过 prompt 或对比学习在一个统一的参数空间内微调特征。这种方法的最大问题是任务间特征干扰严重——以去噪和去模糊为例,前者需要抑制高频噪声,后者恰恰需要恢复高频细节,两者在共享 backbone 中相互拉扯,难以做精细的退化特异建模。第二类是基于专家的架构(WM-MoE、LoRA-IR、DaAIR、AMIR、MEASNet、MoCE-IR 等),用 Mixture-of-Experts 激活任务相关子模块,但已有方法大多把专家限制在局部模块(只换 head 或只换 bottleneck),没有充分利用退化先验;而且它们的路由机制往往假设专家与退化一一对应(rigid mapping),无法处理复合退化,同时在非约束的线性 embedding 空间里,类间距离会出现几何偏差(class-distance bias),导致基于相似度的门控选错专家。除此之外,现有方法普遍忽略两类重要问题:空间非均匀退化(雨线、局部雾)以及训练 crop 与全分辨率推理之间的粒度差距(granularity gap)。

本文的目标是本文目标是提出一个统一的、既能支持细粒度退化识别、又能动态组合专家路径的 all-in-one 复原框架 SLER-IR,具体要实现三件事:1)把 encoder–decoder 的每一层都扩展为多个参数独立的专家节点,并通过层级路由让网络产生 $3^8 = 6561$ 条组合推理路径,从而在不增加推理开销的前提下覆盖复合退化;2)用 hyperspherical embedding 解决线性退化表征的几何偏差,配合三元组对比损失提升退化的可分性;3)设计 Global–Local Granularity Fusion (GLGF) 模块,把全局 CLS token 与局部 patch token 融合,弥合训练 crop 与全分辨率推理的粒度差距,对空间非均匀退化也能稳健路由。

与已有工作不同的是,本文的独特切入角度是同时从路由结构、表征几何和粒度三个层面联合切入,而不是只修其中一块。与 MoCE-IR 等现有专家方法相比,SLER-IR 的根本差异在于:1)专家是层级、参数独立的而非只在局部模块替换;2)路由信号建立在球面 embedding 之上,用余弦相似度选专家,规避线性空间的类间距离偏差;3)训练分两阶段(概率→确定性)配合三元组对比损失,让路由从软选择过渡到硬专精,渐进形成退化特异的组合路径;4)额外引入 GLGF 模块补足空间非均匀退化和 train–test 粒度差异这块之前被忽视的空白。整体定位是「几何一致 + 层级组合 + 全局-局部融合」三位一体。

核心方法

SLER-IR 沿用 PromptIR 的 4-level encoder–decoder 主干(每层 Transformer block 数为 $[4,6,6,8]$,共 8 层有路由决策),把每个 backbone block 扩展为 $C=3$ 个参数独立的专家节点。直觉上:与其让一个共享 backbone 兼顾所有退化类型、然后靠 prompt 弱调制,不如让每一层根据输入退化类型挑选一个最合适的「小专家」,从而获得 $3^8 = 6561$ 条潜在推理路径,组合出退化特异的处理流程。但这条路要跑通,必须解决三个子问题:路由信号从哪儿来(退化表征)、路由怎么决策(门控准则)、决策稳不稳定(训练策略)。SLER-IR 的回答是——用 Degraded CLIP 的 CLS token 加 MLP-L2 得到球面路由向量 $\mathbf{f}\in\mathbb{R}^{K\times d}$;用 cosine similarity + softmax 在球面上挑专家(球面均匀退化嵌入);并设计两阶段训练(Stage 1 概率软路由 + Stage 2 确定性硬路由)配合 GLGF 模块提供全图粒度的内容-退化先验。整体流程可以概括为:CLIP 退化特征 → 球面路由向量 → 层级专家动态选择 → 同时由 GLGF 输出 patch 级复原先验注入 backbone → 重建图像。

核心创新点是把「专家路由」建立在球面退化表征之上,与已有方法的本质区别体现在三点。第一,结构层面,MoCE-IR、WM-MoE 等只在局部模块替换专家,本文是「层级」专家——每一层 backbone block 都扩展为 $C=3$ 个参数独立的专家,推理时每层硬选一个,共 $3^8 = 6561$ 条组合路径,不增加推理开销但获得指数级的路径多样性。第二,几何层面,本文首次把退化表征显式 L2 归一化到单位超球面上做余弦门控,配合三元组对比损失 $\mathcal{L}_{HC,l}$ 约束,使类间在角度空间均匀分离、类内紧凑;而传统方法在无约束线性空间中学表征,存在类距离偏差(Fig.5(b) 的反例:denoise 与 dehaze 被推得过远),会误导相似度路由。第三,粒度层面,提出的 GLGF 模块通过 4×4 crop 网格提取局部退化 CLS token,rearrange 成空间分布图(DSP map),再与全图 Content CLIP 的 CLS token 通过 FiLM 调制后的 CSP map 交叉注意力融合,给 backbone 提供 patch 级的、content-aware 的退化先验,专门解决 train crop 与 full-res 推理的粒度差距以及空间非均匀退化。

方法步骤详情

完整方法分四步。第一步,特征提取与路由向量构造:取 Content CLIP 编码器的全图 CLS token $\mathbf{c}$ 与 patch token $\mathbf{T}_c$;取 Degraded CLIP 编码器对全图的 CLS token,送入 MLP 后做 L2 归一化得到层级路由向量 $\mathbf{f}=\mathrm{L2}(\mathrm{MLP}(\mathrm{CLIP}(x)))\in\mathbb{R}^{K\times d}$。同时为专家中心 $\mathbf{C}\in\mathbb{R}^{C\times d}$ 也做 L2 归一化,使其落在单位超球面上,并通过 Monte Carlo 采样 + 均匀性约束让中心在球面上彼此互不相似(spherical-code 性质)。第二步,路由门控决策:在第 $l$ 层计算余弦相似度 $\mathrm{Sim}_{i,j}=\mathbf{f}_i^\top\mathbf{C}_j^\top/(\|\mathbf{f}_i\|_2\|\mathbf{C}_j\|_2)$,行 softmax 得概率 $\mathbf{p}_{i,:}=\mathrm{Softmax}(\mathrm{Sim}_{i,:})$。Stage 1 用软概率让所有专家收梯度,配合 $\mathcal{L}_1 + \alpha\cdot\mathcal{L}_{HC,l}$ 联合优化 15 epoch;Stage 2 冻结 router,按 $y_i=\arg\max_j \mathbf{p}_{i,j}$ 硬选专家,只用 $\mathcal{L}_1$ 再训 80 epoch。第三步,球面退化对比学习:在每一层对归一化后的样本对相似度 $S_{i,j,l}=\hat{\mathbf{f}}_{i,l}^\top\hat{\mathbf{f}}_{j,l}$ 计算 triplet contrastive loss $\mathcal{L}_{HC,l}=\max(0,\mathbb{E}_N[S]-\mathbb{E}_P[S]+\lambda)$,整体损失 $\mathcal{L}=\mathcal{L}_1 + \alpha\cdot\mathcal{L}_{HC}$(默认 $\alpha=1, \lambda$ 为 margin),迫使同类退化在球面上聚集、异类均匀分开。第四步,GLGF 注入:GLMC 把全图 Content CLIP 的 CLS token 经 MLP 投影生成 $(\gamma,\beta)$ 对 patch token 做 FiLM 调制得 CSP map;同时把 4×4 crop 各自的退化 CLS token rearrange 成 $E\in\mathbb{R}^{4\times 4\times d}$,上采样到 patch 网格得 DSP map $D\in\mathbb{R}^{H\times W\times d}$。CGDF 用 cross-attention(prior 的 query、patch 的 K/V)融合得 patch 级复原先验 $\mathbf{F}'$,再以引导 cross-attention 注入到第 $l$ 层复原特征 $\mathbf{F}_l$,实现局部自适应的复原。

技术新颖性

技术新颖性可以从四个维度看。第一,把 MoE 从「局部模块替换」提升到「层级 backbone 全替换」,并显式枚举 $3^8=6561$ 条组合路径,把 all-in-one 复原从 prompt 微调范式推进到 compositional inference 范式,这在之前的工作中没有完整做过。第二,把退化表征显式约束到单位超球面上并提出 hyperspherical uniform embedding + triplet contrastive loss 的组合,专门解决 class-distance bias 问题,论文 Fig.5(b) 给出反例并量化说明偏差如何误导相似度门控。第三,训练流程上提出两阶段概率→确定性路由切换,避免了硬路由早期多数专家得不到更新的「专家饥饿」问题,是工程上能稳定跑通五任务复原的关键。第四,GLGF 模块首次把「训练 crop 与全分辨率推理的粒度差异」以及「空间非均匀退化」这两点通过 4×4 grid + content FiLM + cross-attention prior 注入的方式系统解决,给后续 all-in-one 复原工作开辟了一个新的模块设计方向(content-guided degradation fusion)。四个点叠加之后才让 SLER-IR 在 PSNR/SSIM 上系统性地超过 MoCE-IR、VLU-Net 等最新基线。

SLER-IR 框架总览
Fig. 3: SLER-IR 框架总览
不同退化下动态路由轨迹可视化
Fig. 4: 不同退化下动态路由轨迹可视化
现有退化路由方法与本文 hyperspherical 表征的对比
Fig. 5: 现有退化路由方法与本文 hyperspherical 表征的对比
GLGF 模块架构图
Fig. 6: GLGF 模块架构图

实验结果

三任务 all-in-one(Table 1):SLER-IR 在 SOTS(去雾)/ Rain100L(去雨)/ CBSD68(去噪,$\sigma=15/25/50$)上平均达到 33.14/0.922,比 MoCE-IR 的 32.73/0.917 高 0.41 dB 和 0.005 SSIM;去雾一项比 MoCE-IR 高 1.66 dB(33.00 vs 31.34),是表中最大单项提升;去噪在所有噪声水平上一致领先($\sigma=15$: 34.25 vs 34.11;$\sigma=25$: 31.68 vs 31.45;$\sigma=50$: 28.45 vs 28.18)。五任务 all-in-one(Table 2):SLER-IR 平均 31.73/0.928,比之前的 MoCE-IR(30.58/0.919)高 1.15 dB 和 0.009,去雾 +2.59 dB(33.43 vs 30.84)、GoPro 去模糊 +1.22 dB(31.27 vs 30.05)、LOL 低光照 +0.96 dB(23.96 vs 23.00),每一项都是当前 SOTA。消融(Table 3)显示 $\mathcal{L}_1$ 单独只有 28.30/0.879,加 classifier loss 28.76/0.884,再换为本文的 hyperspherical contrastive loss 直接跳到 31.73/0.928,验证球面对比损失是最大单一提升源;Table 4 显示 GLGF 模块从 30.64/0.917 提升到 31.73/0.928,贡献 +1.09 dB。Table 5 与 Table 6 的专家数和对比损失权重搜索表明 $C=3$、$\alpha=1$ 是最优组合,$C=2$ 容量不够(31.22),$C=4$ 计算增加但同训练预算下无明显收益(31.32);$\alpha=0.5$ 监督不足、$\alpha=2$ 过强削弱复原目标。视觉对比(Fig.7)覆盖低光、去模糊、去雾、去雨四类场景,SLER-IR 在亮度均衡、纹理锐度、颜色还原、雨线去除全面优于对比方法,没有明显视觉伪影。

三任务 all-in-one 图像复原在 SOTS(去雾)、Rain100L(去雨)、CBSD68(去噪)上的量化对比
Table 1: 三任务 all-in-one 图像复原在 SOTS(去雾)、Rain100L(去雨)、CBSD68(去噪)上的量化对比
损失函数消融
Table 3: 损失函数消融
模型组件消融(GLGF)
Table 4: 模型组件消融(GLGF)
Hyperspherical 对比损失权重 $\alpha$ 消融
Table 6: Hyperspherical 对比损失权重 $\alpha$ 消融
四种模型(AirNet、Restormer、MoCE-IR、SLER-IR)在五类复原任务上的 PSNR/SSIM 量化对比柱状图
Fig. 2: 四种模型(AirNet、Restormer、MoCE-IR、SLER-IR)在五类复原任务上的 PSNR/SSIM 量化对比柱状图
SLER-IR 与当前 SOTA 在低光、去模糊、去雾、去雨任务上的可视化对比
Fig. 7: SLER-IR 与当前 SOTA 在低光、去模糊、去雾、去雨任务上的可视化对比
查看结构化数据
任务指标本文基线提升
三任务平均 (Dehaze+Derain+Denoise) PSNR/SSIM 33.14 / 0.922 MoCE-IR 32.73 / 0.917 +0.41 dB / +0.005
去雾 (SOTS, 三任务设置) PSNR/SSIM 33.00 / 0.982 MoCE-IR 31.34 / 0.979 +1.66 dB / +0.003
去雨 (Rain100L, 三任务设置) PSNR/SSIM 38.34 / 0.983 VLU-Net 38.93 / 0.984 -0.59 dB(次优,但平均仍最高)
去噪 (CBSD68 σ=15/25/50, 三任务平均) PSNR/SSIM 34.25/31.68/28.45 InstructIR 34.15/31.52/28.30 +0.10/+0.16/+0.15 dB(三个噪声水平一致领先)
五任务平均 (Dehaze+Derain+Denoise+Deblur+Low-light) PSNR/SSIM 31.73 / 0.928 MoCE-IR 30.58 / 0.919 +1.15 dB / +0.009
去雾 (SOTS, 五任务设置) PSNR/SSIM 33.43 / 0.985 VLU-Net 30.84 / 0.980 +2.59 dB / +0.005
去模糊 (GoPro) PSNR/SSIM 31.27 / 0.918 MoCE-IR 30.05 / 0.899 +1.22 dB / +0.019
低光照增强 (LOL) PSNR/SSIM 23.96 / 0.861 InstructIR / MoCE-IR 23.00 / 0.836–0.852 +0.96 dB / +0.009–0.025
GLGF 消融 (五任务平均) PSNR/SSIM 31.73 / 0.928 (with GLGF) 30.64 / 0.917 (w/o GLGF) +1.09 dB / +0.011
Hyperspherical 对比损失消融 PSNR/SSIM 31.73 / 0.928 (L1 + LHC) 28.76 / 0.884 (L1 + Lcls) +2.97 dB / +0.044

局限与改进

作者在文中没有单独列 Limitations 章节,但从表述与实验设置可以观察到几个客观边界。第一,路由信号完全依赖 CLIP 提取的退化 CLS token,对未见过的退化类型或与预训练 CLIP 分布差距较大的领域(医学图像、卫星遥感、工业缺陷)泛化能力未验证,CLIP 域外偏差会直接污染路由。第二,4×4 grid 划分是写死的硬编码,对超高分辨率(4K、8K)或超细粒度局部退化的适应性有限,论文没有给出分辨率自适应策略。第三,专家数 $C=3$、路由权重 $\alpha=1$ 是在五任务上 grid search 得到的,迁移到更复杂任务组合(>5 类)时是否仍最优未给出实验。第四,Stage 1 概率路由 + Stage 2 确定性路由的两阶段切换依赖经验阈值(15→80 epoch),不同训练数据量下该切换时机可能需要重新调;作者未讨论 auto-switch 或可学习的阶段切换机制。第五,论文只汇报 PSNR/SSIM,没有在感知质量(LPIPS、NICEst)、下游任务(检测/分割)上验证,意味着方法可能在视觉自然度上未必全面优于 InstructIR 等大模型。第六,推理路径虽然是 $3^8$ 组合但每个路径仍走完整 8 层 backbone,路径多样性 ≠ 参数高效,与真正的 sparse activation 相比仍偏重。

独立分析的弱点

独立分析可以看到几处可以打磨的地方。1)路由可解释性弱:论文给出 $\arg\max$ 的硬选专家索引,但没系统分析「哪种退化→哪类专家路径」的对应关系,Fig.4 只放了可视化未量化路径分布。改进方向是增加专家路径混淆矩阵或频率统计,让 routing 的语义可被审计。2)GLGF 的 4×4 grid 划分过粗,对雨线、刮痕这类细长局部退化,16 个 crop 的 CLS token 粒度不够,容易漏检小区域;可改进为可学习的 adaptive grid 或 attention-based sparse pooling。3)专家中心更新用 Monte Carlo + 均匀性约束,没有可学习的中心漂移率,路由中心可能在长尾退化下塌缩;可考虑 EMA 动量更新或加一个 anti-collapse 正则。4)损失权重 $\alpha$ 与阶段切换 epoch 都是 grid search 超参,对新任务组合需要重调;可改成 cosine ramp 或基于 router entropy 的自适应切换。5)CLIP 编码器在推理时占显存与耗时都不小,论文没有报告其相对主干的开销比例,对实时部署是个隐忧;改进方向是蒸馏一个轻量退化编码器。6)评测只覆盖 5 类合成退化,对真实世界复合退化(雨+低光、夜景雾、噪声+运动模糊)没有给出野外数据测试,复合退化的实际增益只能从 Table 2 的间接对比推断,需要补充 paired real-world benchmark。

未来方向

作者明确给出的后续方向是「Code and models will be publicly released」以及把球面退化嵌入推广到其他视觉任务。除此之外,结合本文成果可延伸出几条值得探索的方向:第一,把球面 expert routing 与稀疏激活(top-1 + 共享 expert)结合,在保持 $3^8$ 路径多样性的同时降低推理 FLOPs;第二,用 hyperspherical embedding 替换 CLIP,直接学习一个 task-agnostic 的退化编码器(如基于 ConvNeXt 或 Mamba),去掉对 CLIP 域的依赖,扩展到非自然图像;第三,引入 contrastive routing memory bank 跨 batch 维护负样本,进一步拉开退化类间距离;第四,把 GLGF 的 4×4 crop 划分替换为可学习的多尺度 grid(如 4×4、8×8、16×16 混合),处理不同尺度的局部退化;第五,把两阶段概率→确定性路由扩展为端到端可微 sparse routing(GSmooth / DSDP),避免 stage 切换的人为超参;第六,在下游高层任务(目标检测、分割、跟踪)上验证 SLER-IR 复原是否带来一致的感知增益,而非只看 PSNR/SSIM;第七,与大视觉-语言模型结合,让退化表征具备语言可解释性,支持用户用自然语言指定退化类型。

复现评估

复现性整体中等偏上。架构层面作者沿用 PromptIR 的 4-level encoder-decoder(block 数 $[4,6,6,8]$),与公开代码风格兼容;训练超参清晰(Stage 1: 15 epoch, batch 10;Stage 2: 80 epoch, batch 20;Adam $\beta=(0.9,0.999)$,初始 lr $2\times 10^{-4}$ cosine decay;对比损失 $\alpha=1$、专家数 $C=3$)。但有几个潜在阻碍:1)作者声明「Code and models will be publicly released」但截至阅读时未给出 GitHub 链接,实际公开与否不可知;2)硬件依赖 H20 GPU,且 Stage 1+Stage 2 共 95 epoch,五任务联合训练的实际时长未披露,复现者算力门槛不低;3)依赖 Degraded CLIP 与 Content CLIP 两个 CLIP 编码器,具体使用何种 CLIP backbone(ViT-B/16? ViT-L/14?)以及是否微调、是否冻结未说明,会显著影响结果;4)4×4 crop 划分、FiLM 调制、cross-attention prior 注入的工程细节需要从正文推断,可能存在隐性实现 trick;5)消融只在五任务上做,三任务与复合退化的进一步消融缺失。综合判断:拥有 PromptIR/Restormer 复现经验、且能访问 H20/A100 集群的研究者可以较高效地复现主线结果,但要完全一致复现 Table 1/2 的具体数字仍需对 CLIP 选择、初始化种子、batch 内多任务采样概率做对齐。