arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2409.04828cs.CVcs.AIcs.MM
收藏

POINTS:以经济策略提升你的视觉语言模型

POINTS: Improving Your Vision-language Model with Affordable Strategies

Pattern Recognition Center, WeChat AI, Tencent Inc(腾讯微信AI模式识别中心);Shanghai Jiao Tong University(上海交通大学);Nanjing University(南京大学)

Yuan Liu, Zhongyin Zhao, Ziyuan Zhuang, Le Tian, Xiao Zhou, Jie Zhou

AI总结:

针对视觉语言模型在透明度、预训练数据探索和微调收益递减方面的问题,提出训练稳健基线、用困惑度筛选预训练数据及在指令微调中应用模型汤等经济策略,构建出性能媲美最先进模型的9B参数模型。

摘要:

近年来,视觉语言模型取得了显著进展,在光学字符识别和几何问题求解等任务中表现出色。然而,仍存在几个关键问题:1)专有模型往往缺乏对其架构的透明度,而开源模型需要对其训练策略进行更详细的消融研究。2)开源工作中的预训练数据探索不足,数据集是经验性地添加的,使得过程繁琐。3)微调通常侧重于添加数据集,导致收益递减。为解决这些问题,我们提出以下贡献:1)我们利用视觉语言模型的最新进展训练了一个稳健的基线模型,引入了有效的改进,并对每种技术进行了全面的消融和验证。2)受近期大型语言模型工作的启发,我们使用困惑度对预训练数据进行过滤,选择最低困惑度的数据进行训练。这种方法使我们能够在精选的100万数据集上进行训练,取得了具有竞争力的性能。3)在视觉指令微调期间,当添加更多数据集带来的改进微乎其微时,我们在不同数据集上使用了模型汤。这些创新产生了一个90亿参数的模型,其性能与最先进的模型相媲美。我们的策略高效且轻量,易于社区采用。

英文摘要:

In recent years, vision-language models have made significant strides, excelling in tasks like optical character recognition and geometric problem-solving. However, several critical issues remain: 1) Proprietary models often lack transparency about their architectures, while open-source models need more detailed ablations of their training strategies. 2) Pre-training data in open-source works is under-explored, with datasets added empirically, making the process cumbersome. 3) Fine-tuning often focuses on adding datasets, leading to diminishing returns. To address these issues, we propose the following contributions: 1) We trained a robust baseline model using the latest advancements in vision-language models, introducing effective improvements and conducting comprehensive ablation and validation for each technique. 2) Inspired by recent work on large language models, we filtered pre-training data using perplexity, selecting the lowest perplexity data for training. This approach allowed us to train on a curated 1M dataset, achieving competitive performance. 3) During visual instruction tuning, we used model soup on different datasets when adding more datasets yielded marginal improvements. These innovations resulted in a 9B parameter model that performs competitively with state-of-the-art models. Our strategies are efficient and lightweight, making them easily adoptable by the community.

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

↑