用几何潜空间建模离散数据,提升生成效率与精度
Generative Modeling of Discrete Data Using Geometric Latent Subspaces
- 在类别分布的指数参数空间中构建几何潜子空间,捕捉变量间统计依赖
- 低维潜空间可准确建模高维离散数据,实现高效生成
- 基于黎曼几何的PCA方法,使潜空间中的测地线为直线,利于流匹配训练
我们提出一种几何潜子空间框架,用于离散数据的生成建模。具体而言,在类别分布的乘积流形指数参数空间中引入潜子空间,作为学习离散数据生成模型的新方法。所得低维潜空间编码了类别变量间的统计依赖性,并消除了冗余自由度。我们在参数域上赋予黎曼几何结构,使得潜子空间与诱导数据流形之间通过等距映射关联,支持一致的流匹配。利用该结构,我们提出一种几何感知的降维目标——几何主成分分析(GPCA),将其形式化为正则化交叉熵最小化问题,鼓励数据与其重构之间的黎曼距离尽可能小。特别地,在诱导几何下,测地线在潜参数空间中表现为直线,使基于流匹配的模型训练更加有效。实验证明,低维潜表示足以准确建模高维离散数据。
原文摘要 · Abstract (English)
We propose a geometric latent-subspace framework for generative modeling of discrete data. Specifically, we introduce latent subspaces in the exponential parameter space of product manifolds of categorical distributions as a novel method for learning generative models of discrete data. The resulting low-dimensional latent space encodes statistical dependencies and removes redundant degrees of freedom among the categorical variables. We equip the parameter domain with a Riemannian geometry such that the latent subspace and induced data manifold are related by isometries enabling consistent flow matching. Exploiting this structure, we propose a geometry-aware dimensionality reduction objective, called geometric PCA (GPCA), which we formulate as a regularized cross-entropy minimization that encourages small Riemannian distances between the data and their reconstructions. In particular, under the induced geometry, geodesics become straight lines in the latent parameter space which makes model training by flow matching effective. Empirical results show that low-dimensional latent representations suffice to accurately model high-dimensional discrete data.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。