Keywords: Large Language Model; Code Generation; Data Visualization
TL;DR: LLMs are bad at complex charts. We built a small, specialized model, PlotCraftor, that fixes this and is now state-of-the-art.
Abstract: Recent Large Language Models (LLMs) have demonstrated remarkable proficiency in code generation. However, their ability to create complex visualizations for scaled and structured data remains largely unevaluated and underdeveloped. To address this gap, we introduce \textbf{PlotCraft}, a new benchmark featuring 1k challenging visualization tasks that cover a wide range of topics, such as finance, scientific research, and sociology. The benchmark is structured around seven high-level visualization tasks and encompasses 48 distinct chart types. Crucially, it is the first to systematically evaluate both single-turn generation and multi-turn refinement across a diverse spectrum of task complexities.
Our comprehensive evaluation of 23 leading LLMs on PlotCraft reveals obvious performance deficiencies in handling sophisticated visualization tasks. To bridge this performance gap, we develope \textbf{SynthVis-30K}, a large-scale, high-quality dataset of complex visualization code synthesized via a collaborative agent framework. Building upon this dataset, we develope \textbf{PlotCraftor}, a novel code generation model that achieves strong capabilities in complex data visualization with a remarkably small size.
Across VisEval, PandasPlotBench, and our proposed PlotCraft, PlotCraftor shows performance comparable to that of leading proprietary approaches. Especially, on hard task, Our model achieves over 50\% performance improvement. We will release the benchmark, dataset, and code at https://anonymous.4open.science/r/PlotCraft-E320.
Primary Area: datasets and benchmarks
Submission Number: 95
Loading