Search Results for author: Zaixi Zhang

Found 29 papers, 19 papers with code

FoldMark: Protecting Protein Generative Models with Watermarking

1 code implementation27 Oct 2024 Zaixi Zhang, Ruofan Jin, Kaidi Fu, Le Cong, Marinka Zitnik, Mengdi Wang

Protein structure is key to understanding protein function and is essential for progress in bioengineering, drug discovery, and molecular biology.

Drug Discovery Protein Structure Prediction

DeltaDock: A Unified Framework for Accurate, Efficient, and Physically Reliable Molecular Docking

1 code implementation15 Oct 2024 Jiaxian Yan, Zaixi Zhang, Jintao Zhu, Kai Zhang, Jianfeng Pei, Qi Liu

Despite these advancements, current methods are often tailored for specific docking settings, and limitations such as the neglect of protein side-chain structures, difficulties in handling large binding pockets, and challenges in predicting physically valid structures exist.

Blind Docking Drug Design +2

Generalized Protein Pocket Generation with Prior-Informed Flow Matching

no code implementations29 Sep 2024 Zaixi Zhang, Marinka Zitnik, Qi Liu

One critical step in this process involves designing protein pockets, the protein interface binding with the ligand.

valid

FlexSBDD: Structure-Based Drug Design with Flexible Protein Modeling

no code implementations29 Sep 2024 Zaixi Zhang, Mengdi Wang, Qi Liu

Structure-based drug design (SBDD), which aims to generate 3D ligand molecules binding to target proteins, is a fundamental task in drug discovery.

Data Augmentation Drug Design +1

Towards Few-shot Self-explaining Graph Neural Networks

1 code implementation14 Aug 2024 Jingyu Peng, Qi Liu, Linan Yue, Zaixi Zhang, Kai Zhang, Yunhao Sha

Subsequently, the predictor mimics the decision-making process, which makes predictions based on the generated explanation.

Model Inversion Attacks Through Target-Specific Conditional Diffusion Models

1 code implementation16 Jul 2024 Ouxiang Li, Yanbin Hao, Zhicai Wang, Bin Zhu, Shuo Wang, Zaixi Zhang, Fuli Feng

To alleviate these issues, leveraging on diffusion models' remarkable synthesis capabilities, we propose Diffusion-based Model Inversion (Diff-MI) attacks.

Image Reconstruction

What Improves the Generalization of Graph Transformers? A Theoretical Dive into the Self-attention and Positional Encoding

no code implementations4 Jun 2024 Hongkang Li, Meng Wang, Tengfei Ma, Sijia Liu, Zaixi Zhang, Pin-Yu Chen

Focusing on a graph data model with discriminative nodes that determine node labels and non-discriminative nodes that are class-irrelevant, we characterize the sample complexity required to achieve a desirable generalization error by training with stochastic gradient descent (SGD).

Graph Learning Node Classification

Structure-based Drug Design Benchmark: Do 3D Methods Really Dominate?

1 code implementation4 Jun 2024 Kangyu Zheng, Yingzhou Lu, Zaixi Zhang, Zhongwei Wan, Yao Ma, Marinka Zitnik, Tianfan Fu

Currently, the field of structure-based drug design is dominated by three main types of algorithms: search-based algorithms, deep generative models, and reinforcement learning.

Drug Design

Deep Geometry Handling and Fragment-wise Molecular 3D Graph Generation

no code implementations15 Mar 2024 Odin Zhang, Yufei Huang, Shichen Cheng, Mengyao Yu, Xujun Zhang, Haitao Lin, Yundian Zeng, Mingyang Wang, Zhenxing Wu, Huifeng Zhao, Zaixi Zhang, Chenqing Hua, Yu Kang, Sunliang Cui, Peichen Pan, Chang-Yu Hsieh, Tingjun Hou

Most earlier 3D structure-based molecular generation approaches follow an atom-wise paradigm, incrementally adding atoms to a partially built molecular fragment within protein pockets.

Graph Generation

FedGT: Federated Node Classification with Scalable Graph Transformer

no code implementations26 Jan 2024 Zaixi Zhang, Qingyong Hu, Yang Yu, Weibo Gao, Qi Liu

However, existing methods have the following limitations: (1) The links between local subgraphs are missing in subgraph federated learning.

Classification Federated Learning +2

Binding-Adaptive Diffusion Models for Structure-Based Drug Design

1 code implementation15 Jan 2024 Zhilin Huang, Ling Yang, Zaixi Zhang, Xiangxin Zhou, Yu Bao, Xiawu Zheng, Yuwei Yang, Yu Wang, Wenming Yang

Then the selected protein-ligand subcomplex is processed with SE(3)-equivariant neural networks, and transmitted back to each atom of the complex for augmenting the target-aware 3D molecule diffusion generation with binding interaction information.

Avg Drug Design

Sparse Attention-Based Neural Networks for Code Classification

no code implementations11 Nov 2023 Ziyang Xiang, Zaixi Zhang, Qi Liu

We introduce an approach named the Sparse Attention-based neural network for Code Classification (SACC) in this paper.

Classification Code Classification

AdaptSSR: Pre-training User Model with Augmentation-Adaptive Self-Supervised Ranking

1 code implementation NeurIPS 2023 Yang Yu, Qi Liu, Kai Zhang, Yuren Zhang, Chao Song, Min Hou, Yuqing Yuan, Zhihao Ye, Zaixi Zhang, Sanshi Lei Yu

Specifically, we adopt a multiple pairwise ranking loss which trains the user model to capture the similarity orders between the implicitly augmented view, the explicitly augmented view, and views from other users.

