小模型在分布外数据上表现差?这篇提出新结构提升鲁棒性。
CoCAViT: Compact Vision Transformer with Robust Global Coordination
- 设计动态全局令牌机制,增强局部与全局特征融合
- 2800万参数模型在ImageNet上达84.0%准确率,跨域性能显著提升
- 适合需要低延迟高鲁棒性的实时视觉应用
近年来,大规模视觉主干网络通过大量预训练展现出学习通用图像特征的强大能力。同时,许多高效架构在特定领域基准上表现可媲美大模型。然而我们发现,小模型在分布外(OOD)数据上的性能下降更为严重,表明现有高效模型泛化能力不足。为此,我们识别出关键的架构瓶颈与不当设计,旨在恢复小模型的鲁棒性。为重建纯窗口注意力的全局感受野,提出协调补丁交叉注意力(CoCA)机制,采用动态、领域感知的全局令牌,以最小计算开销自适应捕捉跨域稳健模式。结合这些改进,提出CoCAViT,一种专为实时鲁棒视觉表征设计的新视觉主干。大量实验证实其有效性:在224×224分辨率下,CoCAViT-28M在ImageNet-1K上达到84.0%的Top-1准确率,并在多个OOD基准上实现显著提升;同时在COCO目标检测上达52.2 mAP,ADE20K语义分割达51.3 mIOU,且保持低延迟。
原文摘要 · Abstract (English)
In recent years, large-scale visual backbones have demonstrated remarkable capabilities in learning general-purpose features from images via extensive pre-training. Concurrently, many efficient architectures have emerged that have performance comparable to that of larger models on in-domain benchmarks. However, we observe that for smaller models, the performance drop on out-of-distribution (OOD) data is disproportionately larger, indicating a deficiency in the generalization performance of existing efficient models. To address this, we identify key architectural bottlenecks and inappropriate design choices that contribute to this issue, retaining robustness for smaller models. To restore the global field of pure window attention, we further introduce a Coordinator-patch Cross Attention (CoCA) mechanism, featuring dynamic, domain-aware global tokens that enhance local-global feature modeling and adaptively capture robust patterns across domains with minimal computational overhead. Integrating these advancements, we present CoCAViT, a novel visual backbone designed for robust real-time visual representation. Extensive experiments empirically validate our design. At a resolution of 224*224, CoCAViT-28M achieves 84.0% top-1 accuracy on ImageNet-1K, with significant gains on multiple OOD benchmarks, compared to competing models. It also attains 52.2 mAP on COCO object detection and 51.3 mIOU on ADE20K semantic segmentation, while maintaining low latency.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。