Branched Optimal Transport Amortization
This work introduces a novel, scalable branched flow-matching algorithm designed to solve the branched optimal transport problem in high dimensions and demonstrates its effectiveness on challenging high-dimensional tasks in biology and image generation.