对比ViT与CNN在少样本几何估计中的表现,发现小数据下CNN更优。
A Comparative Study of Vision Transformers and CNNs for Few-Shot Rigid Transformation and Fundamental Matrix Estimation
- 用ViT和CNN在少样本下估计图像刚性变换和基础矩阵
- 小数据时CNN性能接近甚至超过ViT,大数据时ViT胜出
- ViT跨域泛化更强,适合对鲁棒性要求高的场景
视觉变换器(ViTs)和大规模卷积神经网络(CNNs)通过预训练特征表示重塑了计算机视觉,实现了在多种任务中的强大迁移学习。然而,它们作为几何估计任务的主干网络,在低数据条件下处理图像变形的效率仍不明确。本文研究两个典型任务:1)估计图像对间的二维刚性变换;2)预测立体图像对的基础矩阵,该问题在自动驾驶、机器人和三维场景重建中至关重要。系统比较了大型CNN(ResNet、EfficientNet、CLIP-ResNet)与基于ViT的基座模型(CLIP-ViT变体和DINO)在不同数据规模下的表现,包括少样本情形。这些模型原本用于分类或对比学习,倾向于关注高层语义。而所考虑的任务需要平衡局部与全局特征,挑战了直接将其作为主干的可行性。实验表明,在大数据场景下,如从头训练,ViTs优于CNN;但在小数据情况下,由于归纳偏置和较小容量,CNN表现更优,可媲美ViT。此外,ViTs在跨域评估中展现出更强泛化能力。结果强调了在微调阶段谨慎选择架构的重要性,推动未来研究向融合局部与全局表征的混合架构发展。
原文摘要 · Abstract (English)
Vision-transformers (ViTs) and large-scale convolution-neural-networks (CNNs) have reshaped computer vision through pretrained feature representations that enable strong transfer learning for diverse tasks. However, their efficiency as backbone architectures for geometric estimation tasks involving image deformations in low-data regimes remains an open question. This work considers two such tasks: 1) estimating 2D rigid transformations between pairs of images and 2) predicting the fundamental matrix for stereo image pairs, an important problem in various applications, such as autonomous mobility, robotics, and 3D scene reconstruction. Addressing this intriguing question, this work systematically compares large-scale CNNs (ResNet, EfficientNet, CLIP-ResNet) with ViT-based foundation models (CLIP-ViT variants and DINO) in various data size settings, including few-shot scenarios. These pretrained models are optimized for classification or contrastive learning, encouraging them to focus mostly on high-level semantics. The considered tasks require balancing local and global features differently, challenging the straightforward adoption of these models as the backbone. Empirical comparative analysis shows that, similar to training from scratch, ViTs outperform CNNs during refinement in large downstream-data scenarios. However, in small data scenarios, the inductive bias and smaller capacity of CNNs improve their performance, allowing them to match that of a ViT. Moreover, ViTs exhibit stronger generalization in cross-domain evaluation where the data distribution changes. These results emphasize the importance of carefully selecting model architectures for refinement, motivating future research towards hybrid architectures that balance local and global representations.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。