Learning to Search: A Decision-Based Agent for Knowledge-Based Visual Question Answering

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Chen, Zhuohong, Wu, Zhenxian, Yu, Yunyao, Xu, Hangrui, Liao, Zirui, Liu, Zhifang, Deng, Xiangwen, Jiao, Pen, Wang, Haoqian
Format: Preprint
Published: 2026
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866918436303536128
author Chen, Zhuohong
Wu, Zhenxian
Yu, Yunyao
Xu, Hangrui
Liao, Zirui
Liu, Zhifang
Deng, Xiangwen
Jiao, Pen
Wang, Haoqian
author_facet Chen, Zhuohong
Wu, Zhenxian
Yu, Yunyao
Xu, Hangrui
Liao, Zirui
Liu, Zhifang
Deng, Xiangwen
Jiao, Pen
Wang, Haoqian
contents Knowledge-based visual question answering (KB-VQA) requires vision-language models to understand images and use external knowledge, especially for rare entities and long-tail facts. Most existing retrieval-augmented generation (RAG) methods adopt a fixed pipeline that sequentially retrieves information, filters it, and then produces an answer. Such a design makes it difficult to adapt to diverse question types. Moreover, it separates retrieval from reasoning, making it hard for the model to decide when to search, how to refine queries, or when to stop. As a result, the retrieved evidence is often poorly aligned with the question. To address these limitations, we reformulate KB-VQA as a search-agent problem and model the solving process as a multi-step decision-making procedure. At each step, the agent selects one of four actions-Answer, Image Retrieval, Text Retrieval, and Caption-based on its current information state. We further design an automated pipeline to collect multi-step trajectories that record the agent's reasoning process, tool usage, and intermediate decisions. These trajectories are then used as supervision for fine-tuning. Experiments on InfoSeek and E-VQA demonstrate that our method achieves state-of-the-art performance, consistently outperforming prior baselines and confirming the effectiveness of our framework.
format Preprint
id arxiv_https___arxiv_org_abs_2604_07146
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Learning to Search: A Decision-Based Agent for Knowledge-Based Visual Question Answering
Chen, Zhuohong
Wu, Zhenxian
Yu, Yunyao
Xu, Hangrui
Liao, Zirui
Liu, Zhifang
Deng, Xiangwen
Jiao, Pen
Wang, Haoqian
Computer Vision and Pattern Recognition
Knowledge-based visual question answering (KB-VQA) requires vision-language models to understand images and use external knowledge, especially for rare entities and long-tail facts. Most existing retrieval-augmented generation (RAG) methods adopt a fixed pipeline that sequentially retrieves information, filters it, and then produces an answer. Such a design makes it difficult to adapt to diverse question types. Moreover, it separates retrieval from reasoning, making it hard for the model to decide when to search, how to refine queries, or when to stop. As a result, the retrieved evidence is often poorly aligned with the question. To address these limitations, we reformulate KB-VQA as a search-agent problem and model the solving process as a multi-step decision-making procedure. At each step, the agent selects one of four actions-Answer, Image Retrieval, Text Retrieval, and Caption-based on its current information state. We further design an automated pipeline to collect multi-step trajectories that record the agent's reasoning process, tool usage, and intermediate decisions. These trajectories are then used as supervision for fine-tuning. Experiments on InfoSeek and E-VQA demonstrate that our method achieves state-of-the-art performance, consistently outperforming prior baselines and confirming the effectiveness of our framework.
title Learning to Search: A Decision-Based Agent for Knowledge-Based Visual Question Answering
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2604.07146