rustambekurokov/esm2-35m-protein-localization

ESM-2 35M — Protein Subcellular Localization (Full Fine-Tune)

Fine-tuned ESM-2 35M for protein subcellular localization (10 classes). All parameters trained end-to-end.

Results

Architecture

ESM-2 35M (all trainable) → Mean pooling → LayerNorm → Linear(480→120) → GELU → Linear(120→10)

Training

• —Dataset: DeepLoc 2.0 (17,266 train / 3,700 val / 3,701 test)
• —Strategy: Full fine-tune (all 35M parameters)
• —Epochs: 10
• —Learning rate: 2e-5
• —Hardware: NVIDIA DGX Spark

All Models in This Project

Links

• —Demo: HuggingFace Space
• —Code: GitHub