GPU-accelerated high-performance computing to supercharge foundational deep learning method development for scalable and accurate prediction of protein structures

NIH Pandemic-Era Grants

Pandemic Era Grants

2024

Document text

Principal Investigator: Debswapna  Bhattacharya
Organization: VIRGINIA POLYTECHNIC INST AND ST UNIV
Fiscal Year: 2024
Award: $240,509
Funding agency: National Institute of General Medical Sciences

PROJECT SUMMARY/ABSTRACT:
This supplement aims to acquire a Dell high-performance computing (HPC) server with 8 NVIDIA H100
Graphics Processing Units (GPUs) to supercharge foundational deep learning method development for
scalable and accurate prediction of protein structures, paving the way to genomic-scale computational protein
modeling regardless of evolutionary relationships with previously annotated proteins. Artificial intelligence-
powered methods have led to a paradigm-shift in computational modeling of protein structures, yet even the
most successful approaches for protein structure prediction fail to accurately predict structures of large multi-
domain proteins with complex topologies or proteins with short sequences; and heavily depend on the
availability of evolutionary information that are not always abundant such as with orphan proteins or rapidly
evolving proteins. Work on structure prediction that uses single or few homologous sequences remains
inaccurate and/or inefficient, limiting scaling to genomic protein databases. Latest advances in artificial
intelligence such as foundational deep learning models hold the key to address the limitations. The parent R35
grant of this supplement aims to develop cutting-edge deep learning models to automate genomic-scale
protein structure modeling with the key tasks of: (1) accurate de novo modeling of protein structures beyond
evolutionary relatedness, even with single-sequence input; (2) high-fidelity identification of remotely
homologous proteins despite low sequence similarly to previously annotated proteins; and (3) atomistic
refinement of predicted protein structures to drive them towards experimental resolutions terms of
stereochemical qualities and side-chain positioning. Our substantial progress in the first three years of the
project has demonstrated the feasibility and promise of our approach. However, training and testing
foundational deep learning models leveraging the transformer neural network architectures on evolutionary-
scale molecular data require a large amount of GPU computing power. Using the current GPU resource
available to us, it takes six months for a developer to complete the training and testing of one deep learning
method end to end. While such a speed can yield steady progress, it is not fast enough to unleash the power
of these advanced deep learning methods and realize the full potential and impact of the parent R35 project.
This supplement will enable us to acquire a high-performance computing server consisting of 8 NVIDIA H100
80GB GPUs to significantly speed up the research in the parent R35 project. The requested GPUs can
drastically reduce the time to complete the development a deep learning method from about six months to less
than six weeks, thus dramatically improving the productivity of the developers and in turn accelerating
publication and dissemination of the methods and tools developed in this project. The large shared GPU
memory will enable us to train and optimize robust foundational deep learning methods powered by
transformers having millions of parameters, leading to scalable and accurate prediction of protein structures.

Terms: <AI system><Acceleration><Address><Artificial Intelligence><Biomedical Research><Complex><Computer Models><Computer Reasoning><Computerized Models><Data><Development><Disease><Disorder><Drug Design><Genomics><Grant><High Performance Computing><Homologous Protein><Homologous Sequences><Instruction><Machine Intelligence><Memory><Methods><Modeling><Molecular><Orphan><Parents><Peptide Domain><Position><Positioning Attribute><Productivity><Protein Databases><Protein Domains><Protein Homolog><ProteinHomolog><Proteins><Publications><Research><Research Resources><Resolution><Resources><Scientific Publication><Sequence Homologs><Speed><Structure><Tertiary Protein Structure><Testing><Therapeutic><Time><Training><Work><computational modeling><computational models><computer based models><computerized modeling><deep learning><deep learning based model><deep learning method><deep learning model><deep learning strategy><developmental><drug discovery><high-end computing><improved><method development><molecular scale><neural net architecture><neural network architecture><parent><protein structure><protein structure prediction><protein structures><proteins structure><resolutions><tool>