← Search

Jun Xia

49 accepted papers

2026

Departures: Distributional Transport for Single-Cell Perturbation Prediction with Neural Schrödinger Bridges

AAAI 2026technical

Predicting single-cell perturbation outcomes directly advances gene function analysis and facilitates drug candidate selection, making it a key driver of both basic and translational biomedical research. However, a major bottleneck in this task is the unpaired nature of single-cell data, as the same

Cited by 0SourcePDFScholar
2026

Doloris: Dual Conditional Diffusion Implicit Bridges with Sparsity Masking Strategy for Unpaired Single-Cell Perturbation Estimation

ICLR 2026poster

Estimating single-cell responses across various perturbations facilitates the identification of key genes and enhances drug screening, significantly boosting experimental efficiency. However, single-cell sequencing is a destructive process, making it impossible to capture the same cell's phenotype b…

Cited by 0SourcecodeScholar
2026

Fast Data Mixture Optimization via Gradient Descent

ICLR 2026poster

While large and diverse datasets have driven recent advances in large models, identifying the optimal data mixture for pre-training and post-training remains a significant open problem. We address this challenge with FastMix, a novel framework that automates data mixture discovery while training onl…

Cited by 0SourcecodeScholar
2026

I2Mole: Interaction-aware Invariant Molecular Learning For Generalizable Property Prediction

ICLR 2026poster

Molecular interactions are a common phenomenon in physical chemistry field, which could produce unexpected biochemical properties harmful to humans, such as drug-drug interactions. Machine learning has the potential to deliver rapid and accurate predictions. However, the complexity of molecular stru…

Cited by 0SourceScholar
2026

MM-Spectrum: Multimodal Multi-spectral Molecular Structural Elucidation with a Stable MoE Framework

ICML 2026poster

Inferring molecular structures from multimodal spectroscopic measurements requires integrating complementary yet highly heterogeneous signals. However, the common paradigm of directly concatenating multispectral sequences can exhibit anomalous performance degradation, primarily due to pronounced het…

Cited by 0SourceScholar
2026

Regressor-guided Diffusion Model for De Novo Peptide Sequencing with Explicit Mass Control

AAAI 2026technical

The discovery of novel proteins relies on sensitive protein identification, for which de novo peptide sequencing (DNPS) from mass spectra is a crucial approach. While deep learning has advanced DNPS, existing models inadequately enforce the fundamental mass consistency constraint—that a predicted pe

Cited by 0SourcePDFScholar
2026

SGI: Structured 2D Gaussians for Efficient and Compact Large Image Representation

CVPR 2026

2D Gaussian Splatting has emerged as a novel image representation technique that can support efficient rendering on low-end devices. However, scaling to high-resolution images requires optimizing and storing millions of unstructured Gaussian primitives independently, leading to slow convergence and

Cited by 0SourcecodeScholar
2026

TAMPO: Task- and Model-Aware Automatic Prompt Optimization for Robust and Controllable Auto-Routing in LLM-based Systems

ICML 2026poster

Automatic Prompt Optimization (APO) enables Large Language Models (LLMs) to adapt to specific tasks while minimizing manual engineering costs. However, since existing APO approaches either rely solely on multi-round iterative procedures or use model-specific generators tailored to optimizing prompts…

Cited by 0SourceScholar
2025

A Comprehensive and Systematic Review for Deep Learning-Based De Novo Peptide Sequencing

IJCAI 2025

Tandem mass spectrometry (MS/MS) has revolutionized the field of proteomics, enabling the high-throughput identification of proteins. However, one of the central challenges in mass spectrometry-based proteomics remains peptide identification, especially in the absence of a comprehensive peptide data

Cited by 0SourcePDFScholar
2025

Bridging the Gap between Database Search and \emph{De Novo} Peptide Sequencing with SearchNovo

ICLR 2025poster

Accurate protein identification from mass spectrometry (MS) data is fundamental to unraveling the complex roles of proteins in biological systems, with peptide sequencing being a pivotal step in this process. The two main paradigms for peptide sequencing are database search, which matches experiment…

2025

EDBench: Large-Scale Electron Density Data for Molecular Modeling

NeurIPS 2025poster

