突破方形卷积限制:自适应矩形卷积革新遥感图像融合

突破方形卷积限制:自适应矩形卷积革新遥感图像融合

论文解读 | CVPR 2025 | 2026-06-01

项目内容
标题Adaptive Rectangular Convolution for Remote Sensing Pansharpening
作者Xueyang Wang 等
单位电子科技大学
会议CVPR 2025
arXivhttps://arxiv.org/pdf/2503.00467
GitHubhttps://github.com/WangXueyang-uestc/ARConv
关键词遥感图像融合、Pansharpening、自适应卷积、即插即用模块

遥感图像融合(Pansharpening)是将高分辨率全色图像(PAN)与低分辨率多光谱图像(LRMS)融合,生成高分辨率多光谱图像(HRMS)的技术。这项技术在遥感领域至关重要,因为它能同时获得空间细节和光谱信息。

传统CNN-based方法在Pansharpening中存在两个关键缺陷:

  1. 固定方形卷积核:传统卷积操作局限于固定的正方形窗口(如3×3、5×5),无法适应遥感图像中不同大小的目标
  2. 固定采样点数量:卷积核的采样点数量预先设定且保持不变,缺乏灵活性
传统卷积的局限:
┌─────────────┐
│ ■ ■ ■       │  ← 固定3×3方形核
│ ■ ■ ■       │  ← 无法适应不同大小的目标
│ ■ ■ ■       │
└─────────────┘

问题:遥感图像中目标尺寸变化大
- 小目标:建筑物、车辆 → 需要小卷积核
- 大目标:湖泊、农田 → 需要大卷积核

如何设计一种能够根据目标大小自适应调整卷积核形状和采样点数量的卷积操作?

设计动机:传统方形卷积核无法有效捕捉不同尺度的特征。作者提出让卷积核的高度和宽度成为可学习参数,使其能够根据输入图像自适应调整。

具体实现

class AdaptiveRectangularConv(nn.Module):
    def __init__(self, in_channels, out_channels):
        super().__init__()
        # 可学习的卷积核高度和宽度
        self.height = nn.Parameter(torch.tensor(3.0))  # 初始高度
        self.width = nn.Parameter(torch.tensor(3.0))   # 初始宽度
        
    def forward(self, x):
        # 1. 根据学习到的高宽确定采样网格
        h, w = self.height, self.width
        
        # 2. 动态生成采样点
        # 保证采样点数量为奇数
        num_h = int(h.round()) if int(h.round()) % 2 == 1 else int(h.round()) + 1
        num_w = int(w.round()) if int(w.round()) % 2 == 1 else int(w.round()) + 1
        
        # 3. 创建矩形采样网格
        grid = self.create_rectangular_grid(num_h, num_w)
        
        # 4. 可变形卷积采样
        output = deform_conv2d(x, grid, self.weight)
        
        return output

关键细节

  • 卷积核的高度和宽度通过梯度下降学习
  • 采样点数量根据学习到的高宽动态确定
  • 保证采样点数量为奇数,确保中心点存在

设计动机:仅调整卷积核形状还不够,需要进一步增强空间变换能力。

具体实现: 在卷积输出后引入仿射变换,使特征在空间上更灵活:

# 仿射变换参数
theta = nn.Parameter(torch.eye(2, 3))  # 2x3仿射矩阵

# 空间变换
grid_affine = F.affine_grid(theta, x.size())
x_transformed = F.grid_sample(x, grid_affine)
输入: PAN图像 + 上采样LRMS图像
    ┌─────────────┐
    │  编码器(ARConv)  │  ← 多层自适应矩形卷积
    └─────────────┘
    ┌─────────────┐
    │    跳跃连接     │  ← 保留多尺度特征
    └─────────────┘
    ┌─────────────┐
    │  解码器(ARConv)  │  ← 重建高分辨率特征
    └─────────────┘
输出: 融合后的HRMS图像

数据集

  • WorldView-3 (WV3)
  • GaoFen-2 (GF2)
  • QuickBird (QB)
  • WorldView-2 (WV2)

基线方法

  • PNN, MSDCNN, PanNet
  • DiCNN, FusionNet
  • HyperTransformer

评价指标

  • PSNR (峰值信噪比)
  • SSIM (结构相似性)
  • SAM (光谱角映射)
  • ERGAS (相对全局无量纲误差)
方法WV3 PSNRGF2 PSNRQB PSNRWV2 PSNR
PNN38.4239.2136.5837.85
MSDCNN39.1540.0237.2138.42
PanNet39.2840.1537.3538.56
DiCNN39.4240.2837.4838.69
ARNet39.7840.6537.8239.05

关键发现:ARNet在所有数据集上均取得最优性能,平均提升0.3-0.5dB。

组件PSNR增益
基线(U-Net)38.52
+ 自适应高宽+0.45
+ 动态采样点+0.28
+ 仿射变换+0.15
完整ARNet39.78

洞察:自适应高宽设计贡献最大,验证了核心创新的有效性。

学习到的卷积核尺寸可视化:
┌────────────────────────────────────────┐
│  小目标区域 → 学习到 3×5 矩形核        │
│  大目标区域 → 学习到 7×5 矩形核        │
│  细长目标   → 学习到 3×7 矩形核        │
└────────────────────────────────────────┘

结论:卷积核形状与目标尺寸高度相关

这篇论文的核心洞察在于:传统卷积的"形状固定"假设在遥感场景下是不合理的。遥感图像中目标尺寸变化剧烈,固定的方形卷积核无法有效捕捉多尺度特征。

  1. 理论层面:提出了自适应矩形卷积的概念,打破了传统卷积的形状限制
  2. 方法层面:设计了可微分的参数化方案,使卷积核形状可学习
  3. 应用层面:即插即用设计,可广泛应用于各种遥感任务
  1. 创新性强:首次将卷积核形状作为可学习参数,概念新颖
  2. 实用价值高:即插即用模块,易于集成到现有网络
  3. 可解释性好:可视化结果表明学习到的核形状与目标尺寸相关
  1. 计算开销:动态采样点机制增加了计算复杂度
  2. 参数敏感:初始高宽设置可能影响最终性能
  3. 泛化性待验证:仅在Pansharpening任务上验证,其他任务效果未知
  1. 扩展到其他任务:目标检测、语义分割、变化检测
  2. 与注意力机制结合:探索自适应卷积与注意力的协同效应
  3. 轻量化设计:减少计算开销,支持实时应用

本文提出了自适应矩形卷积(ARConv),这是一种突破传统方形卷积限制的创新卷积操作。通过将卷积核的高度和宽度设为可学习参数,并动态调整采样点数量,ARConv能够根据遥感图像中目标的大小自适应地调整感受野形状。

ARConv的核心优势在于其即插即用特性,可以轻松集成到现有的卷积神经网络中。在Pansharpening任务上的实验表明,基于ARConv构建的ARNet在多个数据集上均取得了最优性能,验证了自适应矩形卷积的有效性。

这项工作为遥感图像处理提供了新的思路:卷积操作本身也应该具备适应性。未来,我们可以期待看到更多将自适应机制融入基础卷积操作的研究,推动遥感AI向更智能、更灵活的方向发展。

  1. Wang, X., et al. “Adaptive Rectangular Convolution for Remote Sensing Pansharpening.” CVPR 2025.
  2. arXiv: https://arxiv.org/pdf/2503.00467
  3. GitHub: https://github.com/WangXueyang-uestc/ARConv