用小模型处理高分辨率图,省算力还提效果
Parameter-Inverted Image Pyramid Networks for Visual Perception and Multimodal Understanding
- 高分辨率图像用小模型处理,低分辨率用大模型,反向分配参数
- 在COCO上达60.0盒AP,ADE20K上59.7 mIoU,算力仅为原方法40%-60%
- 适合视觉与多模态任务,尤其对大模型推理优化显著
图像金字塔广泛用于获取多尺度特征以实现精准视觉感知与理解。但现有方法对多分辨率图像均使用相同的大模型,导致计算开销巨大。为此,我们提出参数倒置图像金字塔网络(PIIP),将预训练模型(ViTs或CNNs)作为分支处理多尺度图像,高分辨率图像由更小的网络分支处理,以平衡计算成本与性能。为融合不同空间尺度信息,我们设计了一种新颖的跨分支特征交互机制。在目标检测、分割、图像分类及多模态理解等任务上,对多种感知模型和代表性多模态大模型LLaVA进行验证。结果表明,相比单分支及现有方法,PIIP在更低计算成本下表现更优。应用于InternViT-6B时,检测与分割性能提升1%-2%,计算量仅需原方法的40%-60%,最终在MS COCO上达到60.0盒AP,ADE20K上59.7 mIoU。在多模态理解方面,PIIP-LLaVA在TextVQA上达73.0%准确率,MMBench上74.5%,仅需2.8M训练数据。
原文摘要 · Abstract (English)
Image pyramids are widely adopted in top-performing methods to obtain multi-scale features for precise visual perception and understanding. However, current image pyramids use the same large-scale model to process multiple resolutions of images, leading to significant computational cost. To address this challenge, we propose a novel network architecture, called Parameter-Inverted Image Pyramid Networks (PIIP). Specifically, PIIP uses pretrained models (ViTs or CNNs) as branches to process multi-scale images, where images of higher resolutions are processed by smaller network branches to balance computational cost and performance. To integrate information from different spatial scales, we further propose a novel cross-branch feature interaction mechanism. To validate PIIP, we apply it to various perception models and a representative multimodal large language model called LLaVA, and conduct extensive experiments on various tasks such as object detection, segmentation, image classification and multimodal understanding. PIIP achieves superior performance compared to single-branch and existing multi-resolution approaches with lower computational cost. When applied to InternViT-6B, a large-scale vision foundation model, PIIP can improve its performance by 1%-2% on detection and segmentation with only 40%-60% of the original computation, finally achieving 60.0 box AP on MS COCO and 59.7 mIoU on ADE20K. For multimodal understanding, our PIIP-LLaVA achieves 73.0% accuracy on TextVQA and 74.5% on MMBench with only 2.8M training data. Our code is released at https://github.com/OpenGVLab/PIIP.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。