arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2410.04064cs.LGcs.AI
收藏

Text2Chart31:利用自动反馈进行图表生成的指令微调

Text2Chart31: Instruction Tuning for Chart Generation with Automatic Feedback

Seoul National University(首尔大学);KAIST AI(韩国科学技术院人工智能实验室);NAVER AI Lab(NAVER人工智能实验室)

Fatemeh Pesaran Zadeh, Juyeon Kim, Jin-Hwa Kim, Gunhee Kim

AI总结:

本文构建覆盖31种图表、含11.1K多模态元组的Text2Chart31数据集,并提出无需人工反馈的强化学习指令微调方法,显著提升小模型的数据可视化生成性能。

摘要:

大语言模型(LLM)已在各类语言任务中展现出强大能力,尤其是通过指令微调方法。然而,LLM在通过图表和图形可视化复杂的真实世界数据方面面临挑战。首先,现有数据集很少覆盖全部图表类型,例如3D、体素和网格图表。其次,监督微调方法没有充分利用丰富数据集中的复杂关系,包括文本、代码和图形之间的关系。为应对这些挑战,我们提出了一种分层流水线和一个用于图表生成的新数据集。我们的数据集Text2Chart31参照Matplotlib库,包含31种独特的图表类型,拥有11.1K个由描述、代码、数据表和图表组成的元组。此外,我们针对图表生成任务引入了一种基于强化学习的指令微调技术,无需人工反馈。实验表明,该方法显著提升了模型性能,使较小模型能够超越较大的开源模型,并在数据可视化任务上与最先进的专有模型相媲美。我们在https://github.com/fatemehpesaran310/Text2Chart31公开了代码和数据集。

英文摘要:

Large language models (LLMs) have demonstrated strong capabilities across various language tasks, notably through instruction-tuning methods. However, LLMs face challenges in visualizing complex, real-world data through charts and plots. Firstly, existing datasets rarely cover a full range of chart types, such as 3D, volumetric, and gridded charts. Secondly, supervised fine-tuning methods do not fully leverage the intricate relationships within rich datasets, including text, code, and figures. To address these challenges, we propose a hierarchical pipeline and a new dataset for chart generation. Our dataset, Text2Chart31, includes 31 unique plot types referring to the Matplotlib library, with 11.1K tuples of descriptions, code, data tables, and plots. Moreover, we introduce a reinforcement learning-based instruction tuning technique for chart generation tasks without requiring human feedback. Our experiments show that this approach significantly enhances the model performance, enabling smaller models to outperform larger open-source models and be comparable to state-of-the-art proprietary models in data visualization tasks. We make the code and dataset available at https://github.com/fatemehpesaran310/Text2Chart31.

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

↑