CellPainTR: Generalizable Representation Learning for Cross-Dataset Cell Painting Analysis
Large-scale biological discovery requires integrating massive, heterogeneous datasets like those from the JUMP Cell Painting consortium, but technical batch effects and a lack of generalizable models remain critical roadblocks. To address this, we introduce CellPainTR, a Transformer-based architecture designed to learn foundational representations of cellular morphology that are robust to batch effects. Unlike traditional methods that require retraining on new data, CellPainTR's design, featuring source-specific context tokens, allows for effective out-of-distribution (OOD) generalization to entirely unseen datasets without fine-tuning. We validate CellPainTR on the large-scale JUMP dataset, where it outperforms established methods like ComBat and Harmony in both batch integration and biological signal preservation. Critically, we demonstrate its robustness through a challenging OOD task on the unseen Bray et al. dataset, where it maintains high performance despite significant domain and feature shifts. Our work represents a significant step towards creating truly foundational models for image-based profiling, enabling more reliable and scalable cross-study biological analysis.
Code (0)
등록된 구현이 없습니다.
Tasks
Representation LearningSimilar Papers 제목 키워드 기반
HuiduRep: A Robust Self-Supervised Framework for Learning Neural Representations from Extracellular Recordings
Extracellular recordings are transient voltage fluctuations in the vicinity of neurons, serving as a fundamental modality in neuroscience for decoding brain activity at single-neuron resolution. Spike sorting, the proces…
Representation LearningContrastive LearningTowards robust and generalizable representations of extracellular data using contrastive learning
Contrastive learning is quickly becoming an essential tool in neuroscience for extracting robust and meaningful representations of neural activity. Despite numerous applications to neuronal population data, there has bee…
SAVE: A Generalizable Framework for Multi-Condition Single-Cell Generation with Gene Block Attention
Modeling single-cell gene expression across diverse biological and technical conditions is crucial for characterizing cellular states and simulating unseen scenarios. Existing methods often treat genes as independent tok…
Automated Annotation of Cell Identities in Dense Cellular Images
Assigning cell identities in dense image stacks is critical for many applications, for comparing data across animals and experiment conditions, and investigating properties of specific cells. Conventional methods are lab…
Learning Cell-Aware Hierarchical Multi-Modal Representations for Robust Molecular Modeling
Understanding how chemical perturbations propagate through biological systems is essential for robust molecular property prediction. While most existing methods focus on chemical structures alone, recent advances highlig…
Molecular Property Prediction