# Long short-term memory-based deep learning model for the discovery of antimicrobial peptides targeting Mycobacterium tuberculosis

## TL;DR

This paper introduces a deep learning model using LSTM networks to discover and generate antimicrobial peptides effective against tuberculosis.

## Contribution

A novel LSTM-based deep learning protocol is proposed for TB-specific antimicrobial peptide discovery using transfer learning and generative modeling.

## Key findings

- A deep learning model achieved 90% accuracy and 0.97 AUC in classifying TB-active peptides.
- A generator trained on TB data produced 94 out of 100 predicted antimicrobial peptides.
- Four generated peptides showed ≥84% identity with known TB-AMPs.

## Abstract

Tuberculosis, caused by Mycobacterium tuberculosis, remains a global health challenge driven by rising antibiotic resistance. Antimicrobial peptides offer a promising alternative due to membrane-disruptive activity and low resistance potential, yet the scarcity of TB-specific AMP data constrains targeted development. We present a reproducible deep learning protocol that integrates long short-term memory networks with transfer learning to classify and generate TB-active peptides.

Classifiers were pretrained on a large corpus of general AMPs and fine-tuned on curated TB-specific sequences using frozen encoder and full backpropagation strategies. We benchmarked four model variants [unidirectional and bidirectional long short-term memories (LSTMs), with and without attention] on a held-out TB test set; the unidirectional LSTM with a frozen encoder achieved the best performance (accuracy 90%, AUC 0.97). In parallel, LSTM-based generative models were trained to produce de novo TB-active peptides. A generator trained exclusively on TB data produced 94 of 100 peptides predicted as antimicrobial by AMP Scanner, outperforming transfer learning-based generators. Generated peptides were evaluated for antimicrobial activity, toxicity, structure, and AMP-like physicochemical traits, and four candidates shared ≥84% identity with known TB-AMPs.

The complete model and data can be found at: https://github.com/linfeng-wang/TB-AMP-design.