TikZilla: Scaling Text-to-TikZ with High-Quality Data and Reinforcement Learning

arXiv:2603.03072 · cs.AI, cs.CL, cs.CV · Submitted 2026-03-03 · Read on arXiv

cs.AI, cs.CL, cs.CV

Submitted: 2026-03-03

Updated: 2026-09-01

Comments: 33 pages, 18 figures, and 13 tables. Models and datasets: https://huggingface.co/collections/nllg/tikzilla . Code: https://github.com/NL2G/TikZilla

Journal ref: Proceedings of the Fourteenth International Conference on Learning Representations (ICLR 2026), 2026

License: http://creativecommons.org/licenses/by-sa/4.0/

The gist: Large language models (LLMs) are increasingly used to assist scientists across diverse workflows.

Terminology

Abstract

Large language models (LLMs) are increasingly used to assist scientists across diverse workflows. A key challenge is generating high-quality figures from textual descriptions, often represented as TikZ programs that can be rendered as scientific images. Prior research has proposed a variety of datasets and modeling approaches for this task. However, existing datasets for Text-to-TikZ are too small and noisy to capture the complexity of TikZ, causing mismatches between text and rendered figures. Moreover, prior approaches rely solely on supervised fine-tuning (SFT), which does not expose the model to the rendered semantics of the figure, often resulting in errors such as looping, irrelevant content, and incorrect spatial relations. To address these issues, we construct DaTikZ-V4, a dataset more than four times larger and substantially higher in quality than DaTikZ-V3, enriched with LLM-generated figure descriptions. Using this dataset, we train TikZilla, a family of small open-source Qwen models (3B and 8B) with a two-stage pipeline of SFT followed by reinforcement learning (RL). For RL, we leverage an image encoder trained via inverse graphics to provide semantically faithful reward signals. Extensive human evaluations with over 1,000 judgments show that TikZilla improves by 1.5-2 points over its base models on a 5-point scale, surpasses GPT-4o by 0.5 points, and matches GPT-5 in the image-based evaluation, while operating at much smaller model sizes.

Sources

Related papers