Unsupervised Transfer Clustering for Mitigating Cold Start in Active Prompt Learning
Organizations: Department of Computing, Federal University of S˜ao Carlos (UFSCar), Sorocaba, SP – Brazil
Abstract
Vision-Language Models (VLMs) are able to achieve impressive zero-shot classification performance by aligning visual and textual representations, but each new task still demands handcrafted prompts. Active Prompt Learning (APL) combines Active Learning (AL) and Prompt Learning (PL) into a single framework, allowing for the usage of the VLM prior knowledge for iteratively querying the most informative images to be labeled. However, the cold-start problem is still relevant for APL methods, where the performance of the initial query can be worse than random sampling. While recent state-of-the-art APL methods mitigate with balanced sampling and multimodal features, they rely on rigid, distance-based clustering to group these features. This simplistic approach can struggle to capture the complex, high-dimensional semantic distributions inherent to VLMs, leading to suboptimal query representativeness. Unsupervised transfer can be applied to these features as a possible alternative, since it is capable of inferring the underlying human labeling of a task without any form of supervision. This way, samples can be grouped in semantically coherent clusters. This paper proposes Unsupervised Transfer Clustering with Selective Querying (UTC+SQ), a framework that enhances a recent APL approach by leveraging state-of-the-art unsupervised transfer model. These models generate high-fidelity pseudo-labels that establish semantically meaningful clusters, allowing for the selection of more relevant samples. Experimental evaluations demonstrate that shifting from distance-based to projection-based clustering improves the representativeness of the queried subset, achieving accuracy gains in 6 of the 8 datasets tested.
Figures & tables
| Dataset | Classes | Train | Validation | Test | Resolution |
|---|---|---|---|---|---|
| CIFAR-10 | 10 | 45,000 | 5,000 | 10,000 | 32 32 |
| STL-10 | 10 | 4,500 | 500 | 8,000 | 96 96 |
| Caltech101 | 100 | 4,128 | 1,649 | 2,465 | 300 200 |
| Flowers102 | 102 | 4,093 | 1,633 | 2,463 | 500 px (smaller side) |
| OxfordPets | 37 | 2,944 | 736 | 3,669 | Multiple |
| Aircraft | 100 | 3,334 | 3,333 | 3,333 | 1–2 MP |
| Method | Flowers102 | OxfordPets | DTD | Caltech101 | Aircraft | EuroSAT | CIFAR-10 | STL-10 | Average |
|---|---|---|---|---|---|---|---|---|---|
| -Means [ 1 ] | 78.92 2.31 | 85.22 3.59 | 51.97 1.92 | 71.87 3.16 | 30.69 0.51 | 65.41 4.54 | 78.60 0.51 | 85.11 1.80 | 68.47 0.92 |
| HUME [ 13 ] | 67.85 0.98 | 82.59 3.20 | 51.48 0.82 | 55.85 0.79 | 20.43 0.58 | 66.33 3.34 | 96.65 0.17 | 94.71 8.01 | 66.99 1.11 |
| TURTLE [ 5 ] | 72.92 0.14 | 91.97 0.06 | 57.23 0.96 | 59.34 0.65 | 32.49 0.08 | 77.47 0.28 | 93.18 0.08 | 98.32 0.00 | 72.86 0.19 |
| Method / Strategy | Flowers102 | OxfordPets | DTD | Caltech101 | Aircraft | EuroSAT | CIFAR-10 | STL-10 | Average |
|---|---|---|---|---|---|---|---|---|---|
| Baseline | |||||||||
| CB+SQ [ 1 ] | 80.05 0.72 | 78.87 4.20 | 43.72 1.69 | 83.38 1.88 | 24.84 0.47 | 51.94 4.72 | 65.58 0.79 | 78.66 1.78 | 63.38 0.39 |
| UTC+SQ w/ HUME (ours) | |||||||||
| Centroids | 78.93 0.18 | 79.33 1.03 | 43.05 1.01 | 78.16 1.20 | 19.58 0.77 | 57.94 7.05 | 80.33 1.14 | 85.50 1.17 | 65.35 0.66 |
| Lowest Entropy | 69.85 1.18 | 70.58 3.72 | 39.36 5.90 | 71.74 0.49 | 17.07 2.30 | 50.91 1.93 | 71.06 4.28 | 84.93 6.32 | 59.44 1.09 |
| Highest Confidence | 73.54 2.77 | 76.47 2.10 | 42.55 0.53 | 75.20 1.00 | 20.37 0.52 | 53.44 5.35 | 74.13 3.80 | 84.86 7.12 | 62.57 1.37 |
Active Learning
Active learning selects the examples, labels, features, or experiments most useful for improving a model under limited budgets.
See all 285 papers