用动态思维链提升视觉语言模型的多模态关键词预测能力
Boosting Multi-modal Keyphrase Prediction with Dynamic Chain-of-Thought in Vision-Language Models
- 用教师模型生成高质量思维链数据,微调小模型增强推理能力
- 动态注入思维链数据,避免模型过度思考,提升推理效率
- 在多个数据集上验证有效,适合需要强推理的多模态任务
多模态关键词预测(MMKP)旨在通过融合多种模态输入信息,超越仅依赖文本的方法,生成一组关键短语。传统多模态方法在处理缺失和未见场景时存在明显局限。此外,现有基准测试因训练与测试集存在显著重叠,高估了模型性能。本文提出利用视觉语言模型(VLMs)完成MMKP任务。首先采用零样本和监督微调(SFT)策略评估VLMs的下限性能。为提升VLMs的复杂推理能力,引入Fine-tune-CoT,利用教师模型生成的高质量思维链(CoT)数据对小型模型进行微调。最后,针对“过度思考”现象,提出动态CoT策略,在训练中自适应注入CoT数据,使模型在推理阶段可灵活启用推理能力。在多个数据集上的实验结果证明所提方法的有效性。代码已开源:https://github.com/bytedance/DynamicCoT。
原文摘要 · Abstract (English)
Multi-modal keyphrase prediction (MMKP) aims to advance beyond text-only methods by incorporating multiple modalities of input information to produce a set of conclusive phrases. Traditional multi-modal approaches have been proven to have significant limitations in handling the challenging absence and unseen scenarios. Additionally, we identify shortcomings in existing benchmarks that overestimate model capability due to significant overlap in training tests. In this work, we propose leveraging vision-language models (VLMs) for the MMKP task. Firstly, we use two widely-used strategies, e.g., zero-shot and supervised fine-tuning (SFT) to assess the lower bound performance of VLMs. Next, to improve the complex reasoning capabilities of VLMs, we adopt Fine-tune-CoT, which leverages high-quality CoT reasoning data generated by a teacher model to finetune smaller models. Finally, to address the "overthinking" phenomenon, we propose a dynamic CoT strategy which adaptively injects CoT data during training, allowing the model to flexibly leverage its reasoning capabilities during the inference stage. We evaluate the proposed strategies on various datasets and the experimental results demonstrate the effectiveness of the proposed approaches. The code is available at https://github.com/bytedance/DynamicCoT.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。