arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2202.08320cs.LG
收藏

TorchDrug:一个强大而灵活的药物发现机器学习平台

TorchDrug: A Powerful and Flexible Machine Learning Platform for Drug Discovery

Mila - Quebec AI Institute(米拉-魁北克人工智能研究所);University of Montreal(蒙特利尔大学);Peking University(北京大学);Shanghai Jiao Tong University(上海交通大学);Tsinghua University(清华大学)
Zhaocheng Zhu, Chence Shi, Zuobai Zhang, Shengchao Liu, Minghao Xu, Xinyu Yuan, Yangtian Zhang, Junkun Chen, Huiyu Cai, Jiarui Lu, Chang Ma, Runcheng Liu, Louis… 展开作者

Zhaocheng Zhu, Chence Shi, Zuobai Zhang, Shengchao Liu, Minghao Xu, Xinyu Yuan, Yangtian Zhang, Junkun Chen, Huiyu Cai, Jiarui Lu, Chang Ma, Runcheng Liu, Louis-Pascal Xhonneux, Meng Qu, Jian Tang

摘要:

机器学习具有巨大的潜力,可以彻底改变药物发现领域,近年来正引起越来越多的关注。然而,缺乏领域知识(例如,哪些任务工作)、标准基准和数据预处理管道是机器学习研究人员在该领域工作的主要障碍。为了促进药物发现的机器学习的进展,我们开发了TorchDrug,这是一个强大而灵活的机器学习平台,用于药物发现,构建在PyTorch之上。TorchDrug对药物发现中的各种重要任务进行基准测试,包括分子性质预测、预训练的分子表示、从头分子设计和优化、逆转录预测和生物医学知识图谱推理。基于几何深度学习(或图机器学习)、深度生成模型、强化学习和知识图谱推理的最新技术被实现用于这些任务。TorchDrug具有分层界面,便于新手和专家在该领域进行定制。教程、基准测试结果和文档可在https://torchdrug.ai上获得。代码在Apache License 2.0下发布。

英文摘要:

Machine learning has huge potential to revolutionize the field of drug discovery and is attracting increasing attention in recent years. However, lacking domain knowledge (e.g., which tasks to work on), standard benchmarks and data preprocessing pipelines are the main obstacles for machine learning researchers to work in this domain. To facilitate the progress of machine learning for drug discovery, we develop TorchDrug, a powerful and flexible machine learning platform for drug discovery built on top of PyTorch. TorchDrug benchmarks a variety of important tasks in drug discovery, including molecular property prediction, pretrained molecular representations, de novo molecular design and optimization, retrosynthsis prediction, and biomedical knowledge graph reasoning. State-of-the-art techniques based on geometric deep learning (or graph machine learning), deep generative models, reinforcement learning and knowledge graph reasoning are implemented for these tasks. TorchDrug features a hierarchical interface that facilitates customization from both novices and experts in this domain. Tutorials, benchmark results and documentation are available at https://torchdrug.ai. Code is released under Apache License 2.0.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