GIST: Generating Image-Specific Text for Fine-grained Object Classification
Lewis, Kathleen M., Mu, Emily, Dalca, Adrian V., Guttag, John
–arXiv.org Artificial Intelligence
Recent vision-language models outperform vision-only models on many image classification tasks. However, because of the absence of paired text/image descriptions, it remains difficult to fine-tune these models for fine-grained image classification. In this work, we propose a method, GIST, for generating image-specific fine-grained text descriptions from image-only datasets, and show that these text descriptions can be used to improve classification. Key parts of our method include 1. prompting a pretrained large language model with domain-specific prompts to generate diverse fine-grained text descriptions for each class and 2. using a pretrained vision-language model to match each image to label-preserving text descriptions that capture relevant visual features in the image. We demonstrate the utility of GIST by fine-tuning vision-language models on the image-and-generated-text pairs to learn an aligned vision-language representation space for improved classification. We evaluate our learned representation space in full-shot and few-shot scenarios across four diverse fine-grained classification datasets, each from a different domain. Our method achieves an average improvement of $4.1\%$ in accuracy over CLIP linear probes and an average of $1.1\%$ improvement in accuracy over the previous state-of-the-art image-text classification method on the full-shot datasets. Our method achieves similar improvements across few-shot regimes. Code is available at https://github.com/emu1729/GIST.
arXiv.org Artificial Intelligence
Aug-4-2023
- Country:
- Asia > Middle East
- Israel > Tel Aviv District > Tel Aviv (0.04)
- Europe > Netherlands
- North Holland > Amsterdam (0.04)
- North America > United States
- Massachusetts (0.04)
- Asia > Middle East
- Genre:
- Research Report > New Finding (0.68)
- Industry:
- Aerospace & Defense > Aircraft (0.67)
- Health & Medicine > Therapeutic Area
- Dermatology (1.00)
- Transportation > Air (0.93)
- Technology: