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

IOTA: Corrective Knowledge-Guided Prompt Learning via Black-White Box Framework

About

Recently, adapting pre-trained models to downstream tasks has attracted increasing interest. Previous Parameter-Efficient-Tuning (PET) methods regard the pre-trained model as an opaque Black Box model, relying purely on data-driven optimization and underutilizing their inherent prior knowledge. This oversight limits the models' potential for effective downstream task adaptation. To address these issues, we propose a novel black-whIte bOx prompT leArning framework (IOTA), which integrates a data-driven Black Box module with a knowledge-driven White Box module for downstream task adaptation. Specifically, the White Box module derives corrective knowledge by contrasting the wrong predictions with the right cognition. This knowledge is verbalized into interpretable human prompts and leveraged through a corrective knowledge-guided prompt selection strategy to guide the Black Box module toward more accurate predictions. By jointly leveraging knowledge- and data-driven learning signals, IOTA achieves effective downstream task adaptation. Experimental results on 12 image classification benchmarks under few-shot and easy-to-hard adaptation settings demonstrate the effectiveness of corrective knowledge and the superiority of our method over state-of-the-art methods.

Shaokun Wang, Yifan Yu, Yuhang He, Weili Guan, Yihong Gong• 2026

Related benchmarks

TaskDatasetResultRank
Image ClassificationFlowers102
Accuracy98.74
478
Image ClassificationDTD
Accuracy73.4
419
Image ClassificationUCF101
Top-1 Acc86.02
404
Image ClassificationCIFAR-100
Accuracy71.37
302
Image ClassificationAircraft
Accuracy52.84
302
Image ClassificationStanfordCars
Accuracy86.41
266
Image ClassificationRESISC45
Accuracy88.27
263
Image ClassificationStanford Dogs
Accuracy77.51
130
Image ClassificationEuroSAT
Accuracy85.49
83
Image Classification12 Downstream Datasets Fine-grained, Natural, Specialized (test)
Accuracy (Fine-grained)72.07
21
Showing 10 of 11 rows

Other info

Follow for update