Improving Generalization in Visual Reasoning via Self-Ensemble

TL;DR

This paper introduces a training-free self-ensemble method that enhances the generalization and reasoning abilities of large vision-language models without additional training, achieving state-of-the-art results on multiple visual reasoning benchmarks.

Contribution

The paper presents a novel self-ensemble technique that leverages internal model capabilities to improve visual reasoning without parameter updates.

Findings

Achieves SOTA performance on SketchyVQA

Improves out-of-distribution VQA accuracy

Enhances model generalization without additional training

Abstract

The cognitive faculty of visual reasoning necessitates the integration of multimodal perceptual processing and commonsense and external knowledge of the world. In recent years, a plethora of large vision-language models (LVLMs) have been proposed, demonstrating outstanding power and exceptional proficiency in commonsense reasoning across diverse domains and tasks. Nevertheless, training such LVLMs requires a lot of costly resources. Recent approaches, instead of training LVLMs from scratch on various large datasets, focus on exploring ways to take advantage of the capabilities of many different LVLMs, such as ensemble methods. In this work, we propose self-ensemble, a novel method that improves the generalization and visual reasoning of the model without updating any parameters, a training-free method. Our key insight is that we realized that LVLM itself can ensemble without the need for any other LVLMs, which helps to unlock their internal capabilities. Extensive experiments on various benchmarks demonstrate the effectiveness of our method in achieving state-of-the-art (SOTA) performance on SketchyVQA, Outside Knowledge VQA, and out-of-distribution VQA tasks.