Summary
This paper builds upon the work of [1], which modifies CLIP's zero-shot classification by replacing class names with attributes generated by GPT. For instance, instead of using the class name "hen," [1] calculates the average cosine similarity between the image and a list of attributes like "two legs," "red, brown, or white feathers," "a small body," etc. This method proved more effective than using the class name alone. The current paper advances this concept in three main ways:
1. Attribute Generation: It enhances attribute generation by incorporating external knowledge from Wikipedia, unlike [1], which relies solely on GPT's internal knowledge. By including a Wikipedia URL in the prompt, the paper enables GPT to generate more accurate class descriptions.
2. Selection of Descriptions: The paper introduces scoring functions to identify the most informative class descriptions, termed "proponent texts." For instance, "red, brown, or white feathers" might be selected as the most defining feature of the class "hen" from the list of attributes.
3. Classification Methodology: Instead of direct zero-shot classification, the paper proposes training a linear classifier atop CLIP image embeddings, followed by further training using CLIP text embeddings derived from these texts. This approach is based on [2], which shows that image and text embeddings are equivalent in CLIP space, with text embeddings serving as "pseudo-images".
The authors demonstrate that on 7 out of 9 datasets, their method for obtaining text descriptions (Steps 1 and 2) outperforms [1] in zero-shot classification. Additionally, on all 9 datasets, Step 3 enhances supervised classification performance compared to models trained solely on images.
Reference:
[1] Visual Classification via Description from Large Language Models. Sachit Menon, Carl Vondrick. ICLR 2023. https://arxiv.org/abs/2210.07183
[2] Diagnosing and Rectifying Vision Models using Language. Yuhui Zhang, Jeff Z. HaoChen, Shih-Cheng Huang, Kuan-Chieh Wang, James Zou, Serena Yeung. ICLR 2023. https://arxiv.org/abs/2302.04269
Strengths
- Incorporating external knowledge in Step 1 and selecting the most informative class description using scoring functions in Step 2 are innovative approaches that could offer deeper insights into model predictions.
- Experimental results show improvements in model performance when the proposed method is used, thereby demonstrating its practical value.
Weaknesses
- The paper is challenging to follow, particularly the underlying motivation. I have to read many times to understand the motivation, methods, and results (see my summary). It requires significant revision for clarity before being published.
- The improvements over the [1] baseline are modest. In the most critical comparison of "Baseline (GPT-3.5)" versus "Ours Zero-shot" (Table 1), which compares the quality of class descriptions, the gains on 7 out of 9 datasets are relatively minor (typically 1-3%).
- The paper needs more in-depth analysis. Section 4.4 and Table 3 indicate a human preference for the class descriptions generated by this paper over [1], but the reasons for this preference remain unexplored. Given the marginal zero-shot improvement, a more thorough analysis of these descriptions could provide additional insights.
- The paper needs more quantitative analysis. Section 4.5 and Figure 6 qualitatively suggest that their class descriptions can reduce bias in text-to-image generation (e.g., for "doctors"), which is interesting but lacks numerical backing. Also, it's unclear why their class descriptions specifically mitigate gender bias in this context.
Questions
- Equation 1 and Figure 4: How does the influence score based solely on images help select the most informative texts? Figure 4 shows a uniform influence of 116.924 on all the texts when computing the "Influence Score", isn't that expected?
- Table 3: The "%" symbol should be removed, or each number should be multiplied by 100.
Rating
3: reject, not good enough
Confidence
5: You are absolutely certain about your assessment. You are very familiar with the related work and checked the math/other details carefully.