English

Mamba-SSM with LLM Reasoning for Feature Selection: Faithfulness-Aware Biomarker Discovery

Quantitative Methods 2026-04-20 v2 Artificial Intelligence

Abstract

Gradient saliency from deep sequence models surfaces candidate biomarkers efficiently, but the resulting gene lists can be contaminated by tissue-composition confounders that degrade downstream classifiers. We study whether LLM chain-of-thought (CoT) reasoning can filter these confounders, and whether reasoning quality is associated with downstream performance. We train a Mamba SSM on TCGA-BRCA RNA-seq and extract the top-50 genes by gradient saliency; DeepSeek-R1 evaluates every candidate with structured CoT to produce a final 17-gene set. On the held-out test split, the raw 50-gene saliency set (no LLM) performs worse than a 5,000-gene variance baseline (AUC 0.832 vs. 0.903), while the LLM-filtered set surpasses it (AUC 0.927), using 294x fewer features. A faithfulness audit (COSMIC CGC, OncoKB, PAM50) shows that 6 of 17 selected genes (35.3%) are validated BRCA biomarkers, while 10 of 16 known BRCA genes present in the input were missed - including FOXA1. This divergence between downstream performance and reasoning faithfulness suggests selective faithfulness in this setting: targeted confounder removal can improve predictive performance without comprehensive recall.

Keywords

Cite

@article{arxiv.2604.14334,
  title  = {Mamba-SSM with LLM Reasoning for Feature Selection: Faithfulness-Aware Biomarker Discovery},
  author = {Pushpa Kumar Balan and Aijing Feng},
  journal= {arXiv preprint arXiv:2604.14334},
  year   = {2026}
}

Comments

9 pages, 4 figures. Accepted at ICLR 2026 Workshop on Logical Reasoning of Large Language Models