arXiv:2409.02513cs.CV2024-09被引 2

用结构化知识指导图像建模,提升深度估计与语义分割性能

SG-MIM: Structured Knowledge Guided Efficient Pre-training for Dense Prediction

  • 通过特征级关系引导,分步融合结构知识与图像信息
  • 在KITTI、NYU-v2、ADE20k上显著优于现有方法
  • 无需额外标注,适配单目深度与语义分割任务

掩码图像建模(MIM)技术重塑了计算机视觉领域,使预训练模型在众多任务中表现卓越。然而,现有MIM方法多依赖单图输入,难以捕捉关键的结构化信息,限制了其在精细特征表达任务中的潜力。为此,本文提出SG-MIM框架,通过融合结构化知识与图像信息,增强密集预测任务的表现。该方法采用轻量级关系引导机制,在特征层面独立引导结构知识,而非传统多模态方法中在像素层面直接拼接。同时,设计选择性掩码策略,最大化通用表征学习与结构知识特定学习之间的协同效应。本方法无需额外标注,具有广泛适用性。在KITTI、NYU-v2和ADE20k数据集上的实验表明,该方法在单目深度估计与语义分割任务中均表现优越。

原文摘要 · Abstract (English)

Masked Image Modeling (MIM) techniques have redefined the landscape of computer vision, enabling pre-trained models to achieve exceptional performance across a broad spectrum of tasks. Despite their success, the full potential of MIM-based methods in dense prediction tasks, particularly in depth estimation, remains untapped. Existing MIM approaches primarily rely on single-image inputs, which makes it challenging to capture the crucial structured information, leading to suboptimal performance in tasks requiring fine-grained feature representation. To address these limitations, we propose SG-MIM, a novel Structured knowledge Guided Masked Image Modeling framework designed to enhance dense prediction tasks by utilizing structured knowledge alongside images. SG-MIM employs a lightweight relational guidance framework, allowing it to guide structured knowledge individually at the feature level rather than naively combining at the pixel level within the same architecture, as is common in traditional multi-modal pre-training methods. This approach enables the model to efficiently capture essential information while minimizing discrepancies between pre-training and downstream tasks. Furthermore, SG-MIM employs a selective masking strategy to incorporate structured knowledge, maximizing the synergy between general representation learning and structured knowledge-specific learning. Our method requires no additional annotations, making it a versatile and efficient solution for a wide range of applications. Our evaluations on the KITTI, NYU-v2, and ADE20k datasets demonstrate SG-MIM's superiority in monocular depth estimation and semantic segmentation.

图像建模深度估计结构知识预训练

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。