Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Incorporating Geo-Diverse Knowledge into Prompting for Increased Geographical Robustness in Object Recognition

About

Existing object recognition models have been shown to lack robustness in diverse geographical scenarios due to domain shifts in design and context. Class representations need to be adapted to more accurately reflect an object concept under these shifts. In the absence of training data from target geographies, we hypothesize that geographically diverse descriptive knowledge of categories can enhance robustness. For this purpose, we explore the feasibility of probing a large language model for geography-based object knowledge, and we examine the effects of integrating knowledge into zero-shot and learnable soft prompting with CLIP. Within this exploration, we propose geography knowledge regularization to ensure that soft prompts trained on a source set of geographies generalize to an unseen target set. Accuracy gains over prompting baselines on DollarStreet while training only on Europe data are up to +2.8/1.2/1.6 on target data from Africa/Asia/Americas, and +4.6 overall on the hardest classes. Competitive performance is shown vs. few-shot target training, and analysis is provided to direct future study of geographical robustness.

Kyle Buettner, Sina Malakouti, Xiang Lorraine Li, Adriana Kovashka• 2024

Related benchmarks

TaskDatasetResultRank
Image ClassificationDollarStreet Europe
Top-1 Balanced Acc65.5
15
Image ClassificationDollarStreet Africa
Top-1 Balanced Acc50.8
15
Image ClassificationDollarStreet Total
Top-1 Balanced Acc57.4
15
object recognitionGeoNet USA
Top-1 Acc54.9
15
object recognitionGeoNet Asia
Top-1 Accuracy52.5
15
Image ClassificationDollarStreet Asia
Top-1 Balanced Acc56
15
Image ClassificationDollarStreet Americas
Top-1 Bal. Acc59.7
15
Image ClassificationDollarStreet Europe (Source)
Top-1 Balanced Acc73.6
12
Image ClassificationDollarStreet Africa (Target)
Top-1 Balanced Accuracy57.2
12
Image ClassificationDollarStreet Asia (Target)
Top-1 Balanced Acc63.8
12
Showing 10 of 14 rows

Other info

Code

Follow for update