Back

Classifying epithelial-mesenchymal transition states in single cell cancer data using large language models

Pan, S.; Withnell, E.; Secrier, M.

2024-08-19 bioinformatics
10.1101/2024.08.16.608311 bioRxiv
Show abstract

Single-cell foundation models (scFMs) have been widely adopted for cell type annotation, yet their suitability for modelling cellular plasticity, where cells transition along continuous, context-dependent state trajectories, remains unclear. Here, we systematically benchmark state-of-the-art scFMs against conventional machine learning and bioinformatics approaches on epithelial-mesenchymal plasticity (EMP), a prototypical plastic cellular process. We show that naive fine-tuning of scFMs often fails to resolve intermediate states and is strongly influenced by tissue- and stimulus-specific signals. We propose a parameter-efficient dual-task adaptation strategy for EMP foundation models (EMP-FM) that combines discrete classification with pseudotime-guided regression, which improves cell state resolution in controlled settings (up to 85% AUROC), but remains sensitive to domain shifts. Across diverse in vitro and in vivo datasets, scFMs do not consistently outperform conventional methods, which often achieve comparable performance with lower complexity. Together, our results delineate both the potential and current limitations of scFMs for modelling cellular plasticity and support their complementary use alongside established bioinformatics approaches.

Matching journals

The top 6 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.