Boundary-Guided Learning for Gene Expression Prediction in Spatial Transcriptomics

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Qu, Mingcheng, Wu, Yuncong, Di, Donglin, Su, Anyang, Su, Tonghua, Song, Yang, Fan, Lei
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910732596019200
author Qu, Mingcheng
Wu, Yuncong
Di, Donglin
Su, Anyang
Su, Tonghua
Song, Yang
Fan, Lei
author_facet Qu, Mingcheng
Wu, Yuncong
Di, Donglin
Su, Anyang
Su, Tonghua
Song, Yang
Fan, Lei
contents Spatial transcriptomics (ST) has emerged as an advanced technology that provides spatial context to gene expression. Recently, deep learning-based methods have shown the capability to predict gene expression from WSI data using ST data. Existing approaches typically extract features from images and the neighboring regions using pretrained models, and then develop methods to fuse this information to generate the final output. However, these methods often fail to account for the cellular structure similarity, cellular density and the interactions within the microenvironment. In this paper, we propose a framework named BG-TRIPLEX, which leverages boundary information extracted from pathological images as guiding features to enhance gene expression prediction from WSIs. Specifically, our model consists of three branches: the spot, in-context and global branches. In the spot and in-context branches, boundary information, including edge and nuclei characteristics, is extracted using pretrained models. These boundary features guide the learning of cellular morphology and the characteristics of microenvironment through Multi-Head Cross-Attention. Finally, these features are integrated with global features to predict the final output. Extensive experiments were conducted on three public ST datasets. The results demonstrate that our BG-TRIPLEX consistently outperforms existing methods in terms of Pearson Correlation Coefficient (PCC). This method highlights the crucial role of boundary features in understanding the complex interactions between WSI and gene expression, offering a promising direction for future research.
format Preprint
id arxiv_https___arxiv_org_abs_2412_04072
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Boundary-Guided Learning for Gene Expression Prediction in Spatial Transcriptomics
Qu, Mingcheng
Wu, Yuncong
Di, Donglin
Su, Anyang
Su, Tonghua
Song, Yang
Fan, Lei
Machine Learning
Spatial transcriptomics (ST) has emerged as an advanced technology that provides spatial context to gene expression. Recently, deep learning-based methods have shown the capability to predict gene expression from WSI data using ST data. Existing approaches typically extract features from images and the neighboring regions using pretrained models, and then develop methods to fuse this information to generate the final output. However, these methods often fail to account for the cellular structure similarity, cellular density and the interactions within the microenvironment. In this paper, we propose a framework named BG-TRIPLEX, which leverages boundary information extracted from pathological images as guiding features to enhance gene expression prediction from WSIs. Specifically, our model consists of three branches: the spot, in-context and global branches. In the spot and in-context branches, boundary information, including edge and nuclei characteristics, is extracted using pretrained models. These boundary features guide the learning of cellular morphology and the characteristics of microenvironment through Multi-Head Cross-Attention. Finally, these features are integrated with global features to predict the final output. Extensive experiments were conducted on three public ST datasets. The results demonstrate that our BG-TRIPLEX consistently outperforms existing methods in terms of Pearson Correlation Coefficient (PCC). This method highlights the crucial role of boundary features in understanding the complex interactions between WSI and gene expression, offering a promising direction for future research.
title Boundary-Guided Learning for Gene Expression Prediction in Spatial Transcriptomics
topic Machine Learning
url https://arxiv.org/abs/2412.04072