RankByGene: Gene-Guided Histopathology Representation Learning Through Cross-Modal Ranking Consistency

Wentao Huang,Meilong Xu,Xiaoling Hu,Shahira Abousamra,Aniruddha Ganguly,Saarthak Kapse,Alisa Yurovsky,Prateek Prasanna,Tahsin Kurc,Joel Saltz,Michael L. Miller,Chao Chen
2024-11-23
Abstract:Spatial transcriptomics (ST) provides essential spatial context by mapping gene expression within tissue, enabling detailed study of cellular heterogeneity and tissue organization. However, aligning ST data with histology images poses challenges due to inherent spatial distortions and modality-specific variations. Existing methods largely rely on direct alignment, which often fails to capture complex cross-modal relationships. To address these limitations, we propose a novel framework that aligns gene and image features using a ranking-based alignment loss, preserving relative similarity across modalities and enabling robust multi-scale alignment. To further enhance the alignment's stability, we employ self-supervised knowledge distillation with a teacher-student network architecture, effectively mitigating disruptions from high dimensionality, sparsity, and noise in gene expression data. Extensive experiments on gene expression prediction and survival analysis demonstrate our framework's effectiveness, showing improved alignment and predictive performance over existing methods and establishing a robust tool for gene-guided image representation learning in digital pathology.
Image and Video Processing,Computer Vision and Pattern Recognition,Quantitative Methods
What problem does this paper attempt to address?