Protein Fold Classification at Scale: Benchmarking and Pretraining

Fuente: arXiv
Guardado en:
Detalles Bibliográficos
Autores principales: Chen, Dexiong, Manolache, Andrei, Niepert, Mathias, Borgwardt, Karsten
Formato: Preprint
Publicado: 2026
Materias:
Acceso en línea:
Etiquetas: Agregar Etiqueta
Sin Etiquetas, Sea el primero en etiquetar este registro!
_version_ 1866911694609973248
author Chen, Dexiong
Manolache, Andrei
Niepert, Mathias
Borgwardt, Karsten
author_facet Chen, Dexiong
Manolache, Andrei
Niepert, Mathias
Borgwardt, Karsten
contents Classifying protein topology is essential for deciphering biological function, but progress is held back by the lack of large-scale benchmarks that avoid duplicates and by models that do not scale well. We introduce TEDBench, a large-scale, non-redundant benchmark for protein fold classification constructed from the Encyclopedia of Domains (TED) and Foldseek-clustered AlphaFold structures. We show that on TEDBench, current protein representation learning methods either require very large models or fail to deliver strong performance. To address this challenge, we propose Masked Invariant Autoencoders (MiAE), a self-supervised framework for protein structure representation learning. MiAE uses an extremely high masking ratio of up to 90% with an $\mathrm{SE(3)}$-invariant encoder and a lightweight decoder that reconstructs backbone coordinates from the latent representation and mask tokens. MiAE scales well and outperforms supervised counterparts and state-of-the-art baselines on TEDBench, establishing a strong recipe for protein fold classification. To test transfer beyond AlphaFold structures, we further benchmark on a curated dataset from experimental structures of CATH v4.4. TEDBench is available at https://github.com/BorgwardtLab/TEDBench.
format Preprint
id arxiv_https___arxiv_org_abs_2605_18552
institution arXiv
publishDate 2026
record_format arxiv
spellingShingle Protein Fold Classification at Scale: Benchmarking and Pretraining
Chen, Dexiong
Manolache, Andrei
Niepert, Mathias
Borgwardt, Karsten
Machine Learning
Biomolecules
Quantitative Methods
Classifying protein topology is essential for deciphering biological function, but progress is held back by the lack of large-scale benchmarks that avoid duplicates and by models that do not scale well. We introduce TEDBench, a large-scale, non-redundant benchmark for protein fold classification constructed from the Encyclopedia of Domains (TED) and Foldseek-clustered AlphaFold structures. We show that on TEDBench, current protein representation learning methods either require very large models or fail to deliver strong performance. To address this challenge, we propose Masked Invariant Autoencoders (MiAE), a self-supervised framework for protein structure representation learning. MiAE uses an extremely high masking ratio of up to 90% with an $\mathrm{SE(3)}$-invariant encoder and a lightweight decoder that reconstructs backbone coordinates from the latent representation and mask tokens. MiAE scales well and outperforms supervised counterparts and state-of-the-art baselines on TEDBench, establishing a strong recipe for protein fold classification. To test transfer beyond AlphaFold structures, we further benchmark on a curated dataset from experimental structures of CATH v4.4. TEDBench is available at https://github.com/BorgwardtLab/TEDBench.
title Protein Fold Classification at Scale: Benchmarking and Pretraining
topic Machine Learning
Biomolecules
Quantitative Methods
url https://arxiv.org/abs/2605.18552