POME: Graph-based embeddings for partially observed mixed-type data
- 1Friedrich-Alexander-University Erlangen-Nuremberg
Abstract
Partially observed mixed-type (POM) data, as often encountered in clinical, epidemiological, and phenotypic datasets, are very common in biomedical research. Yet, advanced data analysis and machine learning based on POM data is complicated by their heterogeneous nature and often substantial fractions of missing values. While one promising way to overcome these issues is to compute vector-valued embeddings for POM datasets which can then be used for downstream analyses, existing embedding methods are mostly not designed for POM data. To address this gap, we developed POME (partially observed mixed-type data embeddings), a self-supervised model that yields low-dimensional representations of both samples and variables, using shared concept learning and a bipartite graph representation of the underlying POM data. We validated POME through extensive experiments on three real-world biomedical datasets, with diverse downstream tasks and objectives: POME achieves state-of-the-art imputation performance and produces high-quality patient representations that support not only unsupervised discovery of well-separated and clinically meaningful patients subgroups but also supervised predictive modeling and zero-shot representation space mining for use cases such as adjuvant therapy modality recommendation. POME is available as a Python package on GitHub (https://github.com/bionetslab/POME) and PyPI (https://pypi.org/project/pome-py).
Related
- Structural identifiability of partially-observed stochastic processes: from single-particle trajectories to total particle density data
- Demographic inference of pathogen-infected populations from partially observed transmission forests
- MOSAIC: Learning Graph Node Embeddings from Spectrally Isolated Dominant Subspaces of Accumulated Diffusion Operators
- Inhibition of cellular NEDDylation partially alters the MHC class I immunopeptidome
- Exploring partially shared valuation-related neural responses between abstract desirability and food incentive evaluation
BibTeX
@misc{woller2026pome,
title = {{POME: Graph-based embeddings for partially observed mixed-type data}},
author = {F. Woller and L. Arend and A. M. Kist and M. List and F. Rahimi and C. Sirocchi and D. B. Blumenthal},
howpublished = {bioRxiv},
year = {2026},
month = sep,
doi = {10.64898/2026.09.10.750629},
url = {https://www.biorxiv.org/content/10.64898/2026.09.10.750629v1},
note = {CC BY},
}
RIS
TY - UNPB
TI - POME: Graph-based embeddings for partially observed mixed-type data
AU - F. Woller
AU - L. Arend
AU - A. M. Kist
AU - M. List
AU - F. Rahimi
AU - C. Sirocchi
AU - D. B. Blumenthal
PY - 2026
DA - 2026/09/10
PB - bioRxiv
DO - 10.64898/2026.09.10.750629
UR - https://www.biorxiv.org/content/10.64898/2026.09.10.750629v1
AB - Partially observed mixed-type (POM) data, as often encountered in clinical, epidemiological, and phenotypic datasets, are very common in biomedical research. Yet, advanced data analysis and machine learning based on POM data is complicated by their heterogeneous nature and often substantial fractions of missing values. While one promising way to overcome these issues is to compute vector-valued embeddings for POM datasets which can then be used for downstream analyses, existing embedding methods are mostly not designed for POM data. To address this gap, we developed POME (partially observed mixed-type data embeddings), a self-supervised model that yields low-dimensional representations of both samples and variables, using shared concept learning and a bipartite graph representation of the underlying POM data. We validated POME through extensive experiments on three real-world biomedical datasets, with diverse downstream tasks and objectives: POME achieves state-of-the-art imputation performance and produces high-quality patient representations that support not only unsupervised discovery of well-separated and clinically meaningful patients subgroups but also supervised predictive modeling and zero-shot representation space mining for use cases such as adjuvant therapy modality recommendation. POME is available as a Python package on GitHub (https://github.com/bionetslab/POME) and PyPI (https://pypi.org/project/pome-py).
ER -