add overview description and metric
Browse files
README.md
CHANGED
@@ -1,5 +1,20 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
# Finetuning AIDO.Tissue for spatial single cell downstream tasks
|
2 |
-
|
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
|