Skip to content

Omniview-Tuning: Boosting Viewpoint Invariance of Vision-Language Pre-training Models

Conference: ECCV 2024
arXiv: 2404.12139
Code: None
Area: Multimodal VLM
Keywords: Viewpoint Invariance, CLIP, Parameter-Efficient Fine-Tuning, Multi-View Dataset, Contrastive Learning

TL;DR

OVT via parameter-efficient fine-tuning significantly improves the robustness of VLP models (e.g., CLIP) to 3D viewpoint changes (averaging +9-10%) by constructing a 4.6-million multi-view image-text dataset, MVCap, and designing a minimax-optimized cross-viewpoint alignment framework, while incurring almost no loss in original performance.

Background & Motivation

Vision-language pre-training (VLP) models, such as CLIP, perform remarkably well under 2D distribution shifts (e.g., style changes, image corruptions). However, recent studies demonstrate that their performance drops significantly under 3D viewpoint changes. For instance, CLIP's accuracy on viewpoint-OOD benchmarks like ImageNet-V+ is much lower than on 2D-OOD benchmarks, posing a challenge for real-world applications requiring multi-view inputs, such as autonomous driving and robotics.

Root Cause: The insufficient coverage of diverse viewpoints in VLP training data prevents models from learning viewpoint-invariant representations.

Limitations of Prior Work:

Data Scarcity: Existing multi-view datasets (e.g., CO3D, MVImgNet) either have insufficient sample sizes, lack category coverage, or lack textual descriptions, making them unsuitable for VLP training.

Method Incompatibility: Traditional adversarial training (e.g., VIAT) treats viewpoint changes as adversarial attacks, introducing a trade-off between robustness and accuracy, and requiring NeRF rendering for adversarial viewpoints, which is computationally expensive (ResNet-50 + 1K objects ≈ 400 GPU hours).

Goal: To enhance viewpoint invariance without compromising the original performance of VLP models, while keeping training highly efficient.

Method

Overall Architecture

OVT introduces two main contributions: (1) constructing a large-scale multi-view image-text pair dataset, MVCap; and (2) designing a parameter-efficient Omniview-Tuning fine-tuning framework that trains VLP models using a cross-viewpoint alignment objective and a minimax optimization strategy.

Key Designs

  1. MVCap Dataset Construction:

    • Function: Creating the first million-scale multi-view image-text dataset, containing more than 4.6 million image-text pairs covering over 100,000 objects and 1,600 categories.
    • Mechanism:
      • Multi-View Image Acquisition: Integrating three data sources: Objaverse (3D synthetic), IM3D (synthetic), and MVImgNet (real-world video). For Objaverse, 24,495 semantically clear 3D objects are filtered using OpenShape semantic embeddings, and 100 random viewpoint images are rendered from the upper hemisphere using Blender. From MVImgNet, real-world scenes with over 30 valid viewpoints are selected.
      • Category-Guided Caption Generation: Automatically generating a description for each image using InstructBLIP-flant5xl. The key innovation is designing a category-guided prompt—"Write a short description for the image, noting that the main instance of the image is a \<category>"—which addresses the issue of class-inconsistent hallucinations of VLLMs under different viewpoints.
    • Design Motivation: Resolving the "chicken-and-egg" dilemma—a viewpoint-invariant model is required to generate training data, yet the model itself lacks viewpoint invariance. Ground-truth category information is injected to bypass this bottleneck.
  2. Cross-Viewpoint Alignment Objective:

    • Function: Adding a viewpoint consistency loss \(\mathcal{L}_{VC}\) on top of the standard image-text contrastive learning loss \(\mathcal{L}_{ITC}\) to directly align the visual embeddings of different viewpoints of the same object.
    • Mechanism: The optimization objective is formulated as \(\min_{\mathbf{W_v}, \mathbf{W_t}} [\mathcal{L}_{ITC} + \lambda \cdot \mathcal{L}_{VC}]\), where \(\mathcal{L}_{VC}\) minimizes the cosine distance between different viewpoint embeddings.
    • Design Motivation: Image-text alignment alone is insufficient to align visual representations across different viewpoints—although variations in textual descriptions across viewpoints are subtle, they can be amplified in high-dimensional space.
  3. Minimax Optimization Strategy:

    • Function: Reformulating the optimization of \(\mathcal{L}_{VC}\) as a minimax game to prevent concept drift and overfitting caused by over-alignment.
    • Mechanism:
      • Maximization Step: Computing an anchor viewpoint embedding \(z_{C_i}^I\) for each object (a weighted centroid, where weights are determined by nearest neighbor distances), and identifying the top-K outlier viewpoints with the maximum cosine distance from the anchor.
      • Minimization Step: Optimizing only these outlier viewpoint embeddings to converge toward the anchor using a margin-based loss \(l(z_{ij}^I, z_{C_i}^I) = \max[d(z_{ij}^I, z_{C_i}^I) + m, 0]\).
    • Design Motivation: Focusing only on the most extreme outlier viewpoints avoids aligning all viewpoint combinations (which is computationally intensive and prone to overfitting), while preserving the original embedding distribution to the greatest extent.
  4. Parameter-Efficient Modules (VIFormer + LoRA):

    • Function: Freezing the original weights of the text and vision encoders, only training additional light-weight parameters.
    • Mechanism:
      • LoRA: Applying low-rank adaptation to the self-attention layers of the vision encoder: \(\tilde{\mathbf{W_v}} = \mathbf{W_v} + \mathbf{BA}\), with rank=8.
      • VIFormer: Adding a learnable self-attention layer after the vision encoder to extract key viewpoint-invariant components \(s^I\).
      • Final Output: \(\tilde{z}^I = \alpha \cdot f_\theta(z^I) + (1-\alpha) \cdot z^I\), where \(\alpha=0.1\) is the residual ratio.
    • Design Motivation: The PEFT approach preserves original performance to the greatest extent while acquiring viewpoint invariance with minimal parameters (e.g., only 6.6M trainable parameters for ViT-B/32).

