PreprintCited by 617Single-cell and spatial transcriptomicsGene expression and cancer classificationCell Image Analysis Techniques

scBERT as a Large-scale Pretrained Deep Language Model for Cell Type Annotation of Single-cell RNA-seq Data

Open access
Yang, Fan, Wang, Wenchuan, Wang, Fang, Fang, Yuan, Tang, Duyu, Huang, Junzhou, Lu, Hui, Yao, Jianhua·2021·Nature Machine Intelligence

Preprint — published as 10.1038/s42256-022-00534-z · Paper page

Abstract

Annotating cell types based on the single-cell RNA-seq data is a prerequisite for researches on disease progress and tumor microenvironment. Here we show existing annotation methods typically suffer from lack of curated marker gene lists, improper handling of batch effect, and difficulty in leveraging the latent gene-gene interaction information, impairing their generalization and robustness. We developed a pre-trained deep neural network-based model scBERT (single-cell Bidirectional Encoder Representations from Transformers) to overcome the challenges. Following BERT’s approach of pre-train and fine-tune, scBERT obtains a general understanding of gene-gene interaction by being pre-trained on huge amounts of unlabeled scRNA-seq data and is transferred to the cell type annotation task of unseen and user-specific scRNA-seq data for supervised fine-tuning. Extensive and rigorous benchmark studies validated the superior performance of scBERT on cell type annotation, novel cell type discovery, robustness to batch effect, and model interpretability.

AI snapshot

Written from the paper’s full text.

TL;DR

scBERT is a pretrained deep language model adapted from BERT/Performer for scRNA-seq cell type annotation, using self-supervised pretraining on unlabeled data and supervised fine-tuning. It reports superior accuracy, robustness to batch effects, novel cell type discovery, and gene-level interpretability across multiple benchmarks.

Key findings

  • Across 9 scRNA-seq datasets covering 17 major organs/tissues, more than 50 cell types, and over 500,000 cells, scBERT surpassed comparison methods in accuracy and macro F1 score on most datasets.
  • On the Zheng68K PBMC dataset, scBERT achieved F1=0.691 and accuracy=0.759 versus best other method F1=0.659 and accuracy=0.704; for highly similar CD8+ cytotoxic T and CD8+/CD45RA+ T cells, F1=0.788 vs. 0.617 (P=9.025e-5) and accuracy=0.801 vs. 0.724 (P=2.265e-5).
  • In cross-cohort pancreas benchmarks, scBERT reached accuracy=0.992 versus scNym=0.904, SciBet=0.985, Seurat=0.984, and SingleR=0.987; it correctly annotated >97% of cells in Muraro and >99% in the other three datasets.
  • For novel cell type discovery, scBERT achieved novel-class accuracy=0.329 vs. SciBet=0.174 and scmap_cluster=0.174, and known-class accuracy=0.942 vs. SciBet=0.784 and scmap_cluster=0.666; scBERT embedding clustering gave ARI=0.95 vs. 0.87 for raw expression.

Methods

  • Model: BERT-style pretrain/fine-tune paradigm with a Performer encoder (6 layers, 10 heads), gene2vec gene embeddings, expression binning into 200-dimensional vectors, self-supervised masking of non-zero expression for reconstruction, and supervised fine-tuning with a 1D convolution plus 3-layer neural network classifier.
  • Data: pretraining on PanglaoDB (209 human single-cell datasets, 74 tissues, 1,126,580 cells) and a large heart dataset (451,513 cells) for hyperparameter sensitivity; benchmarking used 9 datasets, 5-fold cross-validation, leave-one-dataset-out pancreas tests, class-imbalanced PBMC tests, MacParland novel cell type tests, and cross-organ Human Cell Atlas tests.
  • Analysis: accuracy and macro F1 score, Wilcoxon tests, adjusted Rand index, and enrichment analysis; comparisons included marker-based, correlation-based, and ML-based methods.

Limitations

  • Stated: expression binning sacrifices some data resolution; gene interaction network prior knowledge is not explicitly incorporated; masking only non-zero values may lower utilization of single-cell data for pretraining; and there is no unified quantitative evaluation metric for novel cell type detection.
  • Evident from reported results: novel cell type accuracy remains low (0.329), and cross-organ performance was only on par with comparison methods, indicating limited gains in those settings.

Model-written summary. Check it against the paper before relying on it.

Integrity signals

Registry

Crossref · 10.1101/2021.12.05.471261

Cites

67 of 67 references verified0 citation errors · 0 not checkable

Statistics

Nothing to recompute.

Claims the paper makes

