arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2312.02546cs.CV

机器视觉疗法:多模态大语言模型可通过去噪上下文学习增强视觉鲁棒性

Machine Vision Therapy: Multimodal Large Language Models Can Enhance Visual Robustness via Denoising In-Context Learning

Sydney AI Centre, The University of Sydney(悉尼大学悉尼人工智能中心);Institute of Image Communication and Network Engineering, Shanghai JiaoTong University(上海交通大学图像通信与网络工程研究所)

Zhuo Huang, Chang Liu, Yinpeng Dong, Hang Su, Shibao Zheng, Tongliang Liu

AI总结:

本文提出机器视觉疗法及去噪上下文学习(DICL)策略,利用多模态大语言模型纠正视觉模型在OOD场景下的噪声预测,以无监督方式提升模型鲁棒性。

摘要:

尽管诸如对比语言-图像预训练(CLIP)等视觉模型展现出令人印象深刻的泛化性能,但在未进行微调的分布外(OOD)场景下,其零样本鲁棒性仍然受限。与其像通常那样不理想地提供人类监督,不如利用具备强大视觉理解能力的多模态大语言模型(MLLMs)。然而,由于任务不兼容,MLLMs在处理视觉问题时表现挣扎,从而阻碍了其应用。本文中,我们提出有效利用MLLMs实施机器视觉疗法,旨在纠正视觉模型产生的噪声预测。通过使用去噪后的标签进行微调,学习模型的性能可以以无监督方式得到提升。为解决不兼容问题,我们提出了一种新颖的去噪上下文学习(DICL)策略,将视觉任务与MLLMs对齐。具体而言,通过估计一个捕获某一类别被混淆为另一类别概率的转移矩阵,可以构建包含一个正确示例和来自最可能噪声类别的一个错误示例的指令。这种指令可以帮助任何具备ICL能力的MLLMs检测并纠正视觉模型的不正确预测。通过在ImageNet、WILDS、DomainBed及其他OOD数据集上的大量实验,我们仔细验证了本方法的定量和定性有效性。代码可在https://github.com/tmllab/Machine_Vision_Therapy获取。

英文摘要:

Although vision models such as Contrastive Language-Image Pre-Training (CLIP) show impressive generalization performance, their zero-shot robustness is still limited under Out-of-Distribution (OOD) scenarios without fine-tuning. Instead of undesirably providing human supervision as commonly done, it is possible to take advantage of Multi-modal Large Language Models (MLLMs) that hold powerful visual understanding abilities. However, MLLMs are shown to struggle with vision problems due to the incompatibility of tasks, thus hindering their utilization. In this paper, we propose to effectively leverage MLLMs to conduct Machine Vision Therapy which aims to rectify the noisy predictions from vision models. By fine-tuning with the denoised labels, the learning model performance can be boosted in an unsupervised manner. To solve the incompatibility issue, we propose a novel Denoising In-Context Learning (DICL) strategy to align vision tasks with MLLMs. Concretely, by estimating a transition matrix that captures the probability of one class being confused with another, an instruction containing a correct exemplar and an erroneous one from the most probable noisy class can be constructed. Such an instruction can help any MLLMs with ICL ability to detect and rectify incorrect predictions of vision models. Through extensive experiments on ImageNet, WILDS, DomainBed, and other OOD datasets, we carefully validate the quantitative and qualitative effectiveness of our method. Our code is available at https://github.com/tmllab/Machine_Vision_Therapy.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