# Trajectory-Aware Flow Matching for Topology Optimisation

arXiv: https://arxiv.org/abs/2607.14652 · HTML: https://mlsanity.com/paper/2607.14652/

## Key facts

- Cited by 1 papers in the map (data through 14 Oct 2026).
- Placed in Flow matching, time series and science, one of 14 directions in Diffusion models.
- Builds on 19 of its 38 references that are in the map.
- First posted on arXiv 16 Jul 2026.

## Placed in

- Diffusion models / [Flow matching, time series and science](https://mlsanity.com/diffusion/flow-matching-time-series-and-science/)

## Builds on (most cited first, 19 of 19)

- [Denoising Diffusion Probabilistic Models](https://mlsanity.com/paper/2006.11239/) (2020)
- [Flow Matching for Generative Modeling](https://mlsanity.com/paper/2210.02747/) (2022)
- [Flow Straight and Fast: Learning to Generate and Transfer Data with Rectified Flow](https://mlsanity.com/paper/2209.03003/) (2022)
- [Stochastic Interpolants: A Unifying Framework for Flows and Diffusions](https://mlsanity.com/paper/2303.08797/) (2023)
- [Improving and generalizing flow-based generative models with minibatch optimal transport](https://mlsanity.com/paper/2302.00482/) (2023)
- [Flow Matching Guide and Code](https://mlsanity.com/paper/2412.06264/) (2024)
- [Diffusion Models Beat GANs on Topology Optimization](https://mlsanity.com/paper/2208.09591/) (2022)
- [TopologyGAN: Topology Optimization Using Generative Adversarial Networks Based on Physical Fields Over the Initial Domain](https://mlsanity.com/paper/2003.04685/) (2020)
- [Deep learning for determining a near-optimal topological design without any iteration](https://mlsanity.com/paper/1801.05463/) (2018)
- [Neural networks for topology optimization](https://mlsanity.com/paper/1709.09578/) (2017)
- [3D Topology Optimization using Convolutional Neural Networks](https://mlsanity.com/paper/1808.07440/) (2018)
- [Generative Design by Reinforcement Learning: Enhancing the Diversity of Topology Optimization Designs](https://mlsanity.com/paper/2008.07119/) (2020)
- [Real-Time Topology Optimization in 3D via Deep Transfer Learning](https://mlsanity.com/paper/2102.07657/) (2021)
- [Elucidating the Design Choice of Probability Paths in Flow Matching for Forecasting](https://mlsanity.com/paper/2410.03229/) (2024)
- [Diffusing the Optimal Topology: A Generative Optimization Approach](https://mlsanity.com/paper/2303.09760/) (2023)
- [On the Generalization in Topology Optimization via Sensitivity-Conditioned Bernoulli Flow Matching](https://mlsanity.com/paper/2606.02179/) (2026)
- [Real-time topology optimization via learnable mappings](https://mlsanity.com/paper/2311.08473/) (2023)
- [Physics-Informed Transformer for Real-Time High-Fidelity Topology Optimization](https://mlsanity.com/paper/2604.03522/) (2026)
- [Latent Space Diffusion for Topology Optimization](https://mlsanity.com/paper/2508.05624/) (2025)

## Built on by (most cited first, 1 of 1)

- [Conditional Flow Matching for ML-Based Inverse Design Problems](https://mlsanity.com/paper/2609.00863/) (2026)

## Abstract (arXiv)

Topology optimisation (TO) often requires repeated finite element analysis and sensitivity-based material updates, which can be costly when multiple candidate designs are needed under varying physical and design conditions. Generative TO offers a route to rapid design exploration, but existing models may rely on adversarial training, long reverse-diffusion sampling, or external guidance to maintain structural feasibility and physical consistency. This study develops a flow matching-based topology optimisation (FMTO) framework for conditional topology generation. Linear FMTO is first formulated as an endpoint-based baseline by interpolating between a Gaussian source field and the BESO reference topology. To introduce mechanically meaningful intermediate states, a trajectory-aware FMTO formulation is proposed, where volume-fraction-indexed BESO states are used to construct the probability path and target velocity field. This incorporates physics-guided optimisation history into generative flow learning without adding inference-time optimisation. A path--velocity mismatch analysis explains why moderate trajectory weighting can improve generation stability, whereas excessive guidance may over-constrain the learned transport. Numerical examples show that FMTO generates diverse topology candidates with improved compliance-related performance, volume-fraction satisfaction, topology fidelity, and substantially fewer sampling steps than a diffusion-based baseline. Under limited training data, trajectory-aware FMTO achieves the best overall performance with a moderate trajectory weight. Studies on trajectory-anchor density and three-dimensional topology generation further demonstrate the influence of path design and the applicability of the proposed framework beyond two-dimensional problems.
