BASILISC: Bayesian Alignment of Siamese-Integrated Latents for Inferring Single-cell Correspondence
Organizations: The Berlin Institute for Medical Systems Biology, Max Delbrück Center for Molecular Medicine, Germany · Department of Electrical Engineering and Computer Science, Technical University of Berlin, Germany · Department of Biology, Humboldt University of Berlin, Germany · Department of Computer Science, Humboldt University of Berlin, Germany
Abstract
Single-cell assays study the genome one cell at a time. Unimodal assays measure one molecular layer, the newer multimodal assays measure several. Most data are single-modality, so integrating them relies on models trained on multimodal assays, where the cell correspondence across modalities is known. These models return a matching, but rarely a confidence measure. We present BASILISC (Bayesian Alignment of Siamese-Integrated Latents for Inferring Single-cell Correspondence), a framework for matching single-cell representations with uncertainty on the correspondence itself. A BASILISC model is a probabilistic siamese network over the two modalities with a predictive distribution over the correspondence for a cross-modal pair of cells. We instantiate it at two levels: at the cell-type level the question is whether two cells share a cell type, at the cell level which cell is the partner in the other modality. To our knowledge, this is the first uncertainty score on a learned cross-modal correspondence evaluated against a known ground-truth pairing. BASILISC learns a similarity across modalities that reflects cell lineage: an unseen cell type is matched to its lineage relative, matching still works on such held-out types (balanced accuracy 0.86 on unseen against 0.93 on seen types), and the predictive uncertainty rises on these out-of-distribution cells. The uncertainty score also selects high-quality matches for downstream analysis: discarding the least certain fifth of the pairs raises accuracy from 0.93 to 0.98. At the cell level, a deep ensemble of the deterministic matcher reaches 0.062 matching accuracy on the benchmark test set, above the competition winner; on the Bayesian matcher's assignment, the posterior flags the wrong matches that cross a cell-type boundary better than a deterministic read of the same scores (AUROC 0.68 against 0.64).
Figures & tables
| score | any wrong | same-type | cross-type | cross-type, partner is | |
|---|---|---|---|---|---|
| a relative | another family | ||||
| transport-plan posterior | 0.804 | 0.821 | 0.679 | 0.676 | 0.686 |
| top-1 vote | 0.777 | 0.790 | 0.678 | 0.672 | 0.697 |
| Sinkhorn weight, best ( ) | 0.792 | 0.812 | 0.635 | 0.646 | 0.603 |
| Sinkhorn weight, | 0.785 | 0.804 | 0.639 | 0.641 | 0.631 |
| seed stability | 0.701 | 0.707 | 0.656 | 0.654 | 0.662 |
Appendix figures & tables20 assets
Supplementary material from the paper’s appendix.
Appendix
| Multiome | CITE-seq | |
| Modalities | GEX, ATAC | GEX, ADT |
| Cells | 69,249 | 90,261 |
| Features | 13,431 genes, 116,490 peaks | 13,953 genes, 134 proteins |
| Annotated cell types | 22 | 45 |
| Types used | 22 | 38 ( cells) |
| Batches (site donor) | 13 | 12 |
| Multiome , 22 types | ||
|---|---|---|
| Family | Cell types | |
| T | 4 | CD4+ T naive, CD8+ T naive, CD4+ T activated, CD8+ T |
| innate lymphoid | 2 | ILC, NK |
| B | 5 | Lymph prog, Transitional B, Naive CD20+ B, B1 B, Plasma cell |
| mono/DC | 4 | CD14+ Mono, CD16+ Mono, ID2-hi myeloid prog, cDC2 |
| progenitor | 3 | HSC, G/M prog, MK/E prog |
| Multiome | |
|---|---|
| Cell type | Cells |
| CD8+ T | 11,589 |
| CD14+ Mono | 10,843 |
| NK | 6,929 |
| CD4+ T activated | 5,526 |
| Naive CD20+ B | 5,052 |
| Cell type | s1d1 | s1d2 | s1d3 | s2d1 | s2d4 | s2d5 | s3d3 | s3d6 | s3d7 | s3d10 | s4d1 | s4d8 | s4d9 |
|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
| CD8+ T | 725 | 1,632 | 1,020 | 169 | 230 | 752 | 378 | 488 | 110 | 432 | 759 | 4,089 | 805 |
| CD14+ Mono | 921 | 704 | 242 | 1,308 | 1,140 | 1,716 | 855 | 208 | 674 | 1,603 | 1,014 | 458 | – |
| NK | 591 | 982 | 696 | 269 | 136 | 848 | 477 | 126 | 256 | 545 | 1,124 | 421 | 458 |
| CD4+ T activated | 836 | 709 | 392 | 446 | 385 | 493 | 228 | 276 | 102 | 207 | 619 | 684 | 149 |
| Naive CD20+ B | 386 | 121 | 133 | 289 | 590 | 105 | 393 | 18 | 190 | 421 | 626 | 883 | 897 |
| Erythroblast | 797 | 136 | 374 | 478 | 1,485 | 64 | 246 | 84 | 31 | 391 | 474 | 266 | 90 |
| Cell type | s1d1 | s1d2 | s1d3 | s2d1 | s2d4 | s2d5 | s3d1 | s3d6 | s3d7 | s4d1 | s4d8 | s4d9 |
| CD14+ Mono | 734 | 726 | 309 | 2,958 | 1,294 | 2,561 | 2,915 | 2,927 | 6,056 | 637 | 466 | 110 |
| CD4+ T activated | 381 | 848 | 547 | 690 | 364 | 1,009 | 816 | 544 | 380 | 577 | 134 | 676 |
| CD4+ T naive | 591 | 304 | 833 | 883 | 266 | 631 | 681 | 349 | 219 | 583 | 87 | 470 |
| NK | 94 | 458 | 271 | 282 | 76 | 1,128 | 84 | 415 | 1,242 | 511 | 268 | 605 |
| Reticulocyte | 605 | 279 | 929 | 270 | 580 | 116 | 587 | 244 | 291 | 194 | 50 | 127 |
| Erythroblast | 228 | 57 | 228 | 1,202 | 471 | 61 | 622 | 524 | 307 | 150 | 35 | 154 |
| level | data | construction | arms | runs per arm |
|---|---|---|---|---|
| cell type | multiome | mix | BNN, legs, MCD, ENS, DET, -NN, cosine, logistic | 10 |
| cell type | multiome | LOCTO (22 types) | same | 220 |
| cell type | multiome | LOFO (7 families) | BNN | 70 |
| cell type | multiome | LTFO (21 pairs) | BNN | 210 |
| cell type | multiome | few-shot ( , 3 rules) | BNN | 1,260 |
| cell type | multiome | width ladder ( ) | BNN | 660 |
| mix AUROC | error AUROC, margin | gain at 50%, margin | novelty AUROC, | Spearman( , margin) | ||
|---|---|---|---|---|---|---|
| 0 | 0.004 | 0.978 | [ , ] | 0.584 | 0.956 0.027 | |
| 0.01 | 0.042 | 0.978 | – | – | – | 0.984 0.011 |
| 0.1 | 0.262 | 0.976 | [ , ] | 0.612 | 0.989 0.008 | |
| 0.3 | 0.489 | 0.976 | [ , ] | 0.623 | 0.987 0.008 | |
| 1 | 0.750 | 0.976 | [ , ] | 0.627 | 0.986 0.005 |
| method | accuracy | AUROC | Brier | NLL | ECE |
|---|---|---|---|---|---|
| BASILISC (MFVI) | 0.932 0.006 | 0.976 0.002 | 0.053 0.004 | 0.191 0.012 | 0.014 0.004 |
| MC dropout | 0.938 0.003 | 0.981 0.001 | 0.048 0.002 | 0.168 0.007 | 0.017 0.004 |
| deep ensemble (5) | 0.938 0.004 | 0.980 0.002 | 0.048 0.003 | 0.171 0.012 | 0.011 0.001 |
| deterministic (no sampling) | 0.932 0.002 | 0.976 0.002 | 0.052 0.001 | 0.186 0.007 | 0.013 0.002 |
| linear pair classifier | 0.917 0.006 | 0.969 0.002 | 0.063 0.003 | 0.218 0.010 | 0.022 0.005 |
| aligned cosine (Procrustes + Platt) | 0.914 0.004 | 0.965 0.003 | 0.066 0.004 | 0.229 0.011 | 0.013 0.003 |
| cross-batch pairs | within-batch pairs | |||||
|---|---|---|---|---|---|---|
| method | error AUROC | unseen AUROC | both | error AUROC | unseen AUROC | both |
| BASILISC (MFVI) | 0.834 | 0.619 | 69% | 0.833 | 0.635 | 75% |
| MC dropout | 0.843 | 0.707 | 88% | 0.864 | 0.710 | 85% |
| deep ensemble (5) | 0.850 | 0.671 | 82% | 0.870 | 0.679 | 84% |
| deterministic (no sampling) | 0.840 | 0.564 | 53% | 0.856 | 0.582 | 61% |
| linear pair classifier | 0.844 | 0.490 | 45% | 0.852 | 0.451 | 38% |
| multiome | CITE-seq | |
| mix AUROC, cross-batch / within-batch pairs | 0.976 / 0.977 | 0.970 / 0.963 |
| error detection, / margin | 0.856 / 0.858 | 0.795 / 0.797 |
| gain at 50% coverage, / margin | / | / |
| novelty detection, | 0.627 | 0.583 |
| above 0.5 in | 19 of 22 types | 28 of 38 types |
| look-alike partner, median nearest / other | 0.057 / 0.0015 | 0.029 / 0.0006 |
Cross-Modal Alignment
Cross-modal alignment maps representations from different data types into compatible spaces so models can compare, transfer, and combine information.
See all 445 papers