FuzzyCell: Differentiable Cell Ontology Reasoning for Graded Cell Type Annotation
Abstract
*How reliable are single-cell foundation models?* We audit scGPT, Geneformer, CellFM, and scCello on Tabula Sapiens and find that of predictions violate a hard biological constraint encoded in the Cell Ontology (): neurons placed in liver, cells assigned simultaneously to mutually exclusive lineages. These errors are invisible to accuracy and F1, yet they follow from elementary reasoning over the formal axioms of . The root cause is that annotation is treated as flat classification over a label set. Even methods that do use use only its graph: random walks, hierarchical classifiers, PageRank, post-hoc probability propagation, hierarchy-weighted losses. The axioms themselves, disjointness and role restrictions that state which cell types cannot coexist and where each type must reside, are discarded. We recast annotation as ontology-constrained entity linking. FuzzyCell retrieves candidate types with a pretrained encoder and re-ranks them with a differentiable reasoner that evaluates extracted axioms, including asserted disjointness axioms that propagate to effective pairs, inside the training loop. Scoring uses a sigmoidal implication in the family of van Krieken et al., which remains trainable where Goguen and Lukasiewicz implications produce zero gradient on positive-measure regions, and satisfies the fuzzy implication boundary conditions at any finite steepness . On Tabula Sapiens, HLCA, and AIDA v2, FuzzyCell reduces axiom violations by over and improves Macro-F1 by to points over the respective backbone, as a backbone-agnostic module adding under ms per cell. Gains concentrate in the long tail: median per-type F1 rises on common types and on rare types, and the improvement is recall-led, with macro-recall gaining points against for macro-precision. Ablations isolate the source: switching off the reasoner at fixed checkpoint changes of predictions, of which corrections outnumber degradations to ; removing tissue context costs Macro-F1 while removing costs ; removing of axioms at random costs . Beyond hard labels, FuzzyCell outputs graded memberships that track differentiation: along two bone marrow trajectories, terminal-type membership correlates with pseudotime at Spearman and .
est. 32% chance this paper gets accepted at ICLR 2027.
What do you think this paper will get?
All positions stay anonymous.