← Search

Jiaxiang Wu

24 accepted papers

2025

IgGM: A Generative Model for Functional Antibody and Nanobody Design

ICLR 2025poster

Immunoglobulins are crucial proteins produced by the immune system to identify and bind to foreign substances, playing an essential role in shielding organisms from infections and diseases. Designing specific antibodies opens new pathways for disease treatment. With the rise of deep learning, AI-dri…

2024

EBMDock: Neural Probabilistic Protein-Protein Docking via a Differentiable Energy Model

ICLR 2024poster

Protein complex formation, a pivotal challenge in contemporary biology, has recently gained interest from the machine learning community, particularly concerning protein-ligand docking tasks. In this paper, we delve into the equally crucial but comparatively under-investigated domain of protein-prot…

Cited by 10SourcePDFScholar
2024

MorphGrower: A Synchronized Layer-by-layer Growing Approach for Plausible Neuronal Morphology Generation

ICML 2024oral

Neuronal morphology is essential for studying brain functioning and understanding neurodegenerative disorders. As acquiring real-world morphology data is expensive, computational approaches for morphology generation have been studied. Traditional methods heavily rely on expert-set rules and paramete…

2023

DrugOOD: Out-of-Distribution Dataset Curator and Benchmark for AI-Aided Drug Discovery – a Focus on Affinity Prediction Problems with Noise Annotations

AAAI 2023technical

AI-aided drug discovery (AIDD) is gaining popularity due to its potential to make the search for new pharmaceuticals faster, less expensive, and more effective. Despite its extensive use in numerous fields (e.g., ADMET prediction, virtual screening), little research has been conducted on the out-of-…

Cited by 122SourcePDFScholar
2023

Privacy-Preserving Face Recognition Using Random Frequency Components

ICCV 2023poster

The ubiquitous use of face recognition has sparked increasing privacy concerns, as unauthorized access to sensitive face images could compromise the information of individuals. This paper presents an in-depth study of the privacy protection of face images' visual information and against recovery. Dr…

Cited by 21PDFcodeScholar
2023

Probabilistic Knowledge Distillation of Face Ensembles

CVPR 2023poster

Mean ensemble (i.e. averaging predictions from multiple models) is a commonly-used technique in machine learning that improves the performance of each individual model. We formalize it as feature alignment for ensemble in open-set face recognition and generalize it into Bayesian Ensemble Averaging (…

2023

PsyCoT: Psychological Questionnaire as Powerful Chain-of-Thought for Personality Detection

EMNLP 2023long findings

Recent advances in large language models (LLMs), such as ChatGPT, have showcased remarkable zero-shot performance across various NLP tasks. However, the potential of LLMs in personality detection, which involves identifying an individual's personality from their written texts, remains largely unexpl…

Cited by 0SourcecodeScholar
2023

Towards Stable Test-time Adaptation in Dynamic Wild World

ICLR 2023top-5%

Test-time adaptation (TTA) has shown to be effective at tackling distribution shifts between training and testing data by adapting a given model on test samples. However, the online model updating of TTA may be unstable and this is often a key obstacle preventing existing TTA methods from being depl…

2022

Efficient Test-Time Model Adaptation without Forgetting

ICML 2022spotlight

Test-time adaptation provides an effective means of tackling the potential distribution shift between model training and inference, by dynamically updating the model at test time. This area has seen fast progress recently, at the effectiveness of handling test shifts. Nonetheless, prior methods stil…

2022

Energy-Based Learning for Cooperative Games, with Applications to Valuation Problems in Machine Learning

ICLR 2022poster

Valuation problems, such as feature interpretation, data valuation and model valuation for ensembles, become increasingly more important in many machine learning applications. Such problems are commonly solved by well-known game-theoretic criteria, such as Shapley value or Banzhaf value. In this wor…

Cited by 18SourcePDFScholar
2022

Evaluation-Oriented Knowledge Distillation for Deep Face Recognition

CVPR 2022oral

Knowledge distillation (KD) is a widely-used technique that utilizes large networks to improve the performance of compact models. Previous KD approaches usually aim to guide the student to mimic the teacher's behavior completely in the representation space. However, such one-to-one corresponding con…

Cited by 47PDFcodeScholar
2022

GNN-Retro: Retrosynthetic Planning with Graph Neural Networks

AAAI 2022technical

Retrosynthetic planning plays an important role in the field of organic chemistry, which could generate a synthetic route for the target product. The synthetic route is a series of reactions which are started from the available molecules. The most challenging problem in the generation of the synthet…

Cited by 32SourcePDFScholar
2022

Privacy-Preserving Face Recognition with Learnable Privacy Budgets in Frequency Domain

ECCV 2022poster

"Face recognition technology has been used in many fields due to its high recognition accuracy, including the face unlocking of mobile devices, community access control systems, and city surveillance. As the current high accuracy is guaranteed by very deep network structures, facial images often nee…

2022

Self-Supervised Pre-training for Protein Embeddings Using Tertiary Structures

AAAI 2022technical

The protein tertiary structure largely determines its interaction with other molecules. Despite its importance in various structure-related tasks, fully-supervised data are often time-consuming and costly to obtain. Existing pre-training models mostly focus on amino-acid sequences or multiple sequen…

Cited by 40SourcePDFScholar
2021

AdaXpert: Adapting Neural Architecture for Growing Data

ICML 2021spotlight

In real-world applications, data often come in a growing manner, where the data volume and the number of classes may increase dynamically. This will bring a critical challenge for learning: given the increasing data volume or the number of classes, one has to instantaneously adjust the neural model…

2021

PSSM-Distil: Protein Secondary Structure Prediction (PSSP) on Low-Quality PSSM by Knowledge Distillation with Contrastive Learning

AAAI 2021technical

Protein secondary structure prediction (PSSP) is an essential task in computational biology. To achieve the accurate PSSP, the general and vital feature engineering is to use multiple sequence alignment (MSA) for Position-Specific Scoring Matrix (PSSM) extraction. However, when only low-quality PSSM…

2020

Revisiting Parameter Sharing for Automatic Neural Channel Number Search

NeurIPS 2020poster

Recent advances in neural architecture search inspire many channel number search algorithms~(CNS) for convolutional neural networks. To improve searching efficiency, parameter sharing is widely applied, which reuses parameters among different channel configurations. Nevertheless, it is unclear how p…

2019

Exploring Fast and Communication-Efficient Algorithms in Large-Scale Distributed Networks

AISTATS 2019poster

The communication overhead has become a significant bottleneck in data-parallel network with the increasing of model size and data samples. In this work, we propose a new algorithm LPC-SVRG with quantized gradients and its acceleration ALPC-SVRG to effectively reduce the communication complexity wh…

Cited by 25SourcePDFScholar
2018

Error Compensated Quantized SGD and its Applications to Large-scale Distributed Optimization

ICML 2018oral

Large-scale distributed optimization is of great importance in various applications. For data-parallel based distributed learning, the inter-node gradient communication often becomes the performance bottleneck. In this paper, we propose the error compensated quantized stochastic gradient descent alg…

Cited by 286SourcePDFScholar
2016

Quantized Convolutional Neural Networks for Mobile Devices

CVPR 2016poster

Recently, convolutional neural networks (CNN) have demonstrated impressive performance in various computer vision tasks. However, high performance hardware is typically indispensable for the application of CNN models due to the high computation complexity, which prohibits their further extensions. I…

Cited by 1586PDFcodeScholar