Improve model card metadata and links
Browse filesHi! I'm Niels from the community science team at Hugging Face.
This PR improves the model card for the MiAE-L model by:
- Updating the `pipeline_tag` to `graph-ml` for better discoverability.
- Adding a link to the associated paper: [Protein Fold Classification at Scale: Benchmarking and Pretraining](https://huggingface.co/papers/2605.18552).
- Ensuring the GitHub repository link is present.
- Maintaining the existing `library_name` and usage instructions.
Feel free to merge if this looks good!
README.md
CHANGED
|
@@ -1,12 +1,12 @@
|
|
| 1 |
---
|
| 2 |
library_name: tedbench
|
| 3 |
-
tags:
|
| 4 |
-
- protein
|
| 5 |
-
- structure
|
| 6 |
-
- fold-classification
|
| 7 |
-
- tedbench
|
| 8 |
-
pipeline_tag: feature-extraction
|
| 9 |
license: bsd-3-clause
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 10 |
---
|
| 11 |
|
| 12 |
# TEDBench — Pretrained autoencoder (structure only)
|
|
@@ -15,10 +15,12 @@ license: bsd-3-clause
|
|
| 15 |
|
| 16 |
This is a **pretrained MiAE** checkpoint. Use it as a feature extractor or as the starting point for fine-tuning.
|
| 17 |
|
| 18 |
-
|
| 19 |
-
|
| 20 |
-
autoencoder that masks up to 90% of backbone frames and reconstructs the full
|
| 21 |
-
|
|
|
|
|
|
|
| 22 |
|
| 23 |
## Architecture sizes
|
| 24 |
|
|
@@ -59,4 +61,4 @@ model.eval()
|
|
| 59 |
booktitle={Proceedings of the 43rd International Conference on Machine Learning},
|
| 60 |
year={2026}
|
| 61 |
}
|
| 62 |
-
```
|
|
|
|
| 1 |
---
|
| 2 |
library_name: tedbench
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 3 |
license: bsd-3-clause
|
| 4 |
+
pipeline_tag: graph-ml
|
| 5 |
+
tags:
|
| 6 |
+
- protein
|
| 7 |
+
- structure
|
| 8 |
+
- fold-classification
|
| 9 |
+
- tedbench
|
| 10 |
---
|
| 11 |
|
| 12 |
# TEDBench — Pretrained autoencoder (structure only)
|
|
|
|
| 15 |
|
| 16 |
This is a **pretrained MiAE** checkpoint. Use it as a feature extractor or as the starting point for fine-tuning.
|
| 17 |
|
| 18 |
+
This model was presented in the paper [Protein Fold Classification at Scale: Benchmarking and Pretraining](https://huggingface.co/papers/2605.18552).
|
| 19 |
+
|
| 20 |
+
Part of the [TEDBench](https://github.com/BorgwardtLab/TEDBench) benchmark for protein fold classification (ICML 2026). MiAE is an SE(3)-invariant masked autoencoder that masks up to 90% of backbone frames and reconstructs the full structure with a lightweight decoder.
|
| 21 |
+
|
| 22 |
+
- **Code:** [GitHub Repository](https://github.com/BorgwardtLab/TEDBench)
|
| 23 |
+
- **Paper:** [Hugging Face Paper Page](https://huggingface.co/papers/2605.18552)
|
| 24 |
|
| 25 |
## Architecture sizes
|
| 26 |
|
|
|
|
| 61 |
booktitle={Proceedings of the 43rd International Conference on Machine Learning},
|
| 62 |
year={2026}
|
| 63 |
}
|
| 64 |
+
```
|