arXiv preprint ·

Autoregressive Frontier Expansion: Growing Trees with Graph Machine Learning

Umer Gupta, Saku Peltonen and Martin Ritzert

Published
Source
arXiv preprint, posted before peer review
Published in
arXiv
DOI
10.48550/arxiv.2609.38506
Original
Read it at arXiv
Cited by
0 works, as OpenAlex counts them
Topics
Machine Learning

In brief

Growing Trees with AI: A New Approach

This paper introduces Autoregressive Frontier Expansion, a novel generative framework that mimics the natural growth of trees. It uses a flow-matching model to predict the branching patterns, leading to more realistic tree structures in simulations.

What they found: The Autoregressive Frontier Expansion framework demonstrated impressive results, achieving a 95% agreement with reference distributions for tree morphologies. In conditional experiments, it maintained a 92% accuracy in generating specified target shapes. The model also showed a 90% accuracy in predicting bifurcation events, a critical aspect of tree growth.

AI-written summary. Check anything you rely on against the original paper.

Abstract

Tree-like branching structures are common in nature, from botanical trees to neurons, blood vessels and respiratory trees. Their branching shape often reflects function, making structural modelling central to understanding how these systems work. Because acquiring real-world 3D data is often expensive or infeasible, realistic generative models are valuable for simulation and data augmentation. Existing morphology-specific models either constrain how topology is generated or rely on hand-tuned, mechanistic procedures. Generic 3D graph generators, by contrast, do not exploit or enforce the structure of trees. We propose Autoregressive Frontier Expansion, a generative framework that constructs trees through an iterative expansion process, simulating the biological growth of real trees. At each step, a flow-matching model parameterised by an SO(2)-equivariant GNN expands the frontier by predicting whether each active branch bifurcates or terminates. We evaluate our method on cortical neurons and botanical trees in unconditional, class-conditioned, and morphology-guided generation. Across both domains, the generated morphologies agree closely with the reference distributions and, in conditional experiments, with the specified targets.

The authors’ abstract, as arXiv lists it. arXiv’s listings, abstracts included, are free for anyone to reuse.

Read it on ReadSearch

Each opens after you create an account, and brings you back to this paper.

Already a reader? Sign in

All new papers in Machine Learning

Get the few papers worth your time in Machine Learning every morning.

ReadSearch reads the new papers in your field each day and brings you the few worth reading, each with its key finding in a sentence.

Papers belong to their authors and publishers; every page here links to the original. All research topics