agentic-visualization: PaperVizAgent: The Multi-Agent Studio That Tries to Make AI Draw Like a Researcher

Google Research breaks scientific visualization into retrieval, planning, styling, generation, and critique, then loops back until the chart earns publication-grade confidence.

8 min read • View on GitHub • More from google-research

A wide editorial scene of a white workbench turned into an assembly line. Raw notes and data enter from the left, pass through several specialist stations, and emerge on the right as a polished scientific chart under an inspection lamp. It explains that the project treats visualization as a staged editorial process, not a single prompt.
PaperVizAgent treats chart-making like a production line with checks at every station.
Key Takeaways

The chart that knows when it still looks wrong

Most chart generators stop when the code runs. PaperVizAgent keeps going. It renders a draft, looks at the result, and sends the work back if the output still has problems like overlap, clipping, or weak readability.

That is the real story here. The repo does not treat visualization as a prompt problem. It treats it as an editorial problem, where a visual has to survive critique before it counts as finished.

Acting like a creative team of specialized agents, it transforms raw scientific content into publication-quality diagrams and plots through an orchestrated pipeline of Retriever, Planner, Stylist, Visualizer, and Critic agents.

Dawei Zhu, Rui Meng, Yale Song, Xiyu Wei, Sujian Li, Tomas Pfister and Jinsung yoon, Authors · google-research/papervizagent

From one prompt to an assembly line

The repository’s architecture is built around specialization. Instead of asking one model to do everything, it breaks the job into a chain of agents with different responsibilities: retrieval, planning, styling, generation, and critique.

That division of labor matters because scientific visuals fail in different ways. A model can understand the request, yet still choose the wrong example, map the data badly, or produce a plot that looks technically valid but visually sloppy.

The system is not linear. It is a workflow with a quality gate, and the gate can send the output back for another pass.

The global TODO list is the real backbone

The shared memory move is easy to miss, but it is the part that keeps the whole system coherent. The query analyzer writes a global_todo_list, and every downstream agent keeps checking its work against that contract.

That reduces drift. Without a stable target, multi-agent systems can wander. With a shared checklist, each specialist can optimize its own step while still staying accountable to the same intent.

# simplified shape of the workflow
query = user_request()
todo_list = query_analyzer(query)
references = retriever(todo_list)
plan = planner(todo_list, references)
styled_spec = stylist(plan, references)
chart_code = visualizer(styled_spec)
image = render(chart_code)
score, feedback = critic(image, todo_list)
if score < threshold:
    chart_code = refine(chart_code, feedback, todo_list)

Why retrieval beats hallucinated chart logic

The retrieval step is the anti-hallucination move. Instead of trusting the model to invent a good plotting pattern from scratch, the system pulls in grounded examples and uses them as a bias toward valid, familiar structure.

That is especially useful in visualization work, where tiny mistakes matter. A plot can be semantically correct and still be unusable if the axis formatting, marker choice, or layout logic is off.

ApproachStrengthWeaknessBest use caseWhy this repo is different
One-shot LLM chart generationFast to tryOften brittle and style-blindQuick draftsPaperVizAgent adds retrieval and critique before it calls the job done
General-purpose multi-agent builderFlexible across tasksNot tuned for publication visualsBuilding custom agent flowsPaperVizAgent is specialized for scientific illustration quality
Traditional script-first workflowPrecise and reproducibleRequires manual iterationAnalyst-led chartingPaperVizAgent automates the editorial passes around the script
A close-up of a scientific chart on a drafting table with a clipped axis label, an overlapping legend, and a red pencil marking the problem areas. A critic's checklist points to the chart while a curved arrow sends it back toward a refinement box. It explains why visual evaluation matters beyond code correctness.
The visual evaluator catches failures that syntax checks never see.

How the visual evaluator closes the loop

This is where the system stops behaving like a chart bot and starts behaving like a reviewer. The evaluator is not asking whether the code executed. It is asking whether the artifact communicates clearly.

That catches the failures people remember from real work. A label that barely fits. A legend that fights the data. A composition that technically answers the prompt but would never make it into a paper or slide deck.

LayerWhat it optimizesWhat it missesWhat PaperVizAgent adds
Code executionCorrect syntax and runtimeReadability and aestheticsA visual critic that inspects the rendered result
Single-agent generationSpeedContext drift and fragile style choicesSpecialists with separate responsibilities
Generic agent orchestrationFlexibilityWeak quality gatesA publishability threshold and refinement loop

What makes it more than a chart bot

The strongest way to understand PaperVizAgent is as a publication workflow encoded in software. It combines inference, retrieval, design judgment, and revision into a single system that keeps asking whether the output is actually worthy.

That is a narrower ambition than a general agent platform, but it is also a sharper one. The repo is not trying to make every kind of visual artifact. It is trying to make scientific visuals that can survive editorial scrutiny.

Where it sits in the landscape

Compared with general agent builders, PaperVizAgent is less flexible but more opinionated. Compared with plain code-generation tools, it is slower but harder to fool. Compared with traditional visualization scripts, it removes a chunk of the manual back-and-forth that turns drafts into publishable figures.

ProjectPrimary focusIteration styleDifferentiator
PaperVizAgentPublication-quality scientific visualsClosed-loop critique and refinementSpecialized agents plus visual evaluation
General agent buildersBroad workflow compositionUser-designedPlatform breadth
One-shot chart generatorsFast output from textSingle passSpeed
Script-first plottingManual control and precisionHuman-ledDeterminism