Antibody DomainBed: Out-of-Distribution Generalization in Therapeutic Protein Design
Fuente:
arXiv
Enregistré dans:
| Auteurs principaux: | , , , , , , , , , , |
|---|---|
| Format: | Preprint |
| Publié: |
2024
|
| Sujets: | |
| Accès en ligne: | |
| Tags: |
Ajouter un tag
Pas de tags, Soyez le premier à ajouter un tag!
|
| _version_ | 1866917737972891648 |
|---|---|
| author | Tagasovska, Nataša Park, Ji Won Kirchmeyer, Matthieu Frey, Nathan C. Watkins, Andrew Martin Ismail, Aya Abdelsalam Jamasb, Arian Rokkum Lee, Edith Bryson, Tyler Ra, Stephen Cho, Kyunghyun |
| author_facet | Tagasovska, Nataša Park, Ji Won Kirchmeyer, Matthieu Frey, Nathan C. Watkins, Andrew Martin Ismail, Aya Abdelsalam Jamasb, Arian Rokkum Lee, Edith Bryson, Tyler Ra, Stephen Cho, Kyunghyun |
| contents | Machine learning (ML) has demonstrated significant promise in accelerating drug design. Active ML-guided optimization of therapeutic molecules typically relies on a surrogate model predicting the target property of interest. The model predictions are used to determine which designs to evaluate in the lab, and the model is updated on the new measurements to inform the next cycle of decisions. A key challenge is that the experimental feedback from each cycle inspires changes in the candidate proposal or experimental protocol for the next cycle, which lead to distribution shifts. To promote robustness to these shifts, we must account for them explicitly in the model training. We apply domain generalization (DG) methods to classify the stability of interactions between an antibody and antigen across five domains defined by design cycles. Our results suggest that foundational models and ensembling improve predictive performance on out-of-distribution domains. We publicly release our codebase extending the DG benchmark ``DomainBed,'' and the associated dataset of antibody sequences and structures emulating distribution shifts across design cycles. |
| format | Preprint |
| id |
arxiv_https___arxiv_org_abs_2407_21028 |
| institution | arXiv |
| publishDate | 2024 |
| record_format | arxiv |
| spellingShingle | Antibody DomainBed: Out-of-Distribution Generalization in Therapeutic Protein Design Tagasovska, Nataša Park, Ji Won Kirchmeyer, Matthieu Frey, Nathan C. Watkins, Andrew Martin Ismail, Aya Abdelsalam Jamasb, Arian Rokkum Lee, Edith Bryson, Tyler Ra, Stephen Cho, Kyunghyun Biomolecules Machine Learning Machine learning (ML) has demonstrated significant promise in accelerating drug design. Active ML-guided optimization of therapeutic molecules typically relies on a surrogate model predicting the target property of interest. The model predictions are used to determine which designs to evaluate in the lab, and the model is updated on the new measurements to inform the next cycle of decisions. A key challenge is that the experimental feedback from each cycle inspires changes in the candidate proposal or experimental protocol for the next cycle, which lead to distribution shifts. To promote robustness to these shifts, we must account for them explicitly in the model training. We apply domain generalization (DG) methods to classify the stability of interactions between an antibody and antigen across five domains defined by design cycles. Our results suggest that foundational models and ensembling improve predictive performance on out-of-distribution domains. We publicly release our codebase extending the DG benchmark ``DomainBed,'' and the associated dataset of antibody sequences and structures emulating distribution shifts across design cycles. |
| title | Antibody DomainBed: Out-of-Distribution Generalization in Therapeutic Protein Design |
| topic | Biomolecules Machine Learning |
| url | https://arxiv.org/abs/2407.21028 |