HIV-1 Recombinant Classification

Attention-based Transformer encoder for HIV-1 recombinant/non-recombinant classification.

Model

  • Representation: 3-mer tokenization
  • Architecture: Transformer Encoder
  • Transformer layers: 2
  • Attention heads: 8
  • Embedding dimension: 128
  • Feed-forward dimension: 256
  • Maximum input length: 4,395 3-mers
  • Classes:
    • Non-Recombinant
    • Recombinant

Deployment Model

The deployment model uses post-training dynamic INT8 quantization of eligible Linear layers.

The model is intended for CPU-oriented inference.

The embedding and Transformer attention components are not fully quantized.

Source

The deployed model originates from:

Experiment 5 โ€” Best Transformer

The quantized model was generated as part of:

Experiment 6 โ€” Post-Training Dynamic INT8 Quantization

Reproducibility

Source FP32 checkpoint SHA256:

2a5641ef378603c6d9f2e98a1a4a29bc428e6774c92ffb80d39f1c90c01bbb83

Dynamic INT8 checkpoint SHA256:

69935da6113edfb1f6eded7ded6bb6b6219b173aa24325b0e26d8da43012b289

Train-only vocabulary SHA256:

1f626e456022bafa8547b925bdd03907fabcb1d3a8fc09353f008b63536ad0ae

Intended Use

This model is a research prototype for HIV-1 genomic surveillance and recombinant classification research.

It is not a clinical diagnostic tool and should not be used for patient diagnosis or clinical decision-making.

Files

  • model/Transformer_Encoder_best_DYNAMIC_INT8.pt
    • Deployment model
  • model/Transformer_Encoder_best.pt
    • Original FP32 source checkpoint
  • tokenizer/TRAIN_ONLY_3MER_VOCABULARY.csv
    • Training-only 3-mer vocabulary
  • metadata/model_config.json
    • Architecture and deployment metadata
  • metadata/file_manifest.json
    • File integrity manifest

Research Project

HIV-1 Subtype/Recombinant Classification

Attention-Based Deep Learning for HIV-1 Genomic Surveillance and Subtype Classification in Uganda.

Downloads last month

-

Downloads are not tracked for this model. How to track
Inference Providers NEW
This model isn't deployed by any Inference Provider. ๐Ÿ™‹ Ask for provider support