Learning to Name Classes for Vision and Language Models

التفاصيل البيبلوغرافية
العنوان: Learning to Name Classes for Vision and Language Models
المؤلفون: Parisot, Sarah, Yang, Yongxin, McDonagh, Steven
سنة النشر: 2023
المجموعة: Computer Science
مصطلحات موضوعية: Computer Science - Computer Vision and Pattern Recognition
الوصف: Large scale vision and language models can achieve impressive zero-shot recognition performance by mapping class specific text queries to image content. Two distinct challenges that remain however, are high sensitivity to the choice of handcrafted class names that define queries, and the difficulty of adaptation to new, smaller datasets. Towards addressing these problems, we propose to leverage available data to learn, for each class, an optimal word embedding as a function of the visual content. By learning new word embeddings on an otherwise frozen model, we are able to retain zero-shot capabilities for new classes, easily adapt models to new datasets, and adjust potentially erroneous, non-descriptive or ambiguous class names. We show that our solution can easily be integrated in image classification and object detection pipelines, yields significant performance gains in multiple scenarios and provides insights into model biases and labelling errors.
Comment: Accepted to CVPR 2023
نوع الوثيقة: Working Paper
URL الوصول: http://arxiv.org/abs/2304.01830
رقم الأكسشن: edsarx.2304.01830
قاعدة البيانات: arXiv