0 of 34 cited sources checked · 0 supported · 0 flagged

  • Findingown data

    scBERT achieved the highest performance for CD8+ cytotoxic T cells and CD8+/CD45RA+ T cells (F1=0.788 vs 0.617, P=9.025e-5; accuracy=0.801 vs 0.724, P=2.265e-5).

    “and achieved the highest performance (F1 score=0.788 vs. 0.617, P-value=9.025e-5 ;accuracy=0.801vs. 0.724, P-value=2.265e-5 ) for CD8+ cytotoxic T cells and CD8+/CD45RA+ T cells that are highly similar and hard to distinguish in previous studies 35 .”
    Quote in the paperSubstantial claim0 of 1 sources checked
    • cites [35] 2017, Massively parallel digital transcriptional profiling of single cells, Nature Communications 2017 8:1

      Not checked in the free run.

  • Findingown data

    scBERT surpassed the performance of comparison methods in both accuracy and macro F1 score on most of the datasets.

    “scBERT surpassed the performance of comparison methods in both accuracy and macro F1 score on most of the datasets (Fig. 2a, Extended Data Figure 2).”
    Quote in the paperSubstantial claim
  • Findingown data

    Seurat misidentified CD8+ cytotoxic T cells to CD8+/CD45RA+ naive cytotoxic cells in class-imbalanced tests.

    “Seurat misidentifed CD8+ cytotoxic T cells to CD8+/CD45RA+ naive cytotoxic cells and SingleR misclassified all CD19+ B cells due to their rarity, while scBERT performed the least misclassification rate even though high similarity exists between the two cell populations (Fig. 2e, Extended Data Figure 4b).”
    Quote in the paperSubstantial claim
  • Findingreasoning

    On Zheng68K, even the SOTA method could not achieve an accuracy above 0.71.

    “Due to the severe cell type imbalance and the extremely high similarity between subtypes, even the SOTA method could not achieve an accuracy above 0.71.”
    Quote in the paperSubstantial claim
  • Findingown data

    SingleR misclassified all CD19+ B cells due to their rarity in class-imbalanced tests.

    “Seurat misidentifed CD8+ cytotoxic T cells to CD8+/CD45RA+ naive cytotoxic cells and SingleR misclassified all CD19+ B cells due to their rarity, while scBERT performed the least misclassification rate even though high similarity exists between the two cell populations (Fig. 2e, Extended Data Figure 4b).”
    Quote in the paperSubstantial claim
  • May combine two claims. 35
  • Classified as probably not a substantial claim. 19
  • May combine two claims.

    Findingown data

    With only 30% cells from Zheng68K, scBERT beat all comparison methods, and its performance improved rapidly as reference cell number increased.

    “With only 30% cells from Zheng 68K dataset), scBERT beated all comparison moethods, and its performance improved rapidly as reference cell number increased.”
    Quote in the paperSubstantial claim
  • May combine two claims.

    Findingown data

    The list of top-tier methods changes across different tasks and datasets while scBERT is always among it.

    “Notably, the list of the top-tier methods changes across different tasks and datasets while scBERT is always among it.”
    Quote in the paperSubstantial claim
  • May combine two claims.

    Findingreasoning

    Compared to SciBet and scmap_cluster, scBERT achieves superior accuracy on both novel class (scBERT=0.329 vs SciBet=0.174 and scmap_cluster=0.174) and known classes (scBERT=0.942 vs SciBet=0.784 and scmap_cluster=0.666).

    “Compared to SciBet and scmap_cluster, our method achieves the superior accuracy on both the novel class (scBERT=0.329vs. SciBet=0.174and scmap_cluster=0.174)and known classes (scBERT=0.942vs. SciBet=0.784and scmap_cluster=0.666).”
    Quote in the paperSubstantial claim
  • May combine two claims.

    Findingown data

    scBERT correctly annotated most cells (>97%) in the Muraro dataset and over 99% in the other three datasets.

    “scBERT correctly annotated most cells (> 97%) in the Muraro dataset and over 99% in the other three datasets, demonstrating the superb and stable performance of our method in cross-cohort tasks.”
    Quote in the paperSubstantial claim
  • May combine two claims.

    Findingown data

    scNym misclassified the alpha cells as beta cell type and was confused with the beta and delta cells.

    “In the contrast, scNym misclassified the alpha cells as beta cell type and was confused with the beta and delta cells (Fig. 3e-f).”
    Quote in the paperNot a substantial claim
  • May combine two claims.

    Findingreasoning

    scBERT could easily deal with large-scale datasets at both the pre-training and fine-tuning stages, benefiting from mini-batch sampling and the efficient Performer encoder.

    “Conversely, benefiting from mini-batch sampling and the efficient Performer encoder, scBERT could easily deal with large-scale datasets at both the pre-training and the fine-tuning stage.”
    Quote in the paperSubstantial claim
  • May combine two claims.

    Findingreasoning

    scBERT performed the best on novel cell types and achieved first-class performance on known cell types.

    “Not surprisingly, scBERT performed the best on novel cell types and achieved the first-class performance on the known cell types (Fig. 4).”
    Quote in the paperSubstantial claim
  • May combine two claims.

    Findingown data

    ML-based methods tend to learn high-level implicit cell-type-specific patterns, whereas non-ML based methods usually simply find differentially expressed genes using statistics analysis.

    “we could observe that ML-based methods tend to learn high-level implicit cell-type-specific patterns (i.e.discovering some genes with high rank across cell types), whereas non-ML based methods usually simply find differentially expressed genes using statistics analysis.”
    Quote in the paperNot a substantial claim
  • May combine two claims.

    Findingown data

    With deletion of 100% marker genes, scBERT can still efficiently learn informative gene patterns and achieve performance on par with the best performance achieved by comparison methods with all marker genes on the representative Zheng68K dataset.

    “With deletion of 100% marker genes, scBERT can still efficiently learn the informative gene patterns and achieve the performance on par with the best performance achieved by comparison methods with all marker genes on the representative Zheng68K dataset (Extended Data Figure 1b).”
    Quote in the paperSubstantial claim
  • May combine two claims.

    Findingnone given

    CellID_cell performed well on known cell types, while failed to discover any novel cells.

    “CellID_cell performed well on known cell types, while failed to discover any novel cells.”
    Quote in the paperNot a substantial claim
  • May combine two claims.

    Findingown data

    For cross-cohort data, scBERT achieved superior performance with accuracy of 0.992 compared to scNym (accuracy=0.904) with a large margin (about 10%) and outperformed other popular methods (accuracy: SciBet 0.985, Seurat 0.984, SingleR 0.987).

    “For cross-cohort data, scBERT achieved a superior performance with accuracy of 0.992 compared to scNym (accuracy=0.904)with a large margin (about 10%)and outperformed other popular methods (accuracy: SciBet 0.985, Seurat 0.984, SingleR 0.987) (Fig. 3b).”
    Quote in the paperSubstantial claim
  • May combine two claims.

    Findingown data

    On the Muraro dataset, the ARI score for the UMAP based on scBERT embedding is 0.95 and for the UMAP based on raw expression is 0.87.

    “Left UMAP is based on the scBERT embedding of each cell, while right UMAP is based on the raw expression of each cell.”
    • Also stated in: § Figure 5. Model interpretability. · plain
      “ARI=0.95ARI=0.87”
    Quote in the paperSubstantial claim
  • May combine two claims.

    Findingown data

    scBERT embedding exhibits cell-type-specific representation, with an example representation of a gene significantly enriched in alpha cell type.

    “We observed that the scBERT embedding exhibits cell-type-specific representation (the example representation of gene is significantly enriched in alpha cell type), which is naturally suitable for down-streaming cell type annotation task.”
    Quote in the paperSubstantial claim
  • May combine two claims.

    Findingnone given

    The model performs stably with 4 and 6 Performer encoder layers but might suffer from under-fitting or over-fitting when decreasing or increasing the number of layers.

    “Similarly, the model performs stable with 4 and 6 of Performer encoder layers but might suffer from under-fitting or over-fitting problem when decreasing or increasing the number of layers.”
    Quote in the paperNot a substantial claim
  • May combine two claims.

    Findingreasoning

    Small fluctuations of the hyperparameters had little effect on model performance, verifying the robustness of scBERT.

    “Overall, the small fluctuations of the above parameters had little effect on the performance of the model, which also verified the robustness of scBERT.”
    Quote in the paperNot a substantial claim
  • Probably not a substantial claim: may not pass S3 (load-bearing).

    Findingown data

    Enrichment analysis of top-50-attention gene lists revealed interesting relationships between top-enriched terms and the corresponding cell types.

    “Enrichment analysis was performed for the top-50-attention gene lists using various gene-set libraries and the results revealed there were some interesting relationships between top-enriched terms and the corresponding cell types.”
    Quote in the paperNot a substantial claim
    • Rests on an assumption: Attention weights reflect the contribution of each gene and the interaction of gene pairs.
  • May combine two claims.

    Findingown data

    scBERT can generate different representations of the same gene for different cell inputs, while gene2vec generates the same representation of the same gene for different cell inputs.

    “Our model could generate different representations of the same gene for different cell inputs, while gene2vec generated all the same representation of the same gene for different cell inputs.”
    Quote in the paperSubstantial claim
  • May combine two claims.

    Findingown data

    Gene2vec provided an embedding of 200 dimensions and achieved the best performance compared to other dimensions.

    “Second, the gene2vec provided an embedding of 200 dimensions and achieved the best performance compared to other dimensions.”
    Quote in the paperNot a substantial claim
  • May combine two claims.

    Premisecitation

    It is still challenging to distinguish true biological diversity from technical differences and preserve important biological variations despite batch-effect correction methods.

    “Although there exist batch-effect correction methods, it's still challenging to distinguish true biological diversity from technical differences and thus preserve important biological variations 13 .”
    Quote in the paperSubstantial claim0 of 1 sources checked
    • cites [13] source

      Not checked in the free run.

  • May combine two claims.

    Interpretationreasoning

    These uncertainties in performance reflect limitations of comparison methods in generalizability, as well as generalization of our method across all benchmarking datasets.

    “These uncertainties in performance reflect the limitations of the comparison methods in their generalizability, as well as the generalization of our method across all the benchmarking datasets.”
    Quote in the paperSubstantial claim
  • May combine two claims.

    Interpretationown data

    Pretraining enables scBERT to learn general syntax of gene-gene interactions, which helps to remove batch effects across datasets and improve generalizability.

    “Pretraining enables the model to learn general syntax of gene-gene interactions, which helps to remove the batch effects across datasets and improve the generalizability (Extended Data Figure 1a).”
    Quote in the paperSubstantial claim
    • Rests on an assumption: BERT's paradigm of pre-training and fine-tuning enables the use of large-scale unlabeled data to improve the generalizability of AI model.
  • Classified as probably not a substantial claim.

    Premisecitation

    BERT has made breakthrough progress in NLP due to the powerful self-attention mechanism and long-range information integration capability introduced by Transformer layers.

    “It has made breakthrough progress in the fields of natural language processing (NLP) due to the powerful self-attention mechanism and long-range information integration capability introduced by Transformer layers 24,25 .”
    Quote in the paperSubstantial claim0 of 2 sources checked
    • cites [24] Parmar, N. et al. Image Transformer. 4055–4064 Preprint at https://proceedings.mlr.press/v80/parmar18a.html (2018).

      Not checked in the free run.

    • cites [25] 2018, BERT: Pre-training of Deep Bidirectional Transformers for Language Understanding, NAACL HLT 2019 - 2019 Conference of the North American Chapter of the Association for Computational Linguistics: Human Language Technologies - Proceedings of the Conference

      Not checked in the free run.

  • May combine two claims.

    Premisecitation

    Existing marker-gene-based cell type annotation methods depend on manually curated marker genes and are prone to bias and errors.

    “However, selecting the marker genes depends on researchers' prior knowledge, and is therefore prone to bias and errors 7 .”
    Quote in the paperSubstantial claim0 of 1 sources checked
    • cites [7] source

      Not checked in the free run.

  • Probably not a substantial claim: may not pass S4 (based).

    Interpretationnone given

    Fine-tuning ensures that the output embedding for each gene encodes context information more relevant to the transcriptional profiles of the reference dataset.

    “Fine-tuning ensures that the output embedding for each gene encodes context information that is more relevant to the transcriptional profiles of the reference dataset.”
    Quote in the paperNot a substantial claim
  • Classified as probably not a substantial claim.

    Study design and conductnone given

    No manual selection of marker genes was included for an unbiased and fair comparison of all methods.

    “No manual selection of the marker genes was included for an unbiased and fair comparison of all the methods.”
    Quote in the paperSubstantial claim
  • May combine two claims.

    Interpretationown data

    scBERT can correctly discover novel cell types that are not present in original reference datasets and remain accurate prediction performance of other cell types.

    “Taken together, these results suggested that scBERT can correctly discover novel cell types that are not present in original reference datasets and remain accurate prediction performance of other cell types.”
    Quote in the paperSubstantial claim
    • Rests on an assumption: The reference dataset may not cover all cell types present in the query dataset in most tasks.
  • May combine two claims.

    Interpretationown data

    scBERT showed robustness in identifying cells from different sequencing technologies, experiments, different disease states (T2D and health), and even different organs.

    “scBERT showed its robustness in identifying cells from different sequencing technologies, experiments, different disease states (type 2 diabetes (T2D) and health), and even different organs.”
    Quote in the paperSubstantial claim
  • Classified as probably not a substantial claim. May combine two claims.

    Premisecitation

    The frequency of word occurrence in text is valuable information and is often transformed as bag-of-words by term frequency statistic analysis for downstream NLP tasks.

    “It's worth noting that the frequency of the word occurrence in a text is valuable information for text analysis and is often 320 transformed as bag-of-words by term frequency statistic analysis for downstream tasks in the area of 321 NLP 48 .”
    Quote in the paperSubstantial claim0 of 1 sources checked
    • cites [48] 2010, Understanding bag-of-words model: a statistical framework, International Journal of Machine Learning and Cybernetics 2010 1:1

      Not checked in the free run.

  • Probably not a substantial claim: may not pass S3 (load-bearing).

    Interpretationreasoning

    Masking only the non-zero values may lower the utilization of the single-cell data for pre-training, due to their minority.

    “However, masking only the non-zero values may lower the utilization of the single-cell data for pre-training, due to their minority.”
    Quote in the paperNot a substantial claim
  • Probably not a substantial claim: may not pass S3 (load-bearing), S4 (based).

    Interpretationnone given

    scBERT can also be migrated to other tasks by simply modifying output and supervision signals.

    “Beyond that, scBERT can also be migrated to other tasks by simply modifying output and supervision signals.”
    Quote in the paperNot a substantial claim
  • May combine two claims.

    Interpretationreasoning

    scBERT extracts useful attention patterns on gene expressions and interactions from large-scale various scRNA-seq data, alleviating fine-tuning efforts on specific downstream tasks.

    “The scBERT model extracts the useful attention pattern on gene expressions and interactions from a large scale of various scRNA-seq data, alleviating the efforts of the fine-tuning process on the specific downstream tasks.”
    Quote in the paperSubstantial claim
  • Probably not a substantial claim: may not pass S3 (load-bearing).

    Premisecitation

    Word embedding technology helps better represent text by ensuring words with similar meanings have similar representations.

    “The word embedding technology helps to better represent the text by assuring the words with similar meanings have a similar representation 47 .”
    Quote in the paperNot a substantial claim0 of 1 sources checked
    • cites [47] Goldberg, Y. Neural Network Methods for Natural Language Processing. http://dx.doi.org/10.2200/S00762ED1V01Y201703HLT037 10, 1–311 (2017). source

      Not checked in the free run.

  • May combine two claims.

    Interpretationown data

    scBERT has superb and stable performance in cross-cohort tasks.

    “demonstrating the superb and stable performance of our method in cross-cohort tasks.”
    Quote in the paperSubstantial claim
  • Probably not a substantial claim: may not pass S3 (load-bearing).

    Study design and conductcitation

    Well-documented marker lists associated with well-defined cell types from CellMarker were used to avoid bias introduced by marker selection.

    “To avoid bias introduced by marker selection, welldocumented marker lists associated with well-defined cell types from CellMarker 56 were used.”
    Quote in the paperNot a substantial claim0 of 1 sources checked
    • cites [56] source

      Not checked in the free run.

  • Probably not a substantial claim: may not pass S3 (load-bearing).

    Study design and conductnone given

    Accuracy, macro-F1 score, and confusion matrix were employed as evaluation metrics for benchmarking performance for fair comparison.

    “Notably, we employed the accuracy, macro-F1 score, and confusion matrix as evaluation metrics for benchmarking the performance of cell type annotation methods on their classification ability for a fair comparison in this study.”
    Quote in the paperNot a substantial claim
  • May combine two claims.

    Interpretationown data

    scBERT is efficient in learning single-cell-specific representation, which can be used for downstream analysis.

    “indicating the efficiency of scBERT in learning single-cell-specific representation, which can be used for downstream analysis (Fig. 5d).”
    Quote in the paperSubstantial claim
  • Probably not a substantial claim: may not pass S3 (load-bearing).

    Premisecitation

    Gene interactions usually exist in the form of networks, such as gene regulatory networks and biological signaling pathways.

    “Second, gene interactions usually exist in the form of networks (i.e., gene regulatory networks and biological signaling pathways) 43 , and this kind of prior knowledge hasn't been explicitly incorporated in scBERT.”
    Quote in the paperNot a substantial claim0 of 1 sources checked
    • cites [43] source

      Not checked in the free run.

  • May combine two claims.

    Premisecitation

    Gene2vec is based on bulk data and measures the average expression of genes from tissues, which is the sum of cell type-specific gene expression weighted by cell type proportions.

    “Gene2vec is based on bulk data 28 , which measures the average expression of genes from tissues and is the sum of cell type-specific gene expression weighted by cell type proportions 60 .”
    Quote in the paperSubstantial claim0 of 2 sources checked
    • cites [28] 2019, Gene2vec: distributed representation of genes based on co-expression, BMC Genomics 2019 20:1

      Not checked in the free run.

    • cites [60] source

      Not checked in the free run.

  • May combine two claims.

    Premisecitation

    SCD5 has not been reported as a cell-type-specific marker for beta cells, but in a GWAS study, a novel loci for T2D susceptibility was fine-mapped to a coding variant of SCD.

    “For instance, SCD5 hasn't been reported as a cell-type-specific marker for beta cells, but in a GWAS study, a novel loci for T2D susceptibility was fine-mapped to a coding variant of SCD 42 .”
    Quote in the paperNot a substantial claim0 of 1 sources checked
    • cites [42] source

      Not checked in the free run.

  • May combine two claims.

    Interpretationown data

    Extensive and rigorous benchmark studies validated the superior performance of scBERT on cell type annotation, novel cell type discovery, robustness to batch effect, and model interpretability.

    “Extensive and rigorous benchmark studies validated the superior performance of scBERT on cell type annotation, novel cell type discovery, robustness to batch effect, and model interpretability.”
    Quote in the paperSubstantial claim
  • May combine two claims.

    Interpretationown data

    scBERT surpasses existing advanced methods on diverse benchmarks collectively involving 9 single-cell datasets, 17 major organs/tissues, more than 50 cell types, over 500,000 cells, and mainstream single-cell omics technologies, indicating its generalization and robustness.

    “scBERT surpasses the existing advanced methods on diverse benchmarks collectively involving 9 singlecell datasets, 17 major organs/tissues, more than 50 cell types, over 500,000 cells, and the mainstream single-cell omics technologies (i.e., Drop-seq, 10X, SMART-seq, and Sanger-Nuclei), indicating its generalization and robustness.”
    Quote in the paperSubstantial claim
  • Probably not a substantial claim: may not pass S4 (based). May combine two claims.

    Study design and conductreasoning

    Only scRNA-seq datasets that provided highly credible cell type annotation and had been cited by the majority of cell type annotation methods for performance evaluation were included.

    “To this end, we only included scRNA-seq datasets that provided highly credible cell type annotation and had been cited by the majority of the cell type annotation methods for performance evaluation.”
    Quote in the paperNot a substantial claim
  • May combine two claims.

    Premisecitation

    Transformer with global receptive field can effectively learn global representation and long-range dependency without absolute position information, achieving excellent performance on non-sequential data such as images and tables.

    “Transformer with global receptive field could effectively learn global representation and longrange dependency without absolute position information, achieving excellent performance on nonsequential data (i.e., images, tables) 24,27 .”
    Quote in the paperSubstantial claim0 of 2 sources checked
    • cites [24] Parmar, N. et al. Image Transformer. 4055–4064 Preprint at https://proceedings.mlr.press/v80/parmar18a.html (2018).

      Not checked in the free run.

    • cites [27] source

      Not checked in the free run.

  • Classified as probably not a substantial claim.

    Study design and conductnone given

    For each dataset, 5-fold cross-validation was applied to avoid the influence of random results on the conclusion.

    “For each of the datasets, we applied the 5-fold cross-validation strategy to avoid the influence of random results on the conclusion.”
    Quote in the paperSubstantial claim
  • May combine two claims.

    Interpretationreasoning

    scBERT is prone to generate an efficient encoder and provide a general embedding that better represents the gene expression of each cell by revealing critical patterns with lower data noise.

    “Following the BERT's pre-training and fine-tuning paradigm, our method is prone to generate an efficient encoder and provide a general embedding that better represents the gene expression of each cell by revealing critical patterns with lower data noise.”
    Quote in the paperSubstantial claim
  • Probably not a substantial claim: may not pass S3 (load-bearing).

    Interpretationnone given

    An over-parameterized model with 20 attention heads faces a risk of over-fitting, especially when applied to small datasets.

    “However, the over-parameterized model (with 20 attention heads) faces a risk of over-fitting, especially when applying to small datasets.”
    Quote in the paperNot a substantial claim
  • Classified as probably not a substantial claim.

    Study design and conductown data

    The performance of cell type annotation methods was measured by accuracy and F1 score on n=9 datasets using 5-fold cross-validation.

    “Performance of cell type annotation methods measured by accuracy and F1 score on n=9 datasets using 5-fold cross-validation.”
    Quote in the paperSubstantial claim
  • Classified as probably not a substantial claim.

    Study design and conductreasoning

    A strict evaluation method for novel cell types was employed, with accuracy calculated on the union set of cells with the novel cell type label and cells predicted as novel cell types.

    “Notably, we employed a strict evaluation method for novel cell types with the accuracy calculated on the union set of cells with the novel cell type label and the cells that are predicted as novel cell types.”
    Quote in the paperSubstantial claim
  • May combine two claims.

    Interpretationreasoning

    Gene2vec maintains general coexpression patterns of genes but stays away from strong noise and high sparsity of single-cell sequencing.

    “In this regard, gene2vec maintains the general coexpression patterns of genes but stays away from strong noise and high sparsity of single-cell sequencing.”
    Quote in the paperSubstantial claim
  • Probably not a substantial claim: may not pass S3 (load-bearing).

    Premisecitation

    The Zheng68k dataset contains about 68,000 cells within eleven subtypes.

    “It contains about 68, 000 cells within eleven subtypes of cells”
    Quote in the paperNot a substantial claim0 of 1 sources checked
    • cites [35] 2017, Massively parallel digital transcriptional profiling of single cells, Nature Communications 2017 8:1

      Not checked in the free run.

  • Probably not a substantial claim: may not pass S3 (load-bearing).

    Study design and conductnone given

    Cross-validation was employed in all benchmarking experiments.

    “Cross-validation was employed in all the benchmarking experiments and standard deviations were drawn in the figures.”
    Quote in the paperNot a substantial claim
  • Probably not a substantial claim: may not pass S3 (load-bearing).

    Premisecitation

    The Zheng68k dataset is a classic PBMC dataset by 10X CHROMIUM widely used for cell type annotation performance assessment.

    “The Zheng68k is a classic PBMC dataset by 10X CHROMIUM that is widely used for cell type annotation performance acessment35 .”
    Quote in the paperNot a substantial claim0 of 1 sources checked
    • cites [35] 2017, Massively parallel digital transcriptional profiling of single cells, Nature Communications 2017 8:1

      Not checked in the free run.

