Hierarchical cross-entropy loss improves atlas-scale single-cell annotation models
Listed in
This article is not in any list yet, why not save it to one of your lists.Abstract
Accurately annotating cell types is essential for extracting biological insight from single-cell RNA-seq data. Although cell types are naturally organized into hierarchical ontologies, most computational models do not explicitly incorporate this structure into their training objectives. We introduce a hierarchical cross-entropy loss that aligns model objectives with biological structure. Applied to architectures ranging from linear models to transformers, this simple modification significantly improves out-of-distribution performance (12–15%) without added computational cost.