Contrastive Learning Data Augmentation

Full-Atom Protein Pocket Design via Iterative Refinement

1 code implementation NeurIPS 2023 Zaixi Zhang, Zepu Lu, Zhongkai Hao, Marinka Zitnik, Qi Liu

In the initial stage, the residue types and backbone coordinates are refined using a hierarchical context encoder, complemented by two structure refinement modules that capture both inter-residue and pocket-ligand interactions.

Geometric Deep Learning for Structure-Based Drug Design: A Survey

1 code implementation20 Jun 2023 Zaixi Zhang, Jiaxian Yan, Yining Huang, Qi Liu, Enhong Chen, Mengdi Wang, Marinka Zitnik

Structure-based drug design (SBDD) leverages the three-dimensional geometry of proteins to identify potential drug candidates.

Benchmarking Deep Learning +3

An Equivariant Generative Framework for Molecular Graph-Structure Co-Design

1 code implementation12 Apr 2023 Zaixi Zhang, Qi Liu, Chee-Kong Lee, Chang-Yu Hsieh, Enhong Chen

Our extensive investigation reveals that the 2D topology and 3D geometry contain intrinsically complementary information in molecule design, and provide new insights into machine learning-based molecule representation and generation.

3D geometry Drug Design +3

Backdoor Defense via Deconfounded Representation Learning

1 code implementation CVPR 2023 Zaixi Zhang, Qi Liu, Zhicai Wang, Zepu Lu, Qingyong Hu

The other clean model dedicates to capturing the desired causal effects by minimizing the mutual information with the confounding representations from the backdoored model and employing a sample-wise re-weighting scheme.

Backdoor Attack backdoor defense +1

Untargeted Attack against Federated Recommendation Systems via Poisonous Item Embeddings and the Defense

1 code implementation11 Dec 2022 Yang Yu, Qi Liu, Likang Wu, Runlong Yu, Sanshi Lei Yu, Zaixi Zhang

Experiments on two public datasets show that ClusterAttack can effectively degrade the performance of FedRec systems while circumventing many defense methods, and UNION can improve the resistance of the system against various untargeted attacks, including our ClusterAttack.

Contrastive Learning Recommendation Systems

FedRecover: Recovering from Poisoning Attacks in Federated Learning using Historical Information

no code implementations20 Oct 2022 Xiaoyu Cao, Jinyuan Jia, Zaixi Zhang, Neil Zhenqiang Gong

Existing defenses focus on preventing a small number of malicious clients from poisoning the global model via robust federated learning methods and detecting malicious clients when there are a large number of them.

Federated Learning

Hierarchical Graph Transformer with Adaptive Node Sampling

1 code implementation8 Oct 2022 Zaixi Zhang, Qi Liu, Qingyong Hu, Chee-Kong Lee

The Transformer architecture has achieved remarkable success in a number of domains including natural language processing and computer vision.

FLCert: Provably Secure Federated Learning against Poisoning Attacks

no code implementations2 Oct 2022 Xiaoyu Cao, Zaixi Zhang, Jinyuan Jia, Neil Zhenqiang Gong

Our key idea is to divide the clients into groups, learn a global model for each group of clients using any existing federated learning method, and take a majority vote among the global models to classify a test input.

Federated Learning

Model Inversion Attacks against Graph Neural Networks

no code implementations16 Sep 2022 Zaixi Zhang, Qi Liu, Zhenya Huang, Hao Wang, Chee-Kong Lee, Enhong Chen

One famous privacy attack against data analysis models is the model inversion attack, which aims to infer sensitive data in the training dataset and leads to great privacy concerns.

model Reinforcement Learning (RL)

FLDetector: Defending Federated Learning Against Model Poisoning Attacks via Detecting Malicious Clients

1 code implementation19 Jul 2022 Zaixi Zhang, Xiaoyu Cao, Jinyuan Jia, Neil Zhenqiang Gong

FLDetector aims to detect and remove the majority of the malicious clients such that a Byzantine-robust FL method can learn an accurate global model using the remaining clients.

Federated Learning Model Poisoning

ProtGNN: Towards Self-Explaining Graph Neural Networks

1 code implementation2 Dec 2021 Zaixi Zhang, Qi Liu, Hao Wang, Chengqiang Lu, Cheekong Lee

In this work, we propose Prototype Graph Neural Network (ProtGNN), which combines prototype learning with GNNs and provides a new perspective on the explanations of GNNs.

Graph Neural Network

Motif-based Graph Self-Supervised Learning for Molecular Property Prediction

1 code implementation NeurIPS 2021 Zaixi Zhang, Qi Liu, Hao Wang, Chengqiang Lu, Chee-Kong Lee

To bridge this gap, we propose Motif-based Graph Self-supervised Learning (MGSSL) by introducing a novel self-supervised motif generation framework for GNNs.

Molecular Property Prediction Property Prediction +2

GraphMI: Extracting Private Graph Data from Graph Neural Networks

1 code implementation5 Jun 2021 Zaixi Zhang, Qi Liu, Zhenya Huang, Hao Wang, Chengqiang Lu, Chuanren Liu, Enhong Chen

Then we design a graph auto-encoder module to efficiently exploit graph topology, node attributes, and target model parameters for edge inference.

Backdoor Attacks to Graph Neural Networks

2 code implementations19 Jun 2020 Zaixi Zhang, Jinyuan Jia, Binghui Wang, Neil Zhenqiang Gong

Specifically, we propose a \emph{subgraph based backdoor attack} to GNN for graph classification.

Backdoor Attack General Classification +2

Cannot find the paper you are looking for? You can Submit a new open access paper.