52 of 52
  • The paper relies on this without support.

    Marker genes for interested cell types are not always available, and novel cell types do not have marker gene sets yet.

    “Additionally, maker genes for interested cell types are not always available, and novel cell types don't have marker gene sets yet.”
  • The paper relies on this without support.

    Dimensionality reduction like PCA may lose high-dimensional information as well as gene-level independent interpretability.

    “Dimensionality reduction like PCA may lose high-dimensional information as well as gene-level independent interpretability.”
  • The paper relies on this without support.

    Simple classifiers such as fully connected networks could not efficiently capture gene-gene interactions.

    “Besides, simple classifiers such as fully connected networks couldn't efficiently capture gene-gene interactions.”
  • The paper relies on this without support.

    BERT's paradigm of pre-training and fine-tuning enables the use of large-scale unlabeled data to improve the generalizability of AI model.

    “BERT's paradigm of pre-training and fine-tuning enables the use of large-scale unlabeled data to improve the generalizability of AI model.”
  • The paper relies on this without support.

    Existing ML methods select HVG or reduce dimension due to simplified network architecture and low model capacity, hence destroying gene-level interpretability.

    “Existing ML methods select HVG or reduce dimension due to their simplified network architecture and low model capacity hence destroying the gene-level interpretability.”
  • The paper relies on this without support.

    There is currently no research on applying Transformer architectures to gene expression data analysis.

    “To the best of our knowledge, there is currently no research on applying Transformer architectures to gene expression data analysis.”
  • The paper relies on this without support.

    The reference dataset may not cover all cell types present in the query dataset in most tasks.

    “The reference dataset may not cover all cell types present in the query dataset in most tasks.”
  • The paper relies on this without support.

    Gene expression can be considered as the occurrence of each gene in a biological system.

    “The gene expression can also be considered as the occurrence of each gene that has already been 322 well-documented in a biological system.”
  • The paper relies on this without support.

    Shuffling the columns of the input does not change its meaning.

    “Since shuffling the columns of our input does not change its meaning (like the extension of BERT to understand tabular data with TaBERT 27 ), absolute positions are meaningless for gene.”
  • The paper relies on this without support.

    At present, there is no unified quantitative evaluation metric for detection of novel cell types.

    “At present, there is no unified quantitative evaluation metrics for detection of novel cell type.”
  • The paper relies on this without support.

    Cell type annotation task and cell clustering task are not equivalent.

    “Since cell type annotation task and cell clustering task are not equivalent, those metrics assessing the quality and distance of clusters are excluded from this study.”
  • The paper relies on this without support.

    Attention weights reflect the contribution of each gene and the interaction of gene pairs.

    “The attention weights reflect the contribution of each gene and the interaction of gene pairs.”
  • The paper relies on this without support.

    Both marker gene data and gene expression pattern data imply discriminative information for different cell types.

    “Both kinds of methods were reported to achieve good performance on variable cell type annotation tasks, indicating that both types of data imply discriminative information for different cell types.”
  • The paper relies on this without support.

    Top attention genes obtained from the attention-sum vector correspond to a specific cell type compared to other cell types.

    “In this way, we could obtain the top attention genes corresponding to a specific cell type compared to other cell types.”
  • The paper relies on this without support.

    The quadratic computational complexity of the BERT model with Transformer as the basic unit does not scale well to long sequences, while the gene number of scRNA-seq can exceed 20,000.

    “The quadratic computational complexity of the BERT model with Transformer as the 326 basic unit does not scale very well to long sequence while the gene number of scRNA-seq can be up to 327 more than 20,000.”
  • The paper relies on this without support.

    Strong correlations between cell types make it difficult to differentiate them in the Zheng68k dataset.

    “Meanwhile, strong correlations between cell types make it difficult to differentiate them.”
