Relative Entropy Estimation in Function Space: Theory and Applications to Trajectory Inference

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Wang, Chao, Nepote, Luca, Franzese, Giulio, Michiardi, Pietro
Formato: Preprint
Publicado: 2026
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866910157607272448
author Wang, Chao
Nepote, Luca
Franzese, Giulio
Michiardi, Pietro
author_facet Wang, Chao
Nepote, Luca
Franzese, Giulio
Michiardi, Pietro
contents Trajectory Inference (TI) seeks to recover latent dynamical processes from snapshot data, where only independent samples from time-indexed marginals are observed. In applications such as single-cell genomics, destructive measurements make path-space laws non-identifiable from finitely many marginals, leaving held-out marginal prediction as the dominant but limited evaluation protocol. We introduce a general framework for estimating the Kullback-Leibler divergence (KL) divergence between probability measures on function space, yielding a tractable, data-driven estimator that is scalable to realistic snapshot datasets. We validate the accuracy of our estimator on a benchmark suite, where the estimated functional KL closely matches the analytic KL. Applying this framework to synthetic and real scRNA-seq datasets, we show that current evaluation metrics often give inconsistent assessments, whereas path-space KL enables a coherent comparison of trajectory inference methods and exposes discrepancies in inferred dynamics, especially in regions with sparse or missing data. These results support functional KL as a principled criterion for evaluating trajectory inference under partial observability.
format Preprint
id arxiv_https___arxiv_org_abs_2604_20775
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Relative Entropy Estimation in Function Space: Theory and Applications to Trajectory Inference
Wang, Chao
Nepote, Luca
Franzese, Giulio
Michiardi, Pietro
Machine Learning
Trajectory Inference (TI) seeks to recover latent dynamical processes from snapshot data, where only independent samples from time-indexed marginals are observed. In applications such as single-cell genomics, destructive measurements make path-space laws non-identifiable from finitely many marginals, leaving held-out marginal prediction as the dominant but limited evaluation protocol. We introduce a general framework for estimating the Kullback-Leibler divergence (KL) divergence between probability measures on function space, yielding a tractable, data-driven estimator that is scalable to realistic snapshot datasets. We validate the accuracy of our estimator on a benchmark suite, where the estimated functional KL closely matches the analytic KL. Applying this framework to synthetic and real scRNA-seq datasets, we show that current evaluation metrics often give inconsistent assessments, whereas path-space KL enables a coherent comparison of trajectory inference methods and exposes discrepancies in inferred dynamics, especially in regions with sparse or missing data. These results support functional KL as a principled criterion for evaluating trajectory inference under partial observability.
title Relative Entropy Estimation in Function Space: Theory and Applications to Trajectory Inference
topic Machine Learning
url https://arxiv.org/abs/2604.20775