Saved in:
Bibliographic Details
Main Authors: de Campos, Miguel, Krebs, Werner, Gottschalk, Hanno
Format: Preprint
Published: 2026
Subjects:
Online Access:https://arxiv.org/abs/2603.15925
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866918391936188416
author de Campos, Miguel
Krebs, Werner
Gottschalk, Hanno
author_facet de Campos, Miguel
Krebs, Werner
Gottschalk, Hanno
contents Inverse design aims to find design parameters $x$ achieving target performance $y^*$. Generative approaches learn bidirectional mappings between designs and labels, enabling diverse solution sampling. However, standard conditional flow matching (CFM), when adapted to inverse problems by pairing labels with design parameters, exhibits strong sensitivity to their arbitrary ordering and scaling, leading to unstable training. We introduce Diagonal Flow Matching (Diag-CFM), which resolves this through a zero-anchoring strategy that pairs design coordinates with noise and labels with zero, making the learning problem provably invariant to coordinate permutations. This yields order-of-magnitude improvements in round-trip accuracy over CFM and invertible neural network baselines across design dimensions up to $P{=}100$. We develop two architecture-intrinsic uncertainty metrics, Zero-Deviation and Self-Consistency, that enable three practical capabilities: selecting the best candidate among multiple generations, abstaining from unreliable predictions, and detecting out-of-distribution targets; consistently outperforming ensemble and general-purpose alternatives across all tasks. We validate on airfoil, gas turbine combustor, and an analytical benchmark with scalable design dimension.
format Preprint
id arxiv_https___arxiv_org_abs_2603_15925
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Generative Inverse Design with Abstention via Diagonal Flow Matching
de Campos, Miguel
Krebs, Werner
Gottschalk, Hanno
Machine Learning
Inverse design aims to find design parameters $x$ achieving target performance $y^*$. Generative approaches learn bidirectional mappings between designs and labels, enabling diverse solution sampling. However, standard conditional flow matching (CFM), when adapted to inverse problems by pairing labels with design parameters, exhibits strong sensitivity to their arbitrary ordering and scaling, leading to unstable training. We introduce Diagonal Flow Matching (Diag-CFM), which resolves this through a zero-anchoring strategy that pairs design coordinates with noise and labels with zero, making the learning problem provably invariant to coordinate permutations. This yields order-of-magnitude improvements in round-trip accuracy over CFM and invertible neural network baselines across design dimensions up to $P{=}100$. We develop two architecture-intrinsic uncertainty metrics, Zero-Deviation and Self-Consistency, that enable three practical capabilities: selecting the best candidate among multiple generations, abstaining from unreliable predictions, and detecting out-of-distribution targets; consistently outperforming ensemble and general-purpose alternatives across all tasks. We validate on airfoil, gas turbine combustor, and an analytical benchmark with scalable design dimension.
title Generative Inverse Design with Abstention via Diagonal Flow Matching
topic Machine Learning
url https://arxiv.org/abs/2603.15925