Improved Visual Fine-tuning With Natural Language Supervision

Wang Junyang, Xu Yuanhong, Hu Juhua, Yan Ming, Sang Jitao, Qian Qi. Arxiv 2023

[Paper] [Code]    
Attention Mechanism BERT Ethics And Bias Fine Tuning Has Code Model Architecture Pretraining Methods RAG Reinforcement Learning Training Techniques

Fine-tuning a visual pre-trained model can leverage the semantic information from large-scale pre-training data and mitigate the over-fitting problem on downstream vision tasks with limited training examples. While the problem of catastrophic forgetting in pre-trained backbone has been extensively studied for fine-tuning, its potential bias from the corresponding pre-training task and data, attracts less attention. In this work, we investigate this problem by demonstrating that the obtained classifier after fine-tuning will be close to that induced by the pre-trained model. To reduce the bias in the classifier effectively, we introduce a reference distribution obtained from a fixed text classifier, which can help regularize the learned vision classifier. The proposed method, Text Supervised fine-tuning (TeS), is evaluated with diverse pre-trained vision models including ResNet and ViT, and text encoders including BERT and CLIP, on 11 downstream tasks. The consistent improvement with a clear margin over distinct scenarios confirms the effectiveness of our proposal. Code is available at \url{https://github.com/idstcv/TeS}.

Similar Work