16 of 16

Claims other papers make about this one

Each is checked against this paper’s own text: does it say what the citing paper says it does?

  • PowerSGD gradient compression accelerates the training process.

    “In the hardware setup, we find the training process is accelerated by incorporating the PowerSGD (51) gradient compression with rank 32 to reduce the communication loads across different GPUs.”
  • scBERT embeds each gene as the sum of a gene2vec embedding and an expression embedding representing the gene's normalized, binned log-scale expression level.

    “scBERT [7] embeds each gene as the sum of two embeddings: one representing the gene's normalized, binned log-scale expression level, and a second which specifies the gene identity via the previously developed gene2vec [12] embedding, allowing the model to differentiate genes in the otherwise order-less data.”
  • On the PBMC dataset, logistic regression outperformed scBERT in accuracy and F1 score for the difficult-to-distinguish CD8+ cytotoxic T cells and CD8+/CD45RA+ T cells.

    “Even for difficult-to-distinguish CD8+ cytotoxic T cells and CD8+/CD45RA+ T cells [7], logistic regression outperformed scBERT in terms of accuracy and F1 score (Table 1).”
  • In scBERT, the expression embedding, not the gene2vec embedding, drives the variability in the learned representations of a given gene for different cells and is the focus of the masking task used to pre-train the model.

    “The expression embedding - not the gene2vec embedding - drives the variability in the learned representations of a given gene for different cells and is the focus of the masking task used to pre-train the model.”