Existing molecular machine learning force fields (MLFFs) generally focus on the learning of atoms, molecules, and simple quantum chemical properties (such as energy and force), but ignore the importance of electron density (ED) $\rho(r)$ in accurately understanding molecular force fields (MFFs). ED…

Cited by 0SourcecodeScholar
2025

Electron Density-enhanced Molecular Geometry Learning

IJCAI 2025

Electron density (ED), which describes the probability distribution of electrons in space, is crucial for accurately understanding the energy and force distribution in molecular force fields (MFF). Existing machine learning force fields (MLFF) focus on mining appropriate physical quantities from the

2025

GRAPE: Heterogeneous Graph Representation Learning for Genetic Perturbation with Coding and Non-Coding Biotype

IJCAI 2025

Predicting genetic perturbations enables the identification of potentially crucial genes prior to wet-lab experiments, significantly improving overall experimental efficiency. Since genes are the foundation of cellular life, building gene regulatory networks (GRN) is essential to understand and pred

2025

MTGIB-UNet: A Multi-Task Graph Information Bottleneck and Uncertainty Weighted Network for ADMET Prediction

IJCAI 2025

Accurate prediction of ADMET (Absorption, Distribution, Metabolism, Excretion, and Toxicity) properties is crucial in drug development, as these properties directly impact a drug's efficacy and safety. However, existing multi-task learning models often face challenges related to noise interference a

Cited by 0SourcePDFScholar
2025

MeToken: Uniform Micro-environment Token Boosts Post-Translational Modification Prediction

ICLR 2025poster

Post-translational modifications (PTMs) profoundly expand the complexity and functionality of the proteome, regulating protein attributes and interactions that are crucial for biological processes. Accurately predicting PTM sites and their specific types is therefore essential for elucidating protei…

2025

PRESCRIBE: Predicting Single-Cell Responses with Bayesian Estimation

NeurIPS 2025poster

In single-cell perturbation prediction, a central task is to forecast the effects of perturbing a gene unseen in the training data. The efficacy of such predictions depends on two factors: (1) the similarity of the target gene to those covered in the training data, which informs model (epistemic) un…

Cited by 0SourceScholar
2025

ReNovo: Retrieval-Based \emph{De Novo} Mass Spectrometry Peptide Sequencing

ICLR 2025poster

Proteomics is the large-scale study of proteins. Tandem mass spectrometry, as the only high-throughput technique for protein sequence identification, plays a pivotal role in proteomics research. One of the long-standing challenges in this field is peptide identification, which entails determining th…

Cited by 0SourcePDFScholar
2024

A Graph is Worth $K$ Words: Euclideanizing Graph using Pure Transformer

ICML 2024poster

Can we model Non-Euclidean graphs as pure language or even Euclidean vectors while retaining their inherent information? The Non-Euclidean property have posed a long term challenge in graph modeling. Despite recent graph neural networks and graph transformers efforts encoding graphs as Euclidean vec…

2024

AdaNovo: Towards Robust \emph{De Novo} Peptide Sequencing in Proteomics against Data Biases

NeurIPS 2024poster

Tandem mass spectrometry has played a pivotal role in advancing proteomics, enabling the high-throughput analysis of protein composition in biological tissues. Despite the development of several deep learning methods for predicting amino acid sequences (peptides) responsible for generating the obser…

Cited by 0SourcePDFScholar
2024

An Image-enhanced Molecular Graph Representation Learning Framework

IJCAI 2024poster

Extracting rich molecular representation is a crucial prerequisite for accurate drug discovery. Recent molecular representation learning methods achieve impressive progress, but the paradigm of learning from a single modality gradually encounters the bottleneck of limited representation capabilities…

2024

An Online Rcm Adjusting System for Robot-Assisted Retinal Surgeries

IROS 2024poster

In robot-assisted retinal surgery, a Remote Center of Motion (Rcm) allows the surgical instrument to rotate around a distal fixed point without any lateral translations. The Rcm point should be perfectly aligned inside the trocar. Otherwise, unexpected tool translations at the expected remote center…

Cited by 0SourceScholar
2024

Cross-Gate MLP with Protein Complex Invariant Embedding Is a One-Shot Antibody Designer

AAAI 2024technical

