2021/06/07 by Dimitris Kalatzis, Kalatzis, Dimitris, Johan Ziruo Ye +7 · 1 citation
Computer Science · #AI in cancer detection #Digital Imaging for Blood Diseases #Topological and Geometric Data Analysis
paper · pdf · doi:10.48550/arxiv.2106.03500
We present a framework for learning probability distributions on topologically non-trivial manifolds, utilizing normalizing flows. Current methods focus on manifolds that are homeomorphic to Euclidean space, enforce strong structural priors on the learned models or use operations that do not easily scale to high dimensions. In contrast, our method learns distributions on a data manifold by "gluing" together multiple local models, thus defining an open cover of the data manifold. We demonstrate the efficiency of our approach on synthetic data of known manifolds, as well as higher dimensional manifolds of unknown topology, where our method exhibits better sample efficiency and competitive or superior performance against baselines in a number of tasks.