Paper deep dive
scHFBiMamba: a bidirectional Mamba autoencoder for cardiac disease-state classification from single-nucleus RNA-seq data
Ugo Lomoio, Pietro Hiram Guzzi, Pierangelo Veltri
Intelligence
Status: not_run | Model: - | Prompt: - | Confidence: 0%
Entities (0)
Relation Signals (0)
No relation signals yet.
Cypher Suggestions (0)
No Cypher suggestions yet.
Abstract
Abstract:To evaluate whether a compact bidirectional state-space autoencoder improves joint classification of species, cell type, and disease state from single-nucleus RNA-sequencing profiles in heart failure (HF), while yielding interpretable gene attributions. Building on the dense denoising autoencoder (DAE) and MLP framework of Ruz Jurado et al., we developed scHFBiMamba, combining a bidirectional Mamba autoencoder (BiMambaAE) with an MLP head. BiMambaAE maps 16,545-gene profiles to 350-dimensional latent representations using 13.2M parameters, about 14.6-fold fewer than the dense DAE (192.9M). We compared six encoder-classifier configurations on 38,275 held-out cardiac nuclei using McNemar tests with Benjamini-Hochberg correction, and assessed nine classifiers on frozen representations. Biological relevance was evaluated by comparing SHAP-derived differentially explained genes with differentially expressed genes against curated HF-related Gene Ontology keywords. BiMambaAE+MLP achieved the highest aggregate F1: pooled macro-F1 across 13 outputs was 0.966 versus 0.957 for DAE+MLP, and the mean category macro-F1 was 0.977 versus 0.970. It significantly improved cell-type and disease-state accuracy (adjusted p=7.1e-3 and p=4.8e-6, respectively), with species F1 of 1.000. SHAP-derived differentially explained genes achieved a pathway-enrichment benchmark F1 of 0.105 versus 0.053 for differentially expressed genes. Compact bidirectional state-space autoencoders can outperform substantially larger dense autoencoders for joint cardiac transcriptomic classification while reducing parameters. scHFBiMamba enables accurate multi-task prediction and hypothesis-generating gene-level interpretation. Code: this https URL.
Tags
Links
- Source: https://arxiv.org/abs/2610.15117v1
- Canonical: https://arxiv.org/abs/2610.15117v1
Trouble viewing inline? Open PDF directly â
Full Text
55,410 characters extracted from source content.
Expand or collapse full text
scHFBiMamba: a bidirectional Mamba autoencoder for cardiac disease-state classification from single-nucleus RNA-seq data Ugo Lomoio Affiliation: Department of Surgical and Medical Sciences Affiliation: University âMagna GrĂŠciaâ of Catanzaro Affiliation: 88100 Catanzaro, Italy Email: ugo.lomoio@unicz.it Pietro Hiram Guzzi â thanks: Corresponding author. Affiliation: Department of Surgical and Medical Sciences Affiliation: University âMagna GrĂŠciaâ of Catanzaro Affiliation: 88100 Catanzaro, Italy Email: hguzzi@unicz.it Pierangelo Veltri Affiliation: DIMES Affiliation: University of Calabria Affiliation: 88063 Rende, Italy Email: pierangelo.veltri@dimes.unical.it Abstract Objective: To evaluate whether a compact bidirectional state-space autoencoder can improve joint classification of species, cell type, and disease state from single-nucleus RNA-sequencing profiles in heart failure (HF) and related cardiac conditions, while providing biologically interpretable gene attributions. Methods: Building on the dense denoising autoencoder (DAE) and multilayer perceptron (MLP) framework of Ruz Jurado et al., we developed scHFBiMamba, a framework comprising a bidirectional Mamba autoencoder (BiMambaAE) and an MLP classification head. BiMambaAE maps 16,545-gene single-nucleus expression profiles to 350-dimensional representations through learned latent tokens. The autoencoder contains 13.2M parameters, approximately 14.6Ă14.6Ă fewer than the dense DAE (192.9M). We compared six encoderâclassifier configurations on a held-out test set of 38,275 cardiac nuclei, assessed nucleus-level differences using McNemar tests with BenjaminiâHochberg correction, and evaluated nine classifier variants on frozen BiMambaAE representations. Biological relevance was assessed by comparing SHAP-derived differentially explained genes with differentially expressed genes against curated HF-related Gene Ontology keywords. Results: BiMambaAE with an MLP head achieved the highest aggregate F1 among the six configurations under both aggregate definitions: macro-F1 pooled over all 13 outputs was 0.966 versus 0.957 for DAE+MLP, and the arithmetic mean of the three category macro-F1 scores was 0.977 versus 0.970. BiMambaAE+MLP showed significantly higher cell-type and disease-state classification accuracy (BenjaminiâHochberg-adjusted p=7.1Ă10â3p=7.1Ă 10^-3 and p=4.8Ă10â6p=4.8Ă 10^-6, respectively), with a species F1 of 1.000. Among nine classifier variants, the MLP was a practical choice for prediction and gradient-based gene attribution. SHAP-derived differentially explained genes achieved a mean pathway-enrichment benchmark F1 of 0.105, compared with 0.053 for differentially expressed genes. Conclusion: Compact bidirectional state-space autoencoders can outperform substantially larger dense autoencoders for joint cardiac transcriptomic classification while reducing parameter count. scHFBiMamba provides accurate multi-task prediction and hypothesis-generating gene-level interpretation, supporting further evaluation of bidirectional Mamba architectures for single-nucleus HF transcriptomics. Code is available at https://github.com/UgoLomoio/scHFBiMamba. Keywords heart failure â · single-nucleus RNA-seq â · Mamba â · state-space models â · autoencoder â · explainable AI â · SHAP â · differential expression 1 Introduction Heart failure (HF) is a major cause of morbidity and mortality worldwide. Its clinical heterogeneity includes HF with reduced ejection fraction (HFrEF) and HF with preserved ejection fraction (HFpEF); related pressure-overload conditions, including aortic stenosis (AS), provide an additional context for studying cardiac remodeling [1]. Single-nucleus RNA sequencing (snRNA-seq) enables the characterization of cardiac disease at the level of individual cell types [2]. However, the dimensionality, sparsity, and cross-species heterogeneity of cardiac expression atlases complicate the identification of features that distinguish disease states. Ruz Jurado et al. introduced a neural-network framework that classifies each cardiac nucleus by species (human or mouse), cell type (seven classes), and disease state (CTRL, AS, HFrEF, or HFpEF), using normalized expression profiles of 16,545 one-to-one orthologous genes [3]. The framework combines a dense denoising autoencoder (DAE; 16545â5000â2400â350 encoder) with a multilayer perceptron (MLP; 795â230â105 hidden layers). Shapley additive explanations (SHAP) [4] are then used to identify differentially explained genes (DXGs), whose model attributions differ between disease and control groups. The reported DXG analysis identified HF-related pathways that were not recovered using conventional differentially expressed genes (DEGs), motivating the integration of predictive modeling and attribution-based interpretation. The parameter cost of the dense autoencoder motivates the evaluation of alternative architectures. The DAE contains â 193M parameters, with much of this capacity allocated to fully connected input and reconstruction layers. Related biomedical applications have explored denoising convolutional autoencoders for ECG super-resolution [5] and bidirectional Mamba-2 architectures for EEG super-resolution [6]. Alongside attention-based Transformers [7], including those used in single-cell models such as scGPT [8], selective state-space models (SSMs) offer an alternative mechanism for sequence modeling. Mamba [9] and Mamba-2 [10] support input-dependent sequence mixing with linear scaling in sequence length; bidirectional variants have also been investigated in biomedical signal reconstruction [6]. These developments motivate testing whether a compact state-space autoencoder can improve the accuracyâparameter trade-off in cardiac snRNA-seq classification. Single-cell foundation models provide a broader context for this question. Geneformer learns from rank-encoded gene sequences and was pretrained on ⌠30M transcriptomes [11]; scBERT applies BERT-style pretraining to single-cell expression data [12]; the Universal Cell Embedding (UCE) model learns cross-species representations from 36M cells spanning eight species [13]; and scFoundation uses a 100M-parameter architecture covering ⌠20,000 genes and pretraining on over 50M profiles [14]. These approaches target transferable representations across datasets and tasks. In contrast, the present study evaluates a task-specific architecture within an existing cardiac classification and DXG pipeline [3]. The central question is whether this pipeline can retain predictive and attribution-based utility with substantially fewer autoencoder parameters. Pretraining cost, downstream adaptation cost, and attribution support are distinct considerations; neither model size nor pretraining alone precludes the use of a differentiable classifier or gene-level explanations. Related approaches include probabilistic latent-variable models such as scVI [15] and several applications of state-space architectures to single-cell data. These include SC-MAMBA2 [16], GeneMamba [17], scMamba [18], and RegFormer [19]. The architectural feature examined here is the placement of a bidirectional Mamba2 block after an initial expression bottleneck: sequence mixing acts on 32 learned latent tokens, rather than directly on the 16,545 input genes. This reduces the sequence length processed by the SSM while retaining a differentiable mapping from gene expression to classifier outputs. The resulting 13.2M-parameter autoencoder is evaluated against the dense DAE using the same latent dimensionality and a shared downstream attribution workflow. The contribution is therefore a task-specific architectural and empirical comparison, rather than a claim of general superiority over pretrained single-cell models. scHFBiMamba combines a bidirectional Mamba autoencoder (BiMambaAE) with a classifier for cardiac disease-state prediction from snRNA-seq data. The autoencoder projects expression profiles into 32 learned latent tokens, applies a bidirectional Mamba2 layer, and maps the resulting representation to the 350-dimensional latent space used by the DAE baseline. This common dimensionality supports comparison of downstream classifiers, although it does not imply identical latent geometry. The study makes three contributions: âą A compact state-space autoencoder. BiMambaAE contains 13.2M parameters, approximately 14.6Ă14.6Ă fewer than the dense DAE, and is evaluated for downstream species, cell-type, and disease-state classification. âą Encoderâclassifier and classifier-only comparisons. Six encoderâclassifier configurations (M0âM5) and nine classifier variants on frozen BiMambaAE representations are evaluated, including graph convolutional networks (GCN) [20], graph attention networks (GAT) [21], and XGBoost [22]. The test set contains 38,275 nuclei from held-out biological samples, and pairwise differences in prediction correctness are assessed with McNemarâs test [23]. âą Attribution-based pathway evaluation. SHAP-derived DXGs from the six main configurations are compared with expression-based DEGs using enrichment against curated HF-related Gene Ontology keywords. 2 Methods 2.1 Dataset and preprocessing The analysis used the CVD dataset published by Ruz Jurado et al. [3, 24], comprising 272,080 cardiac nuclei from human hearts and mouse models: 144,677 nuclei from healthy human hearts, 40,205 from healthy mouse hearts, 41,689 human and 3,594 mouse nuclei with AS, 19,330 human and 7,046 mouse nuclei with HFrEF, and 1,712 human and 13,827 mouse nuclei with HFpEF. The source preprocessing workflow used CellRanger (v7.0.0) [25] with the hg38 and m10 reference genomes. Barcodes with <<300 or >>6000 detected genes, or >>5% mitochondrial content, were excluded. Normalization, scaling, and integration followed a Seurat (v4.3.0) workflow [26]. Mapping mouse genes to human one-to-one orthologs yielded 16,545 input features. CellTypist annotations [27] were manually curated into seven cell types: cardiomyocytes (CM), endothelial cells (EC), fibroblasts (FB), immune cells (IC), neuronal cells (NC), pericytes (PC), and smooth muscle cells (SMC). Each nucleus has three categorical labels: species (2 classes), cell type (7 classes), and disease state (4 classes: CTRL, AS, HFrEF, HFpEF). These labels are concatenated into a 13-dimensional multi-hot target vector. Partitioning is performed at the biological-sample level, with joint stratification on the speciesĂcell-typeĂdisease label combination: training and validation follow a stratified 80/20 split over samples [3], and all nuclei from a held-out sample remain in the same partition. The test set comprises 38,275 nuclei from complete biological samples held out by species and disease state. Thus, test-set separation is defined at the biological-sample level, whereas the classification metrics reported here are calculated at the nucleus level. Test-set representations are computed with the frozen trained encoder; no part of the pipeline is fit on test-set expression. 2.2 Baseline: dense DAE + MLP (M0) The reference architecture of Ruz Jurado et al. [3] was reimplemented in PyTorch [28]. The dense DAE encoder maps 16,545 normalized expression values through layers of 5000, 2400, and 350 units with ReLU activations. An asymmetric decoder reconstructs the input through hidden layers of 2200 and 5150 units. The MLP maps the 350-dimensional representation to 13 output scores through hidden layers of 795, 230, and 105 units with dropout (p=0.1p=0.1); sigmoid activation yields output probabilities. This DAE+MLP configuration, denoted M0, is the replication baseline. 2.3 Bidirectional Mamba autoencoder (BiMambaAE) BiMambaAE (Figs. 2 and 1) replaces the dense autoencoder with a compact architecture incorporating state-space sequence mixing. The encoder comprises: (i) a linear stem projecting the input to 350 dimensions, followed by LayerNorm, GELU, and dropout (p=0.1p=0.1); (i) a linear expansion reshaped into 32 latent tokens with dimension dmodel=64d_model=64; (i) one bidirectional Mamba layer; and (iv) LayerNorm and a linear projection of the flattened tokens to a 350-dimensional latent vector. The decoder is a two-layer MLP (350â 350â 16545) with LayerNorm, GELU, and dropout. The latent-token bottleneck limits the sequence length processed by the SSM to 32, rather than 16,545. Sequence mixing therefore operates on a compact learned representation, while the input and reconstruction projections account for much of the parameter budget. These tokens are learned features and are not assigned to individual genes or validated gene programs. Bidirectional scans propagate information in both directions along the token sequence; they do not make the representation permutation-invariant or confer a biological interpretation on token order. The bidirectional Mamba layer applies an initial Conv1d operation with kernel size 3, followed by forward and backward Mamba2 blocks [10] (dstate=16d_state=16, dconv=4d_conv=4, expansion factor 2). The backward block processes the reversed sequence, and its output is reversed back to the original token order before summation (flipâprocessâflip), so that corresponding positions are aligned. A final convolution and an intra-layer residual connection link the initial convolution input to the final convolution output. This architecture enables input-dependent mixing among latent tokens in both directions. The complete autoencoder (encoder and decoder) contains 13,242,815 parameters (encoder 7,311,970; decoder 5,930,845), approximately 14.6Ă14.6Ă fewer than the dense DAE (192,903,395); the deployed inference path (BiMambaAE encoder plus MLP head, decoder discarded) contains 7,799,728 parameters, and the complete M3 model contains 13,730,573 (Supplementary File S3). Both autoencoders use mean-squared-error reconstruction loss; no input corruption (masking or noise) was applied during autoencoder training, and the âdenoisingâ designation is retained from the source study. The BiMambaAE decoder ends in a linear (identity-activation) output layer. However, their decoders and optimization settings also differ, so the comparison does not isolate the effect of the Mamba block alone. Figure 1: Architecture of the two main components. Left: the BiMambaAE autoencoder: a linear stem projects the 16,545-gene input to 350 dimensions, reshaped into 32 latent tokens of dimension 64; a bidirectional Mamba2 layer mixes the token sequence (the backward block processes the reversed sequence and its output is reversed back before summation); a projection maps the flattened tokens to the 350-dimensional latent vector; a two-layer MLP decoder reconstructs the input through a linear output layer. Right: the BiMamba classifier head: each latent dimension is embedded as a token with a learned positional embedding, processed by three bidirectional layers of two blocks each with a global residual connection, then mean- and max-pooled and mapped through a 128-unit head to 13 outputs (2 species, 7 cell types, 4 disease states). 2.4 Classifier heads Three classifier heads were evaluated on each encoderâs 350-dimensional representation: (i) the reference MLP (795â230â105â13); (i) a bidirectional Mamba classifier that embeds each latent dimension as a token using a linear projection and a learned positional embedding, applies three bidirectional Mamba layers of two blocks each with a global residual connection, and concatenates mean and max pooling before a 128-unit classification head; and (i) a unidirectional Mamba variant. Each of the 13 outputs is a raw logit trained with BCEWithLogitsLoss; at inference, predictions are formed by argmax within each category group, so exactly one label per category (species, cell type, disease state) is assigned per nucleus. The separate classifier ablation used frozen autoencoder representations and additionally included XGBoost (three multiclass models, one per target category; 300 trees, maximum depth 6, learning rate 0.1) [22]; a two-layer GCN and a two-layer GAT (hidden dimension 64, two attention heads) [20, 21] operating transductively on a Euclidean k-nearest-neighbor graph (k=15k=15) built on the pooled train/validation/test latents, with test labels masked: the graph loss is computed on training nodes, model selection uses validation nodes, and test nodes receive no supervision; and BiMamba variants using class-weighted binary cross-entropy, attention pooling, or both. 2.5 Training protocol Training was sequential, not joint: each autoencoder was first trained with mean-squared-error (MSE) reconstruction loss, then frozen; its latents were cached for the full dataset, and the classifier was trained separately on the cached latents. Neural classifiers used binary cross-entropy (BCE) with logits, with per-class positive weights where indicated. The dense DAE and neural classifiers used Adam [29] (learning rate 10â310^-3, batch size 1024, maximum 500 epochs); BiMambaAE used AdamW [30] (learning rate 10â410^-4, weight decay 10â610^-6, batch size 32, maximum 100 epochs). Neural-network runs used ReduceLROnPlateau scheduling (factor 0.1, patience 25, minimum learning rate 10â710^-7), early stopping (patience 50), automatic mixed precision, and a fixed seed (1234). The six main configurations were M0, DAE+MLP; M1, DAE+Mamba (unidirectional); M2, DAE+BiMamba; M3, BiMambaAE+MLP; M4, BiMambaAE+Mamba; and M5, BiMambaAE+BiMamba. No training wall-clock records were retained in the results bundle, so no training-speed claims are made. 2.6 Attribution, DXG, DEG, and pathway evaluation Gene-level attributions were calculated using SHAP GradientExplainer [4] on the composite encoderâclassifier model. GradientExplainer uses expected gradients to approximate SHAP values; its outputs are not exact Shapley values. The composite modelâs outputs are sigmoid probabilities, so attributions refer to the probability scale. The attribution procedure used a background of 1000 training cells, balanced sampling of 200 test cells per cell type, nsamples=100, and seed 1111; mitochondrial-gene attributions were zeroed post hoc. Differentiability of both the encoder and classifier permits attribution of classifier outputs to the original gene-expression inputs. This property motivated the use of the MLP as the default classifier in the DXG workflow. Following Ruz Jurado et al. [3], attributions were Z-transformed within each cell and compared between disease and control groups within each speciesĂcell-type stratum using Studentâs t-test. Multiple testing was controlled using BenjaminiâHochberg (BH) false discovery rate (FDR) adjustment [31]. DXGs were filtered using adjusted p<0.05p<0.05, SHAP magnitude â„0.01â„ 0.01, cube-root fold-change â„0.1â„ 0.1, and mean expression â„0.1â„ 0.1. DEGs were identified using the Wilcoxon rank-sum test on expression values, with adjusted p<0.05p<0.05, |log2âĄFC|â„0.25| _2FC|â„ 0.25, and detection in â„10%â„ 10\% of cells in either group. Preranked gene set enrichment analysis (GSEA) [32] was performed with GSEApy [33] against GO Biological Process 2023 [34], using 1000 permutations and gene-set sizes of 10â500, with significance at FDR qâ€0.05q†0.05. Genes were ranked by signâĄ(FC)â log10âĄ(|FC|/pbin+1)sign(FC)· _10(|FC|/p_bin+1), where pbin=0.1p_bin=0.1 if the BH-adjusted p<0.05p<0.05 and pbin=1p_bin=1 otherwise. Significant terms were matched case-insensitively (substring matching) to the curated, cell-type-specific reference keywords in Supplementary Table 8 of the source study. For each contrast and method, true positives (TP) are matched significant terms, false positives (FP) are unmatched significant terms, and false negatives (FN) are reference keywords not recovered among the significant terms; the per-contrast F1 is 2âTP/(2âTP+FP+FN)2\,TP/(2\,TP+FP+FN). This evaluation measures recovery of a curated keyword-based benchmark, rather than exhaustive biological ground truth. All 42 possible speciesĂcell-typeĂcondition contrasts, their availability flags, and per-model F1 values are listed in Supplementary File S1. 2.7 Statistical comparison and efficiency benchmarking Two aggregate F1 statistics are reported side by side throughout. The pooled-13 macro-F1 is the macro average over all 13 outputs, weighting each output class equally; the mean-category F1 is the arithmetic mean of the three category macro-F1 scores (cell type, species, disease state). The two differ because the categories contain 7, 2, and 4 classes; both are recomputed at full precision in Supplementary File S2. Test-set predictions were compared pairwise using continuity-corrected McNemar tests [23] for species, cell type, and disease state, with BH adjustment across 45 comparisons [31]. These tests assess paired differences in nucleus-level prediction correctness, not differences in macro-F1, and treat nuclei as independent; within-sample dependence is acknowledged in the Limitations. Per-class and macro-averaged receiver operating characteristic (ROC) curves and areas under the ROC and precisionârecall curves were calculated from prediction scores. Inference time was summarized as mean± over 50 batches of 1024 cells after 10 warm-up batches, with device synchronization around each timed GPU iteration. Parameter counts and estimated floating-point operations (FLOPs) were also recorded. FLOP conventions are stated in the Table 4 caption: dense layers are estimated as 2Ă2Ăparameters, and SSMs as 2ĂparamsĂsequence length2ĂparamsĂsequence length (32 latent tokens for BiMambaAE, 350 for the SSM classifier heads); graph-network estimates depend on the number of edges. XGBoost was benchmarked on CPU, with tree-node counts reported in place of neural-network parameters. These measurements characterize the reported implementations and execution settings rather than a hardware-independent ranking. 3 Results 3.1 Replication of the DAE+MLP baseline The PyTorch implementation of the reference architecture (M0) achieved performance broadly consistent with the source study [3]. On the held-out test set of 38,275 nuclei, the pooled-13 aggregate F1 was 0.957 (mean-category F1 0.970), with category-level F1 values of 0.933 for cell type, 0.9996 for species, and 0.978 for disease state (Table 1, Fig. 3). The source study reported a validation F1 of â 0.95. Because these results concern different evaluation partitions, their agreement supports the use of M0 as a replication comparator but does not establish exact reproducibility. 3.2 BiMambaAE improves classification with a smaller autoencoder Replacing the dense DAE with BiMambaAE while retaining the MLP head (M3) produced the highest aggregate F1 among the six main configurations under both definitions: 0.966 pooled-13 and 0.977 mean-category, with category-level F1 values of 0.946 for cell type, 1.000 for species, and 0.983 for disease state. The corresponding macro ROC-AUC values were 0.9965, 1.0000, and 0.9981 (Table 1, Figs. 3 and 5). M3 was the only configuration with significantly higher nucleus-level classification accuracy than M0 in all three categories: BH-adjusted McNemar p=7.1Ă10â3p=7.1Ă 10^-3 for cell type, p=4.8Ă10â6p=4.8Ă 10^-6 for disease state, and p=7.4Ă10â4p=7.4Ă 10^-4 for species. The remaining configurations had lower cell-type and disease-state accuracy than M0 (all adjusted pâ€1.2Ă10â3p†1.2Ă 10^-3; Table 1). With the dense DAE, replacing the MLP with Mamba heads (M1 and M2) did not improve performance. With BiMambaAE, the Mamba heads (M4 and M5) also underperformed the MLP (M3 vs. M4/M5: adjusted pâ€6.1Ă10â33p†6.1Ă 10^-33 for cell type and pâ€1.0Ă10â50p†1.0Ă 10^-50 for disease state). These comparisons favor the BiMambaAE+MLP configuration under the reported training settings; they do not isolate architecture from optimization effects. The parameter comparison favors BiMambaAE (Fig. 4). The complete autoencoder contains 13.2M parameters, compared with 192.9M for the dense DAE, corresponding to a 14.6Ă14.6Ă reduction. Including the classifier, M3 contains â 13.7M parameters and achieves higher classification performance than the â 193M-parameter DAE-based configurations. Table 1: Encoderâclassifier comparison on the held-out test set of 38,275 nuclei. Category-level macro-F1 and both aggregate F1 statistics are shown: âF1 mean (13)â is the macro-F1 pooled over all 13 outputs (each class weighted equally); âF1 mean (cat.)â is the arithmetic mean of the three category macro-F1 scores. Both aggregates are computed at full precision from the stored per-class scores (Supplementary File S2); displayed category values are rounded, so their arithmetic mean can differ in the last digit. The highest value in each column is in bold; M0 comparator values are underlined where shown. BH-adjusted McNemar tests compare nucleus-level prediction correctness, not F1: M3 exceeds M0 in all three categories (cell type p=7.1Ă10â3p=7.1Ă 10^-3, disease p=4.8Ă10â6p=4.8Ă 10^-6, species p=7.4Ă10â4p=7.4Ă 10^-4). M1, M2, M4, and M5 have lower cell-type and disease-state accuracy than M0 (all adjusted pâ€1.2Ă10â3p†1.2Ă 10^-3); species differences are not significant for M4/M5. Model Encoder Classifier F1 cell type F1 species F1 disease F1 mean (13) F1 mean (cat.) M0 Dense DAE MLP [3] 0.9329 0.9996 0.9780 0.9570 0.9702 M1 Dense DAE Mamba (uni) 0.9240 0.9994 0.9720 0.9504 0.9652 M2 Dense DAE BiMamba 0.9246 0.9990 0.9667 0.9490 0.9634 M3 BiMambaAE MLP 0.9463 1.0000 0.9833 0.9660 0.9765 M4 BiMambaAE Mamba (uni) 0.9078 0.9998 0.9674 0.9403 0.9584 M5 BiMambaAE BiMamba 0.9094 0.9997 0.9613 0.9393 0.9568 Figure 2: scHFBiMamba workflow. Normalized snRNA-seq profiles (16,545 orthologous genes; human and mouse nuclei across CTRL, AS, HFrEF, and HFpEF) are compressed by either the dense DAE baseline or the proposed BiMambaAE into a 350-dimensional latent space. A classifier head predicts 13 outputs spanning species, cell type, and disease state. SHAP attributions through the composite model feed differential explanation (DXG) analysis and GSEA-based pathway evaluation against curated HF-related GO terms. Figure 3: Macro-F1 per category (cell type, species, disease) for the six ablation models on the held-out test set. M3 (BiMambaAE+MLP) achieves the best cell-type (0.946) and disease (0.983) F1 and perfect species classification. Figure 4: Classification performance versus model size. Total parameter count (log scale) is plotted against both aggregate F1 statistics for the six configurations (filled markers: pooled-13 macro-F1; open markers: mean-category F1). BiMambaAE configurations use a 13.2M-parameter autoencoder (orange shading), compared with 192.9M for the dense DAE (blue shading). M3 has the highest aggregate F1 under both definitions among these configurations. Figure 5: Macro-averaged ROC curves on the held-out test set for cell-type (left) and disease-state (right) classification. M3 attains the highest macro ROC-AUC in both categories (0.9965 and 0.9981, respectively). 3.3 Classifier ablation on frozen autoencoder latent spaces Nine classifier variants were trained on frozen DAE and BiMambaAE representations (Table 2, Table 3). The provenance of every row is as follows. In Table 2 (frozen DAE representations), the Mamba and BiMamba (plain BCE) rows are the M1 and M2 main-experiment configurations, reused for k=1k=1, seed 1234 (same encoder, same train split, same seed); the remaining rows come from the dedicated classifier-ablation run on DAE latents. In Table 3 (frozen BiMambaAE representations), the MLP, Mamba, and BiMamba (plain BCE) rows are the M3, M4, and M5 main-experiment configurations, reused under the same conditions; the remaining rows come from the dedicated run on BiMambaAE latents. Because M3 shares the same frozen encoder and sequential training protocol across both experiments, its scores are common to the two tables. An earlier version of the classifier comparison evaluated the BiMambaAE-trained SSM checkpoints on DAE latents (a checkpointâlatent mismatch). That run collapsed to degenerate majority-class predictions (mean F1 â 0.11, with bit-identical species F1 across arms, the signature of a constant majority-class predictor); it is disclosed here because it motivated the per-latent-space presentation. Class imbalance alone is therefore not established as the explanation for the collapsed run. With DAE embeddings, XGBoost achieved the highest mean-category F1 (0.972), followed closely by the MLP (0.970); both exceeded the transductive graph classifiers (GCN 0.913, GAT 0.897). Table 3 compares classifiers trained on frozen 350-dimensional BiMamba-AE representations. A simple MLP achieved the best performance across all tasks (mean-category F1 =0.9765=0.9765), followed by XGBoost (0.97210.9721) and the kkNN-graph models GCN (0.96710.9671) and GAT (0.96500.9650). Species classification was nearly saturated for all models (F1 â„0.9994â„ 0.9994), so differences were driven mainly by cell type (F1 0.89810.8981â0.94630.9463) and disease. Sequence-based heads (Mamba, BiMamba) performed slightly worse (mean-category F1 0.94890.9489â0.96040.9604), and attention did not help. The weighted loss improved BiMamba on cell type (0.9094â0.92200.9094â 0.9220), but the gains remained below the MLP. These results indicate that the frozen embeddings are highly linearly and non-linearly separable, so lightweight classifiers suffice. These results identify sensitivity to the training and aggregation choices in this experiment; they do not establish label imbalance as the sole cause of the collapsed run or imply that all SSM classifiers require the same remedies. The MLP also offered low classifier-only inference latency in the reported benchmark (Table 4): 0.154 ms per 1024-cell batch, compared with 192 ms for BiMamba and 51.3 ms for CPU XGBoost, corresponding to approximate ratios of 1250Ă and 330Ă, respectively. BiMambaAE was about 2.2Ă faster than the dense DAE (3.23±0.95 vs. 7.01±0.04 ms) and used about 14.6Ă14.6Ă fewer parameters (13.2M vs. 192.9M). Its estimated computational cost was also lower: 31.45M FLOPs per nucleus, because the SSM processes only the 32 latent tokens rather than the 16,545 genes, compared with 385.81M for the dense DAE (2Ă2Ăparameters). This is consistent with the 14.6Ă14.6Ă parameter reduction and the observed 2.2Ă speedup. The dense DAE showed lower variability in latency (±0.04 vs. ±0.95 ms). These timings exclude the autoencoder, and the XGBoost comparison involves a different execution device. The MLP additionally supports gradient propagation through the encoder to the original gene inputs. Tree ensembles can be explained using other attribution methods, but they do not provide this differentiable composite path. The MLP was therefore selected as the default head for the DXG workflow. Table 2: Classifier comparison using frozen 350-dimensional DAE representations. Category-level macro-F1 and the mean-category F1 (arithmetic mean of the three category macro-F1 scores) are shown for the held-out test set (single run, seed 1234). The pooled-13 macro-F1 for the main configurations (M0âM2) is reported in Table 1. The highest value in each column is in bold. Classifier F1 cell type F1 species F1 disease Mean cat. F1 XGBoost 0.9381 0.9994 0.9787 0.9721 MLP 0.9362 0.9996 0.9748 0.9702 GCN (kkNN graph) 0.8094 0.9993 0.9309 0.9132 GAT (kkNN graph) 0.7721 0.9994 0.9194 0.8970 Mamba (plain BCE) 0.9240 0.9994 0.9720 0.9504 BiMamba (plain BCE) 0.9246 0.9990 0.9667 0.9490 BiMamba + attention 0.9229 0.9991 0.9649 0.9623 BiMamba + weighted loss 0.8915 0.9987 0.9634 0.9512 BiMamba + weighted + attention 0.8841 0.9983 0.9578 0.9468 Table 3: Classifier comparison using frozen 350-dimensional BiMamba-AE representations. Category-level macro-F1 and the mean-category F1 (arithmetic mean of the three category macro-F1 scores) are shown for the held-out test set (single run, seed 1234). The pooled-13 macro-F1 for the main configurations (M3âM5) is reported in Table 1. The highest value in each column is in bold. Classifier F1 cell type F1 species F1 disease Mean cat. F1 MLP 0.9463 1.0000 0.9833 0.9765 XGBoost 0.9410 0.9998 0.9755 0.9721 GCN (kkNN graph) 0.9276 1.0000 0.9738 0.9671 GAT (kkNN graph) 0.9247 1.0000 0.9704 0.9650 Mamba (plain BCE) 0.9078 0.9998 0.9674 0.9584 BiMamba (plain BCE) 0.9094 0.9997 0.9613 0.9568 BiMamba + attention 0.8981 0.9994 0.9489 0.9489 BiMamba + weighted loss 0.9220 0.9996 0.9595 0.9604 BiMamba + weighted + attention 0.8989 0.9994 0.9500 0.9494 Table 4: Inference efficiency and model complexity. Inference time is mean ± SD per 1024-cell batch (50 batches after 10 warm-up, device-synchronized GPU iterations). XGBoost was benchmarked on CPU (parameter count reports total tree nodes; FLOPs not meaningful). FLOP conventions: dense layers estimated as 2Ă2Ăparameters; SSMs as 2Ă2ĂparametersĂsequence length (32 latent tokens for BiMambaAE, 350 for the SSM classifier heads); GCN/GAT FLOPs scale with graph edges (full-graph transductive pass normalized per 1024 cells). GCN/GAT parameter counts refer to the trained hidden-64 configuration (23,309 and 23,386 parameters); their FLOPs and timings were measured with the hidden-128 benchmark configuration (46,605 and 46,746 parameters), a mismatch disclosed here and in the Limitations. Component Inference time (ms) Parameters FLOPs Autoencoders (encoder and decoder) Dense DAE 7.01± 0.04 192.9M 385.81M BiMambaAE 3.23± 0.95 13.2M 31.45M Classifiers (350-d latent input) MLP 0.15 ± 0.02 487.6K 0.98M Mamba 98.04 ± 0.23 258.5K 181.0M BiMamba 192.01 ± 0.21 424.7K 297.3M GCN 0.07 ± 0.00 23.3K 26.2B GAT 0.78 ± 0.00 23.4K 25.4B XGBoost (CPU) 51.30 ± 1.90 286.6K â 3.4 Latent-space structure UMAP projections [35] of the test-set representations (Fig. 6) suggest that both autoencoders separate major cell types and retain some within-cell-type structure associated with disease state. This qualitative visualization is consistent with the classification results, but does not establish preservation of the full latent geometry or quantify biological fidelity. Figure 6: UMAP projections of the 350-dimensional test-set representations. Left: dense DAE (M0). Right: BiMambaAE (M3). Top: nuclei colored by cell type. Bottom: nuclei colored by disease state. Both representations show cell-type structure despite the approximately 14.6Ă14.6Ă difference in total autoencoder parameter count. 3.5 DXG-based enrichment shows higher average agreement with the curated pathway reference SHAP analysis identified genes contributing to disease-state predictions. For HFrEF, highly attributed genes in M3 included cardiomyopathy- and remodeling-related loci such as MYH7, TTN, RYR2, ANKRD1, ACTC1, MYL2, FHL2, LAMA2, and PDE3A (Fig. 7), consistent with the reference study [3]. HFpEF attributions highlighted a different set, including MALAT1, NEAT1, KCND2, CAMK1D, and PRKN. DXG analysis yielded 51,127 geneâcontrast calls for M3 across 33 speciesĂcell-typeĂcondition contrasts, compared with 37,572 for M0. These totals count calls across contrasts rather than unique genes. For human cardiomyocytes in HFrEF vs. CTRL, 1820 DXGs and 3576 DEGs shared 769 genes (Fig. 8). DXG-exclusive genes included DMD, SORBS2, MLIP, PLCL1, and ANK3. These attributions describe model behavior and do not establish causal involvement in disease. Against the curated GO keyword reference (Fig. 9), DXG-based GSEA achieved approximately twice the mean per-contrast F1 of DEG-based GSEA: 0.105 for M3-DXG over its 33 available contrasts vs. 0.053 for DEG over its 37 available contrasts. A sensitivity analysis restricted to the 31 contrasts available to both methods preserves the direction of the difference (0.112 vs. 0.051). Mean scores ranged from 0.095â0.108 for the three BiMambaAE configurations and from 0.076â0.086 for the DAE-based configurations. Examples favoring DXGs included mouse CM in HFpEF (F1 0.385 vs. 0.177), mouse IC in AS (0.500 vs. 0.000), mouse EC in AS (0.400 vs. 0.000), human FB in HFrEF (0.267 vs. 0.066), and human NC in HFrEF (0.200 vs. 0.000). The advantage was not uniform: DEGs achieved a higher score for human CM in HFrEF (0.310 vs. 0.091). The low absolute mean scores and variation across contrasts support interpreting DXGs as a complementary source of candidate pathways. Of the 42 possible contrasts, 39 were evaluable; the three unevaluable contrasts (human SMC in HFpEF, mouse NC in AS, mouse SMC in HFrEF) are flagged in Supplementary File S1. Figure 7: SHAP beeswarm plots for the 20 highest-ranked gene attributions to the HFrEF (left) and HFpEF (right) outputs of M3 (BiMambaAE+MLP). Point color represents normalized expression. Highly attributed HFrEF genes include MYH7, TTN, RYR2, ANKRD1, ACTC1, and MYL2; attribution magnitude reflects contribution to model predictions rather than experimental validation of disease mechanisms. Figure 8: Overlap between M3 DXGs (SHAP-based) and DEGs (expression-based Wilcoxon test) for human cardiomyocytes in HFrEF vs. CTRL. The sets contain 1051 DXG-exclusive genes, 2807 DEG-exclusive genes, and 769 shared genes. Figure 9: GSEA F1 scores against the curated, cell-type-specific GO keyword reference for HFpEF (left) and HFrEF (right). DXG-based rankings from each model are compared with DEG-based rankings. Scores describe agreement with the selected reference and vary across cell types and disease contrasts. 4 Discussion The results support the use of a compact bidirectional state-space autoencoder within this cardiac transcriptomic classification pipeline. BiMambaAE reduces the complete autoencoder parameter count by approximately 14.6Ă14.6Ă, while the BiMambaAE+MLP configuration improves the prediction metrics relative to the replicated dense DAE+MLP baseline. A possible explanation is that the compact latent-token representation provides a useful constraint on feature learning before input-dependent sequence mixing. However, the present comparisons do not distinguish this effect from changes in decoder capacity, optimization, batch size, or training duration. Dedicated component ablations under matched training conditions are needed to attribute the gains specifically to bidirectionality or state-space mixing. The reported species F1 of 1.000 and improvements in cell-type classification characterize this held-out test set and do not establish generalization to independent cohorts. Classifier choice remained consequential. MLP and XGBoost produced similar scores on frozen BiMambaAE representations, whereas the graph classifiers and unweighted BCE-trained SSM heads performed less well. The improved scores obtained with class weighting or attention pooling indicate that SSM-head performance is sensitive to training and aggregation choices in this setting. The MLP offers a practical combination of predictive performance, low measured classifier latency (0.15 ms/batch), and differentiability with respect to the original expression inputs. This final property enables the gradient-based attribution workflow used for DXG analysis. The pathway analysis extends the DXG evaluation of Ruz Jurado et al. [3] to a state-space autoencoder. The approximately twofold increase in average reference-based GSEA F1 relative to DEGs suggests that prediction-oriented attributions can prioritize biological information not captured by expression differences alone. Nevertheless, attribution identifies features used by a fitted model; it does not independently validate their mechanistic relevance. DXG-exclusive genes, including DMD in HFrEF cardiomyocytes, provide candidates for follow-up investigation. Because DEGs perform better in some contrasts, both analyses should be reported, and any benefit from combining their gene sets should be tested explicitly. Relationship to single-cell foundation models. Geneformer, UCE, and scFoundation target broadly transferable representations learned from large pretraining corpora [11, 13, 14]. scHFBiMamba instead evaluates a task-specific autoencoder within a defined cardiac classification and interpretation workflow. Its 13.2M-parameter autoencoder and differentiable classifier provide a compact implementation for this setting. No foundation-model baseline was evaluated, so the present results do not establish advantages in accuracy, transferability, computational cost, or attribution quality over pretrained alternatives. Future work could assess frozen and fine-tuned foundation-model embeddings under the same sample-level split and downstream evaluation protocol. Limitations. Experiments were restricted to one two-species dataset. Independent cohort validation, repeated training runs, and evaluation of sensitivity to data partitioning are needed to characterize robustness. Although test samples were held out at the biological-sample level, nucleus-level metrics and significance tests do not account for dependence among nuclei from the same donor or animal. Sample-aware uncertainty estimates would therefore strengthen inference. The reported DXG and DEG tests also require careful interpretation because their observational units are cells rather than independent biological replicates. Further limitations concern the keyword-based enrichment reference, implementation-specific efficiency measurements, and incomplete isolation of architectural effects. External validation could include an independent dilated-cardiomyopathy atlas such as that of Koenig et al. [2], provided that source-sample overlap is excluded. Most configurations were trained with one seed and one training/validation split, and the classifier comparison reports single runs; its rows derive from different runs and latent spaces, as disclosed in the Results, which limits direct comparability across tables. The GSEA evaluation depends on curated reference keywords, produces modest absolute F1 values, and favors DXGs inconsistently across contrasts. The current BiMambaAE implementation depends on CUDA kernels from mamba_ssm, limiting CPU-only deployment. Token count, layer depth, and state size were selected heuristically and warrant systematic sensitivity analysis. The GCN/GAT FLOPs and timings were measured with a hidden-128 benchmark configuration rather than the trained hidden-64 models (Table 4), and benchmark hardware and software versions are not archived with the results bundle. Comparisons with pretrained single-cell foundation models remain future work. 5 Conclusion scHFBiMamba provides a compact autoencoder-based framework for classifying cardiac nuclei and analyzing gene-level model attributions. In the reported experiments, BiMambaAE with an MLP head achieved higher classification scores than the dense DAE+MLP comparator while reducing the total autoencoder parameter count by approximately 14.6Ă14.6Ă. DXG-derived rankings showed higher average agreement with a curated HF-related pathway reference than DEG-derived rankings, with substantial variation across contrasts. These findings motivate further evaluation using matched training controls, sample-aware statistical analysis, and independent cohorts. Data and code availability The CVD dataset is publicly available at https://huggingface.co/datasets/mruzjurado/CVD_dataset [24]. The scHFBiMamba code is available at https://github.com/UgoLomoio/scHFBiMamba. Acknowledgments The authors thank Ruz Jurado et al. for making the CVD dataset and analysis code publicly available. This work was supported by the OFIDIAPlus project, INTERREG GREECEâITALY 2021â2027 PROGRAMME. All scientific content, results, and interpretations were verified, modified and approved by the authors. Declaration of interests The authors declare that they have no known competing financial interests or personal relationships that could have appeared to influence the work reported in this paper. Ethics statement This study did not require ethical approval because it used a publicly available, de-identified dataset and did not involve direct interaction with human subjects. Supplementary Material Figure 10: Supplementary Figure S1: Ablation heatmap across the six configurations (rows) and ten performance metrics (columns): macro-F1 per category, both aggregate F1 statistics (âF1 mean (13 outputs)â and âF1 mean (categories)â), mean precision and recall, and macro ROC-AUC per category, all on the held-out test set. Figure 11: Supplementary Figure S2: Training and validation curves for the replicated baseline M0 (left) and the proposed M3 (BiMambaAE+MLP, right). References [1] G. Savarese, P. M. Becher, L. H. Lund, P. Seferovic, G. M. C. Rosano, and A. J. S. Coats (2023) Global burden of heart failure: a comprehensive and updated review of epidemiology. Nature Reviews Cardiology 20 (2), p. 122â134. External Links: Document Cited by: §1. [2] A. L. Koenig, I. Shchukina, J. Amrute, P. S. Andhey, K. Zaitsev, L. Lai, G. Bajpai, A. Bredemeyer, G. Smith, C. Jones, E. Terrebonne, S. L. Rentschler, M. N. Artyomov, and K. J. Lavine (2022) Single-cell transcriptomics reveals cell-type-specific diversification in human heart failure. Nature Cardiovascular Research 1 (3), p. 263â280. External Links: Document Cited by: §1, §4. [3] M. Ruz Jurado, D. Rodriguez Morales, E. Genetzakis, F. B. Ardakani, L. Zanders, A. Fischer, F. Buettner, M. H. Schulz, S. Dimmeler, and D. John (2025) Decoding heart failure subtypes with neural networks via differential explanation analysis. Briefings in Bioinformatics 26 (6), p. bbaf581. External Links: ISSN 1477-4054, Document, Link Cited by: §1, §1, §2.1, §2.1, §2.2, §2.6, §3.1, §3.5, Table 1, §4. [4] S. M. Lundberg and S. Lee (2017) A unified approach to interpreting model predictions. In Advances in Neural Information Processing Systems 30 (NeurIPS), p. 4765â4774. Cited by: §1, §2.6. [5] U. Lomoio, P. Veltri, P. H. Guzzi, and P. LiĂČ (2025) Design and use of a denoising convolutional autoencoder for reconstructing electrocardiogram signals at super resolution. Artificial Intelligence in Medicine 160, p. 103058. Cited by: §1. [6] U. Lomoio, P. LiĂł, P. H. Guzzi, and P. Veltri (2026) Bidirectional mamba-2 boosts eeg super-resolution via regression and diffusion. Bioinformatics, p. btag169. Cited by: §1. [7] A. Vaswani, N. Shazeer, N. Parmar, J. Uszkoreit, L. Jones, A. N. Gomez, Ć. Kaiser, and I. Polosukhin (2017) Attention is all you need. In Advances in Neural Information Processing Systems 30 (NeurIPS), p. 5998â6008. Cited by: §1. [8] H. Cui, C. Wang, H. Maan, K. Pang, F. Luo, N. Duan, and B. Wang (2024) scGPT: toward building a foundation model for single-cell multi-omics using generative AI. Nature Methods 21 (8), p. 1470â1480. External Links: Document Cited by: §1. [9] A. Gu and T. Dao (2024) Mamba: linear-time sequence modeling with selective state spaces. In Conference on Language Modeling (COLM), External Links: 2312.00752 Cited by: §1. [10] T. Dao and A. Gu (2024) Transformers are SSMs: generalized models and efficient algorithms through structured state space duality. In Proceedings of the 41st International Conference on Machine Learning (ICML), PMLR, Vol. 235. External Links: 2405.21060 Cited by: §1, §2.3. [11] C. V. Theodoris, L. Xiao, A. Chopra, M. D. Chaffin, Z. R. Al Sayed, M. C. Hill, H. Mantineo, E. M. Brydon, Z. Zeng, X. S. Liu, and P. T. Ellinor (2023) Transfer learning enables predictions in network biology. Nature 618 (7965), p. 616â624. External Links: Document Cited by: §1, §4. [12] F. Yang, W. Wang, F. Wang, Y. Fang, D. Tang, J. Huang, H. Lu, and J. Yao (2022) ScBERT as a large-scale pretrained deep language model for cell type annotation of single-cell RNA-seq data. Nature Machine Intelligence 4 (10), p. 852â866. External Links: Document Cited by: §1. [13] Y. Rosen, Y. Roohani, A. Agrawal, L. SamotorÄan, T. S. Consortium, S. R. Quake, and J. Leskovec (2026) Universal cell embedding provides a foundation model for cell biology. Nature. External Links: Document Cited by: §1, §4. [14] M. Hao, J. Gong, X. Zeng, C. Liu, Y. Guo, X. Cheng, T. Wang, J. Ma, X. Zhang, and L. Song (2024) Large-scale foundation model on single-cell transcriptomics. Nature Methods 21 (8), p. 1481â1491. External Links: Document Cited by: §1, §4. [15] R. Lopez, J. Regier, M. B. Cole, M. I. Jordan, and N. Yosef (2018) Deep generative modeling for single-cell transcriptomics. Nature Methods 15 (12), p. 1053â1058. External Links: Document Cited by: §1. [16] Y. Zhao, B. Zhao, F. Zhang, C. He, W. Wu, and L. Lai (2024) SC-MAMBA2: leveraging state-space models for efficient single-cell ultra-long transcriptome modeling. bioRxiv. External Links: Document Cited by: §1. [17] C. Qi, H. Fang, S. Jiang, et al. (2025) GeneMamba: an efficient and effective foundation model on single cell data. arXiv preprint arXiv:2504.16956. External Links: Document Cited by: §1. [18] Z. Yuan, S. Jiao, Y. Xiao, and J. Peng (2025) ScMamba: a scalable foundation model for single-cell multi-omics integration beyond highly variable feature selection. arXiv preprint arXiv:2506.20697. External Links: Document Cited by: §1. [19] L. Hu, H. Qin, Y. Zhang, Y. Lu, P. Qiu, Z. Guo, L. Cao, and W. Jiang (2026) RegFormer: a single-cell foundation model powered by gene regulatory hierarchies. Nature Communications. External Links: Document Cited by: §1. [20] T. N. Kipf and M. Welling (2017) Semi-supervised classification with graph convolutional networks. In International Conference on Learning Representations (ICLR), External Links: 1609.02907 Cited by: 2nd item, §2.4. [21] P. VeliÄkoviÄ, G. Cucurull, A. Casanova, A. Romero, P. LiĂČ, and Y. Bengio (2018) Graph attention networks. In International Conference on Learning Representations (ICLR), External Links: 1710.10903 Cited by: 2nd item, §2.4. [22] T. Chen and C. Guestrin (2016) XGBoost: a scalable tree boosting system. In Proceedings of the 22nd ACM SIGKDD International Conference on Knowledge Discovery and Data Mining, p. 785â794. External Links: Document Cited by: 2nd item, §2.4. [23] Q. McNemar (1947) Note on the sampling error of the difference between correlated proportions or percentages. Psychometrika 12 (2), p. 153â157. External Links: Document Cited by: 2nd item, §2.7. [24] M. Ruz Jurado (2025) CVD_dataset: data sets and labels for âdecoding heart failure subtypes with neural networks via differential explanation analysisâ. Note: https://huggingface.co/datasets/mruzjurado/CVD_datasetHugging Face dataset Cited by: §2.1, Data and code availability. [25] G. X. Y. Zheng, J. M. Terry, P. Belgrader, P. Ryvkin, Z. W. Bent, R. Wilson, S. B. Ziraldo, T. D. Wheeler, G. P. McDermott, J. Zhu, M. T. Gregory, J. Shuga, L. Montesclaros, J. G. Underwood, D. A. Masquelier, S. Y. Nishimura, M. Schnall-Levin, P. W. Wyatt, C. M. Hindson, R. Bharadwaj, A. Wong, K. D. Ness, L. W. Beppu, H. J. Deeg, C. McFarland, K. R. Loeb, W. J. Valente, N. G. Ericson, E. A. Stevens, J. P. Radich, T. S. Mikkelsen, B. J. Hindson, and J. H. Bielas (2017) Massively parallel digital transcriptional profiling of single cells. Nature Communications 8, p. 14049. External Links: Document Cited by: §2.1. [26] Y. Hao, S. Hao, E. Andersen-Nissen, W. M. Mauck, S. Zheng, A. Butler, M. J. Lee, A. J. Wilk, C. Darby, M. Zager, P. Hoffman, M. Stoeckius, E. Papalexi, E. P. Mimitou, J. Jain, A. Srivastava, T. Stuart, L. M. Fleming, B. Yeung, A. J. Rogers, J. M. McElrath, C. A. Blish, R. Gottardo, P. Smibert, and R. Satija (2021) Integrated analysis of multimodal single-cell data. Cell 184 (13), p. 3573â3587.e29. External Links: Document Cited by: §2.1. [27] C. DomĂnguez Conde, C. Xu, L. B. Jarvis, D. B. Rainbow, S. B. Wells, T. Gomes, S. K. Howlett, O. Suchanek, K. Polanski, H. W. King, et al. (2022) Cross-tissue immune cell analysis reveals tissue-specific features in humans. Science 376 (6594), p. eabl5197. External Links: Document Cited by: §2.1. [28] A. Paszke, S. Gross, F. Massa, A. Lerer, J. Bradbury, G. Chanan, T. Killeen, Z. Lin, N. Gimelshein, L. Antiga, A. Desmaison, A. Kopf, E. Yang, Z. DeVito, M. Raison, A. Tejani, S. Chilamkurthy, B. Steiner, L. Fang, J. Bai, and S. Chintala (2019) PyTorch: an imperative style, high-performance deep learning library. In Advances in Neural Information Processing Systems 32 (NeurIPS), p. 8024â8035. Cited by: §2.2. [29] D. P. Kingma and J. Ba (2015) Adam: a method for stochastic optimization. In International Conference on Learning Representations (ICLR), External Links: 1412.6980 Cited by: §2.5. [30] I. Loshchilov and F. Hutter (2019) Decoupled weight decay regularization. In International Conference on Learning Representations (ICLR), External Links: 1711.05101 Cited by: §2.5. [31] Y. Benjamini and Y. Hochberg (1995) Controlling the false discovery rate: a practical and powerful approach to multiple testing. Journal of the Royal Statistical Society: Series B (Methodological) 57 (1), p. 289â300. External Links: Document Cited by: §2.6, §2.7. [32] A. Subramanian, P. Tamayo, V. K. Mootha, S. Mukherjee, B. L. Ebert, M. A. Gillette, A. Paulovich, S. L. Pomeroy, T. R. Golub, E. S. Lander, and J. P. Mesirov (2005) Gene set enrichment analysis: a knowledge-based approach for interpreting genome-wide expression profiles. Proceedings of the National Academy of Sciences 102 (43), p. 15545â15550. External Links: Document Cited by: §2.6. [33] Z. Fang, X. Liu, and G. Peltz (2023) GSEApy: a comprehensive package for performing gene set enrichment analysis in Python. Bioinformatics 39 (1), p. btac757. External Links: Document Cited by: §2.6. [34] S. A. Aleksander, J. Balhoff, S. Carbon, J. M. Cherry, H. J. Drabkin, D. Ebert, M. Feuermann, P. Gaudet, N. L. Harris, D. P. Hill, R. Lee, H. Mi, S. Moxon, C. J. Mungall, A. Muruganugan, T. Mushayahama, P. W. Sternberg, P. D. Thomas, K. Van Auken, J. Ramsey, D. A. Siegele, R. L. Chisholm, P. Fey, M. C. Aspromonte, M. V. Nugnes, F. Quaglia, S. Tosatto, M. Giglio, S. Nadendla, G. Antonazzo, H. Attrill, G. dos Santos, S. Marygold, V. Strelets, C. J. Tabone, J. Thurmond, P. Zhou, S. Ahmed, P. Asanitthong, D. Luna Buitrago, M. N. Erdol, M. C. Gage, M. Ali Kadhum, K. Y. Li, S. Long, A. Michishita, A. Pesala, A. Pritazahra, S. C. C. Saverimuttu, R. Su, K. E. Thurlow, R. C. Lovering, C. Logie, S. Oliferenko, J. Blake, K. Christie, L. Corbani, M. E. Dolan, H. J. Drabkin, D. P. Hill, L. Ni, D. Sitnikov, C. Smith, A. Cuzick, J. Seager, L. Cooper, J. Elser, P. Jaiswal, P. Gupta, P. Jaiswal, S. Naithani, M. Lera-Ramirez, K. Rutherford, V. Wood, J. De Pons, M. Dwinell, G. T. Hayman, M. L. Kaldunski, A. E. Kwitek, S. J. F. Laulederkind, M. A. Tutaj, M. Vedi, S. Wang, P. DâEustachio, L. Aimo, K. Axelsen, A. Bridge, N. Hyka-Nouspikel, A. Morgat, S. Aleksander, et al. (2023) The Gene Ontology knowledgebase in 2023. Genetics 224 (1), p. iyad031. External Links: Document Cited by: §2.6. [35] L. McInnes, J. Healy, and J. Melville (2018) UMAP: uniform manifold approximation and projection for dimension reduction. arXiv preprint arXiv:1802.03426. Cited by: §3.4.