Antibodies are crucial proteins produced by the immune system in response to foreign substances or antigens. The specificity of an antibody is determined by its complementarity-determining regions (CDRs), which are located in the variable domains of the antibody chains and form the antigen-binding s…

2024

Deciphering RNA Secondary Structure Prediction: A Probabilistic K-Rook Matching Perspective

ICML 2024poster

The secondary structure of ribonucleic acid (RNA) is more stable and accessible in the cell than its tertiary structure, making it essential for functional prediction. Although deep learning has shown promising results in this field, current methods suffer from poor generalization and high complexit…

2024

Deep Graph Neural Networks via Posteriori-Sampling-based Node-Adaptative Residual Module

NeurIPS 2024poster

Graph Neural Networks (GNNs), a type of neural network that can learn from graph-structured data through neighborhood information aggregation, have shown superior performance in various downstream tasks. However, as the number of layers increases, node representations becomes indistinguishable, whic…

2024

End-to-end Learnable Clustering for Intent Learning in Recommendation

NeurIPS 2024poster

Intent learning, which aims to learn users' intents for user understanding and item recommendation, has become a hot research spot in recent years. However, existing methods suffer from complex and cumbersome alternating optimization, limiting performance and scalability. To this end, we propose a n…

2024

FlexMol: A Flexible Toolkit for Benchmarking Molecular Relational Learning

NeurIPS 2024poster

Molecular relational learning (MRL) is crucial for understanding the interaction behaviors between molecular pairs, a critical aspect of drug discovery and development. However, the large feasible model space of MRL poses significant challenges to benchmarking, and existing MRL frameworks face limit…

2024

KW-Design: Pushing the Limit of Protein Design via Knowledge Refinement

ICLR 2024poster

Recent studies have shown competitive performance in protein inverse folding, while most of them disregard the importance of predictive confidence, fail to cover the vast protein space, and do not incorporate common protein knowledge. Given the great success of pretrained models on diverse protein-r…

2024

Learning Complete Protein Representation by Dynamically Coupling of Sequence and Structure

NeurIPS 2024poster

Learning effective representations is imperative for comprehending proteins and deciphering their biological functions. Recent strides in language models and graph neural networks have empowered protein models to harness primary or tertiary structure information for representation learning. Neverthe…

Cited by 0SourcePDFScholar
2024

MMGNN: A Molecular Merged Graph Neural Network for Explainable Solvation Free Energy Prediction

IJCAI 2024poster

In this paper, we address the challenge of accurately modeling and predicting Gibbs free energy in solute-solvent interactions, a pivotal yet complex aspect in the field of chemical modeling. Traditional approaches, primarily relying on deep learning models, face limitations in capturing the intrica…

Cited by 5SourcePDFScholar
2024

NovoBench: Benchmarking Deep Learning-based \emph{De Novo} Sequencing Methods in Proteomics

NeurIPS 2024poster

Tandem mass spectrometry has played a pivotal role in advancing proteomics, enabling the analysis of protein composition in biological tissues. Many deep learning methods have been developed for \emph{de novo} peptide sequencing task, i.e., predicting the peptide sequence for the observed mass spect…

2024

ProtGO: Function-Guided Protein Modeling for Unified Representation Learning

NeurIPS 2024poster

Protein representation learning is indispensable for various downstream applications of artificial intelligence for bio-medicine research, such as drug design and function prediction. However, achieving effective representation learning for proteins poses challenges due to the diversity of data moda…

Cited by 0SourcePDFScholar
2024

WaveAttack: Asymmetric Frequency Obfuscation-based Backdoor Attacks Against Deep Neural Networks

NeurIPS 2024poster

Due to the increasing popularity of Artificial Intelligence (AI), more and more backdoor attacks are designed to mislead Deep Neural Network (DNN) predictions by manipulating training samples or processes. Although backdoor attacks have been investigated in various scenarios, they still suffer from…

2023

CVT-SLR: Contrastive Visual-Textual Transformation for Sign Language Recognition With Variational Alignment

CVPR 2023highlight

Sign language recognition (SLR) is a weakly supervised task that annotates sign videos as textual glosses. Recent studies show that insufficient training caused by the lack of large-scale available sign datasets becomes the main bottleneck for SLR. Most SLR works thereby adopt pretrained visual modu…

