Influ-BERT: An Interpretable Model for Enhancing Low-Frequency Influenza A virus Subtype Recognition
Ye, R.; Li, L.; Song, S.
Show abstract
Influenza A Virus (IAV) poses a persistent threat to global public health due to its broad host adaptability, frequent anti-genic variation, and potential for cross-species transmission. Accurate identification of IAV subtypes is essential for effective epidemic surveillance and precise disease control. Here, we present Influ-BERT, a domain-adaptive pretrained model based on the Transformer architecture. Optimized from DNABERT-2, Influ-BERT was developed using a dedicated corpus of approximately 900,000 influenza genome sequences. We constructed a custom Byte Pair Encoding (BPE) tokenizer, and employed a two-stage training strategy involving domain-adaptive pretraining followed by task-specific fine-tuning. This approach significantly enhanced identification performance for IAV subtypes. Experimental results demonstrate that Influ-BERT outperforms both traditional machine learning approaches and general genomic language models, such as DNABERT-2 and MegaDNA, in the task of IAV subtype identification. The model achieved F1-scores consistently above 97% and exhibited stable performance gains for subtypes that are underrepresented in sequencing data, including H5N8, H5N1, H7N9, and H9N2. Beyond subtype identification, Influ-BERT was successfully applied to additional tasks including respiratory virus identification, IAV pathogenicity prediction, and identification of IAV genomic fragments and functional genes, demonstrating robust performance throughout. Further interpretability analysis using sliding window perturbation confirmed that the model focuses on biologically significant genomic regions, providing insight into its improved predictive capability. Contactsongshh@big.ac.cn (Song S), atrv@lncc.br (Ana Tereza Ribeiro Vasconcelos)
Matching journals
The top 7 journals account for 50% of the predicted probability mass.
Similar papers in this journal
Similar papers in this journal
Similar papers in this journal
- Phylogenetic-informed graph deep learning to classify dynamic transmission clusters in infectious disease epidemics 94%
- Improving protein function prediction by learning and integrating representations of protein sequences and function labels 93%
- Nucleotide augmentation for machine learning-guided protein engineering 93%
Similar papers in this journal
- IPEV: Identification of Prokaryotic and Eukaryotic Virus-derived sequences in virome using deep learning 96%
- Profiling the baseline performance and limits of machine learning models for adaptive immune receptor repertoire classification 93%
- Knowledge Graph-based Thought: a knowledge graph enhanced LLMs framework for pan-cancer question answering 93%
"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.