The microscope is the mask: privileged views and labels from a cryo-ET forward model
Organizations: MRC Laboratory of Molecular Biology Cambridge, UK
Abstract
We explore the use of simulated data for training a model for protein annotation in crowded cryo-electron tomography volumes reconstructed from images collected at limited tilt angles and severely corrupted by the measurement operator. Firstly, we leverage the corruptions imposed by the forward model to generate domain-specific augmented paired views of the exact same scene for an invariance objective integrated into the LeJEPA self-supervised training framework. Secondly, we use additional information from the simulation pipeline such as the positions and identity of proteins in the simulated volumes to inform the architecture of the model and the loss function, so that semantic information is localised at protein positions in the resulting dense feature volume. The resulting model, CARNIVAL, is evaluated without finetuning on classification and detection tasks in real tomograms, using a benchmark dataset containing multiple protein types and two tomogram processing types. We show that CARNIVAL outperforms a state-of-the-art model trained using a contrastive objective on simulated data but without forward model-based paired views or privileged information.
Figures & tables
| Tomogram | Filtered | Denoised | ||
|---|---|---|---|---|
| CARNIVAL | TomoTwin | CARNIVAL | TomoTwin | |
| TS_5_2 | 0.508 | 0.431 | 0.529 | 0.410 |
| TS_14_1 | 0.389 | 0.311 | 0.357 | 0.311 |
| TS_43_9 | 0.542 | 0.437 | 0.465 | 0.425 |
Appendix figures & tables8 assets
Supplementary material from the paper’s appendix.
Appendix
| Transformation | Form | Sampled from |
| Measurement operator | ||
| Bulk medium (ice / cytoplasm) | on the volume before projection, with so structure displaces medium; fill level | |
| Large-scale medium variation | generalised to a smooth field of the same mean, zero-mean unit-std low-passed by , | |
| Tilt scheme (missing wedge) | projections at in steps of ; the wedge follows from | |
| Defocus (CTF) | per tilt, with varying with depth along the beam | |
| CTF envelope | from partial coherence, tilt-independent, multiplied into the CTF | |
| Filtered | Denoised | # particles | |||
|---|---|---|---|---|---|
| CARNIVAL | TomoTwin | CARNIVAL | TomoTwin | ||
| ribosome | 0.98 | 0.96 | 0.98 | 0.95 | 10820 |
| VLP | 0.95 | 0.82 | 0.94 | 0.78 | 1061 |
| thyroglobulin | 0.82 | 0.81 | 0.83 | 0.69 | 5368 |
| -galactosidase | 0.72 | 0.68 | 0.73 | 0.51 | 3725 |
| apoferritin | 0.98 | 0.92 | 0.98 | 0.93 | 9149 |
| Filtered | Denoised | |||
| Protein type | CARNIVAL | TomoTwin | CARNIVAL | TomoTwin |
| TS_5_2 | ||||
| ribosome | ||||
| VLP | ||||
| thyroglobulin | ||||
| -galactosidase | ||||
Protein Representation Learning
Protein representation learning builds computational features that capture sequence, structure, function, and molecular relationships.
See all 64 papers