2023

Deep Manifold Graph Auto-Encoder For Attributed Graph Embedding

ICASSP 2023accepted

Representing graph data in a low-dimensional space for subsequent tasks is the purpose of attributed graph embedding. Most existing neural network approaches learn latent representations by minimizing reconstruction errors. Rare work considers the data distribution and the topological structure of l…

Cited by 0SourceScholar
2023

Dink-Net: Neural Clustering on Large Graphs

ICML 2023poster

Deep graph clustering, which aims to group the nodes of a graph into disjoint clusters with deep neural networks, has achieved promising progress in recent years. However, the existing methods fail to scale to the large graph with million nodes. To solve this problem, a scalable deep graph clusterin…

2023

Mole-BERT: Rethinking Pre-training Graph Neural Networks for Molecules

ICLR 2023poster

Recent years have witnessed the prosperity of pre-training graph neural networks (GNNs) for molecules. Typically, atom types as node attributes are randomly masked, and GNNs are then trained to predict masked types as in AttrMask \citep{hu2020strategies}, following the Masked Language Modeling (MLM)…

2023

Temporal Attention Unit: Towards Efficient Spatiotemporal Predictive Learning

CVPR 2023poster

Spatiotemporal predictive learning aims to generate future frames by learning from historical frames. In this paper, we investigate existing methods and present a general framework of spatiotemporal predictive learning, in which the spatial encoder and decoder capture intra-frame features and the mi…

2023

Understanding the Limitations of Deep Models for Molecular property prediction: Insights and Solutions

NeurIPS 2023poster

Molecular Property Prediction (MPP) is a crucial task in the AI-driven Drug Discovery (AIDD) pipeline, which has recently gained considerable attention thanks to advancements in deep learning. However, recent research has revealed that deep models struggle to beat traditional non-deep ones on MPP. I…

Cited by 37SourcePDFScholar
2023

Wordreg: Mitigating the Gap between Training and Inference with Worst-Case Drop Regularization

ICASSP 2023accepted

Dropout has emerged as one of the most frequently used techniques for training deep neural networks (DNNs). Although effective, the sampled sub-model by random dropout during training is inconsistent with the full model (without dropout) during inference. To mitigate this undesirable gap, we propose…

Cited by 0SourceScholar
2022

Eliminating Backdoor Triggers for Deep Neural Networks Using Attention Relation Graph Distillation

IJCAI 2022poster

Due to the prosperity of Artificial Intelligence (AI) techniques, more and more backdoors are designed by adversaries to attack Deep Neural Networks (DNNs). Although the state-of-the-art method Neural Attention Distillation (NAD) can effectively erase backdoor triggers from DNNs, it still suffers fr…

2022

ProGCL: Rethinking Hard Negative Mining in Graph Contrastive Learning

ICML 2022spotlight

Contrastive Learning (CL) has emerged as a dominant technique for unsupervised representation learning which embeds augmented versions of the anchor close to each other (positive samples) and pushes the embeddings of other samples (negatives) apart. As revealed in recent studies, CL can benefit from…

2022

Stacked Multi-Scale Attention Network for Image Colorization

ICASSP 2022accepted

Deep convolutional networks (CNNs) show their potential in image colorization for producing plausible results. Recently, the attention mechanism further boosts the performances of CNNs by constructing channel and spatial interactions. However, existing attention methods are performed in a single-sca…

Cited by 0SourceScholar
2022

Using Context-to-Vector with Graph Retrofitting to Improve Word Embeddings

ACL 2022long

Although contextualized embeddings generated from large-scale pre-trained models perform well in many tasks, traditional static embeddings (e.g., Skip-gram, Word2Vec) still play an important role in low-resource and lightweight settings due to their low computational cost, ease of deployment, and st…

2020

Microscope-Guided Autonomous Clear Corneal Incision

ICRA 2020poster

Clear Corneal Incision, a challenging step in cataract surgery, and important to the overall quality of the surgery. New surgeons usually spend one full year trying to perfect their incision, but even after such rigorous training deficient incisions can still occur. This paper proposes an autonomous…

Cited by 6SourceScholar