activity
20182024
most citedATOMO: Communication-efficient Learning via Atomic Sparsification

198 citations · 730 across the 23 of their papers we have counts for

collaborators

23 papers

cs.LG2024

Unsupervised Discovery of Steerable Factors When Graph Deep Generative Models Are Entangled

Shengchao Liu, Chengpeng Wang, Jiarui Lu +5

Deep generative models (DGMs) have been widely developed for graph data. However, much less investigation has been carried out on understanding the latent space of such pretrained…

cs.LG2023★ 8 cited

Symmetry-Informed Geometric Representation for Molecules, Proteins, and Crystalline Materials

Shengchao Liu, Weitao Du, Yanjing Li +10

Artificial intelligence for scientific discovery has recently generated significant interest within the machine learning and scientific communities, particularly in the domains of…

cs.LG2023★ 1 cited

GIMLET: A Unified Graph-Text Model for Instruction-Based Molecule Zero-Shot Learning

Haiteng Zhao, Shengchao Liu, Chang Ma +5

Molecule property prediction has gained significant attention in recent years. The main bottleneck is the label insufficiency caused by expensive lab experiments. In order to allev…

q-bio.BM2023★ 18 cited

ChatGPT-powered Conversational Drug Editing Using Retrieval and Domain Feedback

Shengchao Liu, Jiongxiao Wang, Yijin Yang +4

Recent advancements in conversational large language models (LLMs), such as ChatGPT, have demonstrated remarkable promise in various domains, including drug discovery. However, exi…

cs.LG2023★ 6 cited

A Group Symmetric Stochastic Differential Equation Model for Molecule Multi-modal Pretraining

Shengchao Liu, Weitao Du, Zhiming Ma +2

Molecule pretraining has quickly become the go-to schema to boost the performance of AI-based drug discovery. Naturally, molecules can be represented as 2D topological graphs or 3D…

cs.LG2023★ 26 cited

A Text-guided Protein Design Framework

Shengchao Liu, Yanjing Li, Zhuoxinran Li +10

Current AI-assisted protein design mainly utilizes protein sequential and structural information. Meanwhile, there exists tremendous knowledge curated by humans in the text format…

cs.LG2023★ 2 cited

Improving Domain Generalization with Domain Relations

Huaxiu Yao, Xinyu Yang, Xinyi Pan +3

Distribution shift presents a significant challenge in machine learning, where models often underperform during the test stage when faced with a different distribution than the one…

cs.LG2022★ 22 cited

Multi-modal Molecule Structure-text Model for Text-based Retrieval and Editing

Shengchao Liu, Weili Nie, Chengpeng Wang +6

There is increasing adoption of artificial intelligence in drug discovery. However, existing studies use machine learning to mainly utilize the chemical structures of molecules but…

cs.LG2022★ 1 cited

Flaky Performances when Pretraining on Relational Databases

Shengchao Liu, David Vazquez, Jian Tang +1

We explore the downstream task performances for graph neural network (GNN) self-supervised learning (SSL) methods trained on subgraphs extracted from relational databases (RDBs). I…

cs.LG2022★ 19 cited

Molecular Geometry Pretraining with SE(3)-Invariant Denoising Distance Matching

Shengchao Liu, Hongyu Guo, Jian Tang

Molecular representation pretraining is critical in various applications for drug and material discovery due to the limited number of labeled molecules, and most existing work focu…

cs.LG2022★ 7 cited

Evaluating Self-Supervised Learning for Molecular Graph Embeddings

Hanchen Wang, Jean Kaddour, Shengchao Liu +3

Graph Self-Supervised Learning (GSSL) provides a robust pathway for acquiring embeddings without expert labelling, a capability that carries profound implications for molecular gra…

cs.LG2022

An Empirical Study of Retrieval-enhanced Graph Neural Networks

Dingmin Wang, Shengchao Liu, Hanchen Wang +5

Graph Neural Networks (GNNs) are effective tools for graph representation learning. Most GNNs rely on a recursive neighborhood aggregation scheme, named message passing, thereby th…

cs.LG2022★ 27 cited

MolGenSurvey: A Systematic Survey in Machine Learning Models for Molecule Design

Yuanqi Du, Tianfan Fu, Jimeng Sun +1

Molecule design is a fundamental problem in molecular science and has critical applications in a variety of areas, such as drug discovery, material science, etc. However, due to th…

q-bio.BM2022★ 7 cited

Structured Multi-task Learning for Molecular Property Prediction

Shengchao Liu, Meng Qu, Zuobai Zhang +2

Multi-task learning for molecular property prediction is becoming increasingly important in drug discovery. However, in contrast to other domains, the performance of multi-task lea…

cs.LG2022★ 32 cited

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

Zhaocheng Zhu, Chence Shi, Zuobai Zhang +12

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., wh…

cs.LG2021★ 160 cited

Pre-training Molecular Graph Representation with 3D Geometry

Shengchao Liu, Hanchen Wang, Weiyang Liu +3

Molecular graph representation learning is a fundamental problem in modern drug and material discovery. Molecular graphs are typically modeled by their 2D topological structures, b…

cs.LG2021★ 2 cited

Attentive Walk-Aggregating Graph Neural Networks

Mehmet F. Demirel, Shengchao Liu, Siddhant Garg +2

Graph neural networks (GNNs) have been shown to possess strong representation power, which can be exploited for downstream prediction tasks on graph-structured data, such as molecu…

cs.CL2021★ 1 cited

BERT4SO: Neural Sentence Ordering by Fine-tuning BERT

Yutao Zhu, Jian-Yun Nie, Kun Zhou +3

Sentence ordering aims to arrange the sentences of a given text in the correct order. Recent work frames it as a ranking problem and applies deep neural networks to it. In this wor…

cs.CL2021★ 7 cited

Neural Sentence Ordering Based on Constraint Graphs

Yutao Zhu, Kun Zhou, Jian-Yun Nie +2

Sentence ordering aims at arranging a list of sentences in the correct order. Based on the observation that sentence order at different distances may rely on different types of inf…

cs.LG2020★ 52 cited

Learning To Navigate The Synthetically Accessible Chemical Space Using Reinforcement Learning

Sai Krishna Gottipati, Boris Sattarov, Sufeng Niu +9

Over the last decade, there has been significant progress in the field of machine learning for de novo drug design, particularly in deep generative models. However, current generat…

cs.LG2019★ 30 cited

Bad Global Minima Exist and SGD Can Reach Them

Shengchao Liu, Dimitris Papailiopoulos, Dimitris Achlioptas

Several works have aimed to explain why overparameterized neural networks generalize well when trained by Stochastic Gradient Descent (SGD). The consensus explanation that has emer…

cs.LG2018★ 104 cited

N-Gram Graph: Simple Unsupervised Representation for Graphs, with Applications to Molecules

Shengchao Liu, Mehmet Furkan Demirel, Yingyu Liang

Machine learning techniques have recently been adopted in various applications in medicine, biology, chemistry, and material engineering. An important task is to predict the proper…

stat.ML2018★ 198 cited

ATOMO: Communication-efficient Learning via Atomic Sparsification

Hongyi Wang, Scott Sievert, Zachary Charles +3

Distributed model training suffers from communication overheads due to frequent gradient updates transmitted between compute nodes. To mitigate these overheads, several studies pro…