Self-check

Percentages, sums and reported differences re-computed from the paper’s own numbers.

Each contradiction between two passages listed here was confirmed by an independent check.

  • Class-imbalanced test requests more cells than Zheng68K contains

    Major

    The four cell types are listed in the order CD19+ B, CD8+ cytotoxic T, CD34+, and CD8+/CD45RA+ cells, so the stated reference sample requires 10,000 CD19+ B cells and 10,000 CD34+ cells. The paper says those types make up 8.6% and 0.4% of about 68,000 cells-roughly 5,850 and 270 cells, respectively.

  • MacParland dataset has two different reported sizes and cell-type counts

    Minor

    The Methods describe the MacParland dataset as having 8,444 cells in 20 populations, but the novel-cell-type experiment describes it as having 8,434 cells in 14 types. No filtering or subset selection is stated to reconcile the counts.

2 of 2

Cites

Flagged first, then by citations

Research works printed but not linked: 8

  1. 10.1038/s41467-020-16839-9 · DOI · Paper page
  2. 10.1038/s41467-020-16839-9 · DOI · Paper page
  3. 10.1186/s13059-019-1727-y · DOI · Paper page
  4. Parmar, N. et al. Image Transformer. 4055–4064 Preprint at https://proceedings.mlr.press/v80/parmar18a.html (2018).
  5. 10.1038/s41467-020-16839-9 · DOI · Paper page
  6. 10.1186/s13059-019-1727-y · DOI · Paper page
  7. Do Transformers Really Perform Badly for Graph Representation? https://proceedings.neurips.cc/paper/2021/hash/f1c1592588411002af340cbaedd6fc33-Abstract.html.
  8. 10.1186/s13059-020-1935-5 · DOI · Paper page

