gongjing commited on
Commit
e3c4731
·
verified ·
1 Parent(s): fd7ce66

add overview description and metric

Browse files
Files changed (1) hide show
  1. README.md +16 -1
README.md CHANGED
@@ -1,5 +1,20 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  # Finetuning AIDO.Tissue for spatial single cell downstream tasks
2
- In this file, we introduce how to finetune and evaluate our pre-trained AIDO.Tissue foundation models for downstream tasks. These tasks can be classified into the following categories:
3
 
4
  * **Sequence-level classification tasks**: niche label type prediction
5
  * **Sequence-level regression tasks**: cell density prediction
 
1
+ # Overview
2
+ The AIDO.Tissue model is develpoed on spatial single-cell transcriptomic data. To incorporate spatial cell information, K nearest neighbor cells (K=8 in our case) are retrieved for each center cell. The center cell and neighbor cell expression vectors are concatenated as model input. Two-dimension positional rotary embedding are introduced to encode both the gene and cell information. The first dim is gene index and second dim is cell index. The overall training scheme is similar to scFoundation with an asymmetric encoder-decoder architecture.
3
+
4
+ # Results
5
+
6
+ We evelauted the model on two spatial data task, including predicting niche label and cell density. The metrics are as below:
7
+
8
+ | Task | F1-score |
9
+ | -------- | ------- |
10
+ | niche label type prediction | 0.67 |
11
+
12
+ | Task | Mean absolute error | R square |
13
+ | -------- | ------- | ------- |
14
+ | cell density prediction | 4.44 | 0.55|
15
+
16
  # Finetuning AIDO.Tissue for spatial single cell downstream tasks
17
+ We introduce how to finetune and evaluate our pre-trained AIDO.Tissue foundation models for downstream tasks. These tasks can be classified into the following categories:
18
 
19
  * **Sequence-level classification tasks**: niche label type prediction
20
  * **Sequence-level regression tasks**: cell density prediction