Towards Global Optimal Visual In-Context Learning Prompt Selection

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Xu, Chengming, Liu, Chen, Wang, Yikai, Yao, Yuan, Fu, Yanwei
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866909343870353408
author Xu, Chengming
Liu, Chen
Wang, Yikai
Yao, Yuan
Fu, Yanwei
author_facet Xu, Chengming
Liu, Chen
Wang, Yikai
Yao, Yuan
Fu, Yanwei
contents Visual In-Context Learning (VICL) is a prevailing way to transfer visual foundation models to new tasks by leveraging contextual information contained in in-context examples to enhance learning and prediction of query sample. The fundamental problem in VICL is how to select the best prompt to activate its power as much as possible, which is equivalent to the ranking problem to test the in-context behavior of each candidate in the alternative set and select the best one. To utilize more appropriate ranking metric and leverage more comprehensive information among the alternative set, we propose a novel in-context example selection framework to approximately identify the global optimal prompt, i.e. choosing the best performing in-context examples from all alternatives for each query sample. Our method, dubbed Partial2Global, adopts a transformer-based list-wise ranker to provide a more comprehensive comparison within several alternatives, and a consistency-aware ranking aggregator to generate globally consistent ranking. The effectiveness of Partial2Global is validated through experiments on foreground segmentation, single object detection and image colorization, demonstrating that Partial2Global selects consistently better in-context examples compared with other methods, and thus establish the new state-of-the-arts.
format Preprint
id arxiv_https___arxiv_org_abs_2405_15279
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Towards Global Optimal Visual In-Context Learning Prompt Selection
Xu, Chengming
Liu, Chen
Wang, Yikai
Yao, Yuan
Fu, Yanwei
Computer Vision and Pattern Recognition
Visual In-Context Learning (VICL) is a prevailing way to transfer visual foundation models to new tasks by leveraging contextual information contained in in-context examples to enhance learning and prediction of query sample. The fundamental problem in VICL is how to select the best prompt to activate its power as much as possible, which is equivalent to the ranking problem to test the in-context behavior of each candidate in the alternative set and select the best one. To utilize more appropriate ranking metric and leverage more comprehensive information among the alternative set, we propose a novel in-context example selection framework to approximately identify the global optimal prompt, i.e. choosing the best performing in-context examples from all alternatives for each query sample. Our method, dubbed Partial2Global, adopts a transformer-based list-wise ranker to provide a more comprehensive comparison within several alternatives, and a consistency-aware ranking aggregator to generate globally consistent ranking. The effectiveness of Partial2Global is validated through experiments on foreground segmentation, single object detection and image colorization, demonstrating that Partial2Global selects consistently better in-context examples compared with other methods, and thus establish the new state-of-the-arts.
title Towards Global Optimal Visual In-Context Learning Prompt Selection
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2405.15279