arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2302.06148cs.CV
收藏

CoMAE:小规模RGB-D数据集上的单模型混合预训练

CoMAE: Single Model Hybrid Pre-training on Small-Scale RGB-D Datasets

State Key Laboratory for Novel Software Technology, Nanjing University(南京大学计算机软件新技术国家重点实验室);MYbank, Ant Group(蚂蚁集团网商银行)

Jiange Yang, Sheng Guo, Gangshan Wu, Limin Wang

摘要:

当前的RGB-D场景识别方法通常使用相同的Places或ImageNet预训练训练RGB和深度模态的两个独立骨干。然而,预先训练的深度网络仍然受到基于RGB的模型的影响,这可能导致次优解决方案。在本文中,我们提出了一种用于RGB和深度模态的单模型自监督混合预训练框架,称为CoMAE。我们的CoMAE提出了一个课程学习策略,以统一两个流行的自监督表示学习算法:对比学习和掩码图像建模。具体来说,我们首先建立一个补丁级对齐任务,通过跨模态对比学习预训练两种模态共享的单个编码器。然后,将预先训练的对比编码器传递到多模态掩码自编码器,以从生成的角度捕获更精细的上下文特征。此外,我们的单模型设计,而不需要融合模块是非常灵活和强大的推广到单峰的情况下,在训练和测试阶段。在SUN RGB-D和NYUDv 2数据集上的大量实验证明了我们的CoMAE对于RGB和深度表示学习的有效性。此外,我们的实验结果表明,CoMAE是一个数据有效的表示学习器。虽然我们只使用小规模和未标记的训练集进行预训练,但我们的CoMAE预训练模型仍然可以与超大规模和监督式RGB数据集预训练的最先进方法竞争。代码将在https://github.com/MCG-NJU/CoMAE上发布。

英文摘要:

Current RGB-D scene recognition approaches often train two standalone backbones for RGB and depth modalities with the same Places or ImageNet pre-training. However, the pre-trained depth network is still biased by RGB-based models which may result in a suboptimal solution. In this paper, we present a single-model self-supervised hybrid pre-training framework for RGB and depth modalities, termed as CoMAE. Our CoMAE presents a curriculum learning strategy to unify the two popular self-supervised representation learning algorithms: contrastive learning and masked image modeling. Specifically, we first build a patch-level alignment task to pre-train a single encoder shared by two modalities via cross-modal contrastive learning. Then, the pre-trained contrastive encoder is passed to a multi-modal masked autoencoder to capture the finer context features from a generative perspective. In addition, our single-model design without requirement of fusion module is very flexible and robust to generalize to unimodal scenario in both training and testing phases. Extensive experiments on SUN RGB-D and NYUDv2 datasets demonstrate the effectiveness of our CoMAE for RGB and depth representation learning. In addition, our experiment results reveal that CoMAE is a data-efficient representation learner. Although we only use the small-scale and unlabeled training set for pre-training, our CoMAE pre-trained models are still competitive to the state-of-the-art methods with extra large-scale and supervised RGB dataset pre-training. Code will be released at https://github.com/MCG-NJU/CoMAE.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