Hyperbolic Graph Representation Learning for Differential Diagnosis on Biomedical Knowledge Graphs

arXiv:2609.18481 · cs.AI, cs.LG · Submitted 2026-09-16 · Read on arXiv

cs.AI, cs.LG

Submitted: 2026-09-16

Updated: 2026-09-16

License: http://creativecommons.org/licenses/by/4.0/

The gist: Biomedical knowledge graphs combine ontology-derived hierarchies with transversal associations among heterogeneous entities such as phenotypes, diseases, genes, proteins, and patients.

Terminology

Abstract

Biomedical knowledge graphs combine ontology-derived hierarchies with transversal associations among heterogeneous entities such as phenotypes, diseases, genes, proteins, and patients. This hybrid structure raises the question of whether hyperbolic embeddings, which naturally capture tree-like organization, remain useful beyond purely hierarchical graphs. We present a preliminary study of hyperbolic graph representation learning for Mendelian-disease differential diagnosis on a patient-integrated biomedical graph. Experiments on isolated ontology subgraphs show that hyperbolic models achieve strong performance in substantially lower dimensions than Euclidean baselines. We then evaluate the models on a link-prediction task that ranks candidate diseases for each patient. Results suggest that hyperbolic embeddings can exploit biomedical hierarchical structure while supporting diagnostic reasoning over heterogeneous patient-level graphs.

Sources

Related papers