Loss & Training

  • Total Loss: \(\mathcal{L} = \mathcal{L}_{ITC} + \lambda \cdot \mathcal{L}_{VC}\), with \(\lambda=1.0\).
  • Training Data: A mixture of MVCap and the ImageNet-1K training set.
  • Prior to each epoch, anchor viewpoint embeddings and outlier viewpoint sets are calculated (maximization), followed by computing both losses and updating parameters within the batch (minimization).
  • Training budget: Approximately 35K iterations with a batch size of 512 for ViT-B/32.

Key Experimental Results

Main Results: Zero-Shot Classification (Different Architectures & Distributions)

Model Clean Avg. 2D-OOD Avg. Viewpoint-OOD Avg. Overall Avg.
OpenCLIP ViT-B/32 74.5 52.2 42.8 54.6
OVT-OpenCLIP ViT-B/32 71.2 49.7 52.4 (+9.6) 56.0
OpenCLIP ViT-B/16 76.8 53.6 46.4 57.0
OVT-OpenCLIP ViT-B/16 74.9 52.9 56.6 (+10.2) 59.6
OpenCLIP ViT-L/14 81.9 56.8 53.4 61.9
OVT-OpenCLIP ViT-L/14 81.8 57.3 62.3 (+8.9) 65.1

Bridging the Performance Gap: ImageNet-1K vs ImageNet-V+

Model IN-1K IN-V+ Gap
OpenCLIP ViT-B/32 66.5 37.1 29.4
OVT-OpenCLIP ViT-B/32 67.8 59.5 8.3
OpenCLIP ViT-L/14 75.2 53.2 22.0
OVT-OpenCLIP ViT-L/14 77.3 69.8 7.5

Key Findings

  • The improvement of OVT on Viewpoint-OOD is highly significant: on ViT-B/32, the accuracy on ImageNet-V+ surges from 37.1% to 59.5% (+22.4%), while that on ImageNet-1K increases by only 1.3%.
  • Performance loss on 2D-OOD is negligible, with only a 0.2% drop for ViT-L/14.
  • OVT is effective across different VLP architectures (CLIP, MetaCLIP, BLIP) and different vision encoders (ViT-B/32, B/16, L/14).
  • When acting as the vision encoder for a VLLM (LLaVA), OVT-CLIP also enhances viewpoint robustness in image captioning and visual question answering tasks.
  • The BLIP architecture similarly gains a substantial improvement (Viewpoint-OOD Avg. +8.6%).

Highlights & Insights

  • MVCap Dataset: It is the first million-scale multi-view image-text pair dataset specifically designed for VLP viewpoint invariance, featuring a simple yet effective category-guided prompting strategy.
  • Minimax Optimization: The ingenuity lies in optimizing only extreme outlier viewpoints instead of all viewpoint pairs. This saves computation and avoids overfitting, which represents an elegant adaptation of adversarial training ideas to the VLP context.
  • Outstanding Parameter Efficiency: By training only about 4-12M parameters (comprising 3-4% of total parameters), the training costs are substantially lower than traditional schemes like VIAT.
  • The experimental design is comprehensive, covering Clean, 2D-OOD, and Viewpoint-OOD distributions, making the findings highly convincing.

Limitations & Future Work

  • Synthetic data accounts for a large portion of the MVCap dataset, leaving a domain gap with real-world scenarios.
  • Category-Guided captions depend on ground-truth category labels, limiting the applicability to objects of unseen categories.
  • Currently, LoRA is only applied to the vision encoder while the text encoder is completely frozen, which might limit viewpoint adaptation on the textual side.
  • The selection of top-K outlier viewpoints is fixed prior to each epoch, which may lack dynamicity.
  • VIAT (2023) first applied NeRF adversarial training for viewpoint robustness, but at an extremely high computational cost; OVT drastically reduces this cost using PEFT.
  • The residual design of CLIP-Adapter inspired the VIFormer module.
  • The success of LoRA in NLP has been transferred to the viewpoint adaptation scenario of vision encoders.

Rating

  • Novelty: ⭐⭐⭐⭐ The first study to systematically address VLP viewpoint invariance, leveraging a dual-pronged approach of both data and methodology.
  • Experimental Thoroughness: ⭐⭐⭐⭐⭐ Evaluating 8 VLP variants on over 12 benchmark datasets, multiple downstream tasks, and providing detailed ablation studies.
  • Writing Quality: ⭐⭐⭐⭐ The problem definition is clear, and the narrative logic from data to method flows smoothly.
  • Value: ⭐⭐⭐⭐ It reveals the viewpoint vulnerability of VLPs and offers a reproducible solution, while the MVCap dataset possesses independent value.