提出GAGA框架,实现数据流形上的几何感知生成与插值。
Geometry-Aware Generative Autoencoders for Warped Riemannian Metric Learning and Generative Modeling on Data Manifolds
- 构建基于流形学习的神经嵌入空间,学习变形黎曼度量。
- 在真实和模拟数据上提升30%轨迹推断性能。
- 适合单细胞测序等高维数据的生成与轨迹分析。
单细胞转录组学和空间基因组学等领域的高维数据快速增长,带来科学发现机遇的同时也引发计算与统计挑战。传统方法难以实现几何感知的数据生成、有意义轨迹的插值以及种群间的可行路径传输。为此,我们提出几何感知生成自编码器(GAGA),将可扩展的流形学习与生成建模相结合。GAGA构建一个尊重流形内在几何的神经网络嵌入空间,并在数据空间中学习一种新的变形黎曼度量。该度量由数据流形上的点及非流形负样本共同决定,能刻画整个潜在空间的有意义几何。基于此度量,GAGA可均匀采样流形点、沿测地线生成点,并通过测地线引导流实现跨种群插值。在模拟和真实数据集上表现优异,单细胞种群级轨迹推断相比最先进方法提升30%。
原文摘要 · Abstract (English)
Rapid growth of high-dimensional datasets in fields such as single-cell RNA sequencing and spatial genomics has led to unprecedented opportunities for scientific discovery, but it also presents unique computational and statistical challenges. Traditional methods struggle with geometry-aware data generation, interpolation along meaningful trajectories, and transporting populations via feasible paths. To address these issues, we introduce Geometry-Aware Generative Autoencoder (GAGA), a novel framework that combines extensible manifold learning with generative modeling. GAGA constructs a neural network embedding space that respects the intrinsic geometries discovered by manifold learning and learns a novel warped Riemannian metric on the data space. This warped metric is derived from both the points on the data manifold and negative samples off the manifold, allowing it to characterize a meaningful geometry across the entire latent space. Using this metric, GAGA can uniformly sample points on the manifold, generate points along geodesics, and interpolate between populations across the learned manifold using geodesic-guided flows. GAGA shows competitive performance in simulated and real-world datasets, including a 30% improvement over the state-of-the-art methods in single-cell population-level trajectory inference.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。