Phylo-HS: A phylogenetic hierarchical softmax for taxonomic classification
Menegaux, R.
Show abstract
Taxonomic binning -assigning taxonomic labels to DNA sequencing reads - is a core component of metagenomics data analysis. While machine learning approaches offer competitive speed and accuracy, their scalability is hindered by the growing number of referenced genomes and species. A major bottleneck lies in the final softmax layer of neural network models, which computes probabilities and gradients for all outcome classes. To address this, we propose Phylo-HS, a hierarchical softmax method that leverages the taxonomic tree to group classes into meaningful clusters. Phylo-HS achieves an order of magnitude speed improvement on a dataset with 5,000 classes and improves classification accuracy compared to frequency-based hierarchical softmax methods. By integrating phylogenetic structure into the model, Phylo-HS effectively balances scalability and accuracy for large-scale metagenomic analysis.
Matching journals
The top 2 journals account for 50% of the predicted probability mass.
Similar papers in this journal
- Large-scale Inference of Cell Lineage Trees and Genotype Calling from Noisy Single-Cell Data Using Efficient Local Search 96%
- TREE-QMC: Improving quartet graph construction for scalable and accurate species tree estimation from gene trees 95%
- Haplotype-aware sequence alignment to pangenome graphs 94%
"Similar papers" are the closest papers from that journal in the model's embedding space. They show what the match is built on, but the ranking comes mostly from a classifier over the whole training set, not from these examples alone.