Back

A Computational Approach to Interpreting the Embedding Space of Dimension Reduction

Zhang, B.; Uno, K.; Kodama, H.; Himori, K.; Matsui, Y.

2024-06-27 bioinformatics
10.1101/2024.06.23.600292 bioRxiv
Show abstract

Nonlinear dimension reduction methods are widely applied in studies analyzing gene and protein expression, by revealing patterns of discrete groups and continuous orders in high-dimensional data. However, the tools are limited to understanding the obtained embedding structures of biological mechanisms, hindering the full exploitation of data. Here, we propose a novel framework to interpret embedding systematically by identifying and mapping associated biological functions. The method performs statistical tests and visualizes significantly enriched functions essential for the organization of the embedding structure, by applying it to the embedding results of two datasets: the Genotype Tissue Expression dataset and a Caenorhabditis elegans embryogenesis dataset, one capturing distinct cluster structures and the other capturing continuous developmental trajectories. We identified the associated functions for interpreting the two embeddings and confirmed it as a useful explainable AI tool in exploratory data analysis by providing annotations to the embedding space.

Matching journals

The top 5 journals account for 50% of the predicted probability mass.

50% of probability mass above

"Similar papers" are the closest papers from that journal in the model's embedding space. They show what the match is built on, but the ranking comes mostly from a classifier over the whole training set, not from these examples alone.