Cited by

Sorted by citations

Authors
Yang, F., Wang, W., Wang, F., Fang, Y., Tang, D., Huang, J., Lu, H., & Yao, J.
Affiliations
  • Tencent (China) (CN)
  • Shanghai Jiao Tong University (CN)
  • Harvard University (US)
  • The University of Texas at Arlington (US)
Corresponding author
  • Lu, Hui
  • Yao, Jianhua
Venue
Nature Machine Intelligence
Publisher
openRxiv
Published
2021-12-07
Volume
4
Issue
10
Pages
852-866
ISSN
2522-5839 (ISSN-L)
OpenAlex
W4297243391
Cited by
617
References
67
Citations by year
2026: 172 · 2025: 233 · 2024: 158 · 2023: 52 · 2022: 2
FWCI
36.41
Type
journal article
Language
en
Open access
Locations
  • Nature Machine Intelligence (publishedVersion)
  • bioRxiv (Cold Spring Harbor Laboratory) (acceptedVersion)
Funders
  • Yale University
  • Shanghai Jiao Tong University
Topics
  • Life Sciences › Biochemistry, Genetics and Molecular Biology › Molecular Biology › Single-cell and spatial transcriptomics
  • Life Sciences › Biochemistry, Genetics and Molecular Biology › Molecular Biology › Gene expression and cancer classification
  • Life Sciences › Biochemistry, Genetics and Molecular Biology › Biophysics › Cell Image Analysis Techniques
Keywords
cell type annotation, single-cell RNA sequencing, transformer models, self-supervised pre-training, gene-gene interaction, marker genes, fine-tuning, model interpretability

About these checks

Registry, retraction, Cites and statistics results are computed, not judged, with one exception: a printed reference that no index lists is read once by a model before it counts as a citation error. Claims are extracted by a model and kept only when their quote appears verbatim in the text; each cited source is then checked, with a model’s help, for whether it says what the claim relies on. Those verdicts can be wrong — read the claim’s quote and the cited source before relying on one.

Claims read by deepseek-v4.1-flash under definition version 3 (extract-v3) on 2026-10-10.

Classified by jev-1.13-20260917 (gold-guideline-v0.2) on 2026-10-10.

Statistics and arithmetic are recomputed, not judged.

Confidence bands are shown only for fields calibrated on our annotated test set. None is calibrated yet, so no band is shown.

What counts as a claim ›How we measure confidence ›How we check this ›

Are you an author, or do you see a problem with this page? Request a review