← Search

Junbin Gao

42 accepted papers

2026

ATOM: A Pretrained Neural Operator for Multitask Molecular Dynamics

ICLR 2026poster

Molecular dynamics (MD) simulations underpin modern computational drug discovery, materials science, and biochemistry. Recent machine learning models provide high-fidelity MD predictions without the need for repeated quantum-mechanical force calculations, enabling significant speedups over conventio…

Cited by 0SourcecodeScholar
2026

Aligning Collaborative View Recovery and Tensorial Subspace Learning via Latent Representation for Incomplete Multi-View Clustering

ICLR 2026poster

Multi-view data usually suffer from partially missing views in open scenarios, which inevitably degrades clustering performance. The incomplete multi-view clustering (IMVC) has attracted increasing attention and achieved significant success. Although existing imputation-based IMVC methods perform we…

Cited by 0SourceScholar
2026

BiOTPrompt: Bidirectional Optimal Transport Guided Prompting for Disease Evolution-aware Radiology Report Generation

CVPR 2026

Radiology report generation (RRG) aims to automatically describe medical images via free-text reports. In clinical practice, comparing current and prior chest X-rays is essential for assessing disease progression, motivating the development of longitudinal RRG methods. However, most existing approac

Cited by 0SourcecodeScholar
2026

Capacity-Agnostic Parameter Isolation for Continual Graph Learning

ICML 2026poster

Existing parameter isolation-based methods in continual learning employ diverse designs to learn more tasks within a limited model capacity. However, most of their designs inevitably incur substantial computational overhead if their model capacity is enlarged to accommodate further tasks as the task…

Cited by 0SourceScholar
2026

Deep Multi-view Graph Clustering via Attribute-aware Bidirectional Structural Refinement and Pseudo-label Guided Multi-level Fusion

ICML 2026poster

Deep multi-view graph clustering (DMGC) typically leverages graph neural networks for representation learning, but most existing methods excessively depend on local and static graph structures and only utilize simplistic cross-view fusion strategies. To this end, this paper proposes **A**ttribute-aw…

Cited by 0SourceScholar
2026

Dual-channel Dynamic Graph Neural Networks with Adaptive Adjacency Learning and Multi-scale Representation Fusion

ICML 2026poster

Graph neural networks (GNNs) have been demonstrated to be powerful tools for analyzing structural graph data. However, most existing methods usually rely on fixed adjacency structures for information propagation, lacking strong adaptability to the latent semantic relationships that exist but are not…

Cited by 0SourceScholar
2026

Expanding the Chaos: Neural Operator for Stochastic (Partial) Differential Equations

ICML 2026poster

Stochastic differential equations (SDEs) and stochastic partial differential equations (SPDEs) are fundamental for modeling stochastic dynamics across the natural sciences and modern machine learning. Learning their solution operators with deep learning models promises fast solvers and new perspecti…

Cited by 0SourceScholar
2026

GeoEvo: Identity-Aware Potential Game with Geometric Evolution for Personalized Multimodal Federated Learning

ICML 2026poster

We reconceptualize Personalized Multimodal Federated Learning (PMFL) by treating missing modalities as intrinsic structural identities that constrain each client to a distinct Riemannian submanifold, rather than deficiencies to be compensated. To resolve the tension between identity preservation and…

Cited by 0SourceScholar
2026

Information Geometry Loss for Time Series Forecasting

ICML 2026poster

Time series forecasting fundamentally involves learning probability distributions over future observations. However, existing loss functions rely on point-wise Euclidean metrics, neglecting the intrinsic geometric structure of probability distributions. This leads to suboptimal alignment between pre…

Cited by 0SourceScholar
2026

Learning Manifold and Itô Dynamics with Branched Neural Rough Differential Equations

ICML 2026poster

Neural rough differential equations (NRDEs) learn continuous-time dynamics from irregularly sampled sequences by encoding the input path with signature features, providing robustness to discretisation and sampling irregularity. However, existing NRDEs implicitly rely on algebraic identities that can…

Cited by 0SourceScholar
2026

S$^3$GNN: Efficient Global Mixing and Local Message Passing for Long-Range Graph Learning

ICML 2026spotlight

Message-passing neural networks (MPNNs) often suffer from an information bottleneck when capturing long-range dependencies, leading to the oversquashing (OSQ) phenomenon. Alongside spatial connectivity enrichment (e.g., rewiring), recent studies have shown that spectral filtering can yield strong lo…

Cited by 0SourceScholar
2026

Two Modalities Are Better Than One: Efficient Adversarial Purification via Multimodal Diffusion Models

ICML 2026poster

Adversarial purification uses generative models to restore clean data distributions from unseen attacks without retraining classifiers. However, unimodal diffusion-based approaches struggle to preserve semantic consistency, while recent multimodal variants rely on computationally expensive adversari…

Cited by 0SourceScholar
2026

WATS: Wavelet-Aware Temperature Scaling for Reliable Graph Neural Networks

ICLR 2026poster

Graph Neural Networks (GNNs) have demonstrated strong predictive performance on relational data; however, their confidence estimates often misalign with actual predictive correctness, posing significant limitations for deployment in safety-critical settings. While existing graph-aware calibration me…

Cited by 0SourcecodeScholar
2025

A Riemannian Approach to Ground Metric Learning for Optimal Transport

ICASSP 2025accepted

Optimal transport (OT) theory has attracted much attention in machine learning and signal processing applications. OT defines a notion of distance between probability distributions of source and target data points. A crucial factor that influences OT-based distances is the ground metric of the embed…

Cited by 0SourceScholar
2025

ACT as Human: Multimodal Large Language Model Data Annotation with Critical Thinking

NeurIPS 2025poster

Supervised learning relies on high-quality labeled data, but obtaining such data through human annotation is both expensive and time-consuming. Recent work explores using large language models (LLMs) for annotation, but LLM-generated labels still fall short of human-level quality. To address this pr…

Cited by 0SourceScholar
2025

Diffusing to the Top: Boost Graph Neural Networks with Minimal Hyperparameter Tuning

ICLR 2025poster

Graph Neural Networks (GNNs) are proficient in graph representation learning and achieve promising performance on versatile tasks such as node classification and link prediction. Usually, a comprehensive hyperparameter tuning is essential for fully unlocking GNN's top performance, especially for com…

2025

HC-LLM: Historical-Constrained Large Language Models for Radiology Report Generation

AAAI 2025technical

Radiology report generation (RRG) models typically focus on individual exams, often overlooking the integration of historical visual or textual data, which is crucial for patient follow-ups. Traditional methods usually struggle with long sequence dependencies when incorporating historical informatio…

2025

Hybrid-Collaborative Augmentation and Contrastive Sample Adaptive-Differential Awareness for Robust Attributed Graph Clustering

NeurIPS 2025poster

Due to its powerful capability of self-supervised representation learning and clustering, contrastive attributed graph clustering (CAGC) has achieved great success, which mainly depends on effective data augmentation and contrastive objective setting. However, most CAGC methods utilize edges as auxi…

Cited by 0SourcecodeScholar
2025

Language-Driven Multi-Label Zero-Shot Learning with Semantic Granularity

ICCV 2025poster

Recent methods learn class-unified prompt contexts by image data to adapt CLIP to zero-shot multi-label image classification, which achieves impressive performance. However, simply tuning prompts is insufficient to deal with novel classes across different semantic granularity levels. This limitation…

2025

When Graph Neural Networks Meet Dynamic Mode Decomposition

ICLR 2025poster

Graph Neural Networks (GNNs) have emerged as fundamental tools for a wide range of prediction tasks on graph-structured data. Recent studies have drawn analogies between GNN feature propagation and diffusion processes, which can be interpreted as dynamical systems. In this paper, we delve deeper int…

Cited by 0SourcePDFScholar
2024

AutoFGNN: A Framework for Extracting All Frequency Information from Large-Scale Graphs

ICASSP 2024accepted

As a powerful model for deep learning on graph-structured data, the scalability limitation of Graph Neural Networks (GNNs) are receiving increasing attention. To tackle this limitation, two categories of scalable GNNs have been proposed: sampling-based and model simplification methods. However, samp…

Cited by 0SourceScholar
2024

Diffusion Models Demand Contrastive Guidance for Adversarial Purification to Advance

ICML 2024poster

In adversarial defense, adversarial purification can be viewed as a special generation task with the purpose to remove adversarial attacks and diffusion models excel in adversarial purification for their strong generative power. With different predetermined generation requirements, various types of…

Cited by 6SourcePDFScholar
2024

Graph Neural Networks with Soft Association between Topology and Attribute

AAAI 2024technical

Graph Neural Networks (GNNs) have shown great performance in learning representations for graph-structured data. However, recent studies have found that the interference between topology and attribute can lead to distorted node representations. Most GNNs are designed based on homophily assumptions,…

2023

Riemannian Accelerated Gradient Methods via Extrapolation

AISTATS 2023poster

In this paper, we propose a convergence acceleration scheme for general Riemannian optimization problems by extrapolating iterates on manifolds. We show that when the iterates are generated from the Riemannian gradient descent method, the scheme achieves the optimal convergence rate asymptotically a…

Cited by 10SourcePDFScholar
2022

OTExtSum: Extractive Text Summarisation with Optimal Transport

NAACL 2022findings

Extractive text summarisation aims to select salient sentences from a document to form a short yet informative summary. While learning-based methods have achieved promising results, they have several limitations, such as dependence on expensive training and lack of interpretability. Therefore, in th…

2021

Hierarchical Graph Convolution Network for Traffic Forecasting

AAAI 2021technical

Traffic forecasting is attracting considerable interest due to its widespread application in intelligent transportation systems. Given the complex and dynamic traffic data, many methods focus on how to establish a spatial-temporal model to express the non-stationary traffic patterns. Recently, the l…

2021

How Framelets Enhance Graph Neural Networks

ICML 2021spotlight

This paper presents a new approach for assembling graph neural networks based on framelet transforms. The latter provides a multi-scale representation for graph-structured data. We decompose an input graph into low-pass and high-pass frequencies coefficients for network training, which then defines…

2021

On Riemannian Optimization over Positive Definite Matrices with the Bures-Wasserstein Geometry

NeurIPS 2021poster

In this paper, we comparatively analyze the Bures-Wasserstein (BW) geometry with the popular Affine-Invariant (AI) geometry for Riemannian optimization on the symmetric positive definite (SPD) matrix manifold. Our study begins with an observation that the BW metric has a linear dependence on SPD mat…

2019

Double Nuclear Norm Based Low Rank Representation on Grassmann Manifolds for Clustering

CVPR 2019poster

Unsupervised clustering for high-dimension data (such as imageset or video) is a hard issue in data processing and data mining area since these data always lie on a manifold (such as Grassmann manifold). Inspired of Low Rank representation theory, researchers proposed a series of effective clusterin…

Cited by 25PDFScholar
2016

Kernel Sparse Subspace Clustering on Symmetric Positive Definite Manifolds

CVPR 2016poster

Sparse subspace clustering (SSC), as one of the most successful subspace clustering methods, has achieved notable clustering accuracy in computer vision tasks. However, SSC applies only to vector data in Euclidean space. As such, there is still no satisfactory approach to solve subspace clustering…

Cited by 134PDFScholar
2016

Mixture of Bilateral-Projection Two-Dimensional Probabilistic Principal Component Analysis

CVPR 2016poster

The probabilistic principal component analysis (PPCA) is built upon a global linear mapping, with which it is insufficient to model complex data variation. This paper proposes a mixture of bilateral-projection probabilistic principal component analysis model (mixB2DPPCA) on 2D data. With multi-compo…

Cited by 5PDFScholar
2016

Proximal Riemannian Pursuit for Large-Scale Trace-Norm Minimization

CVPR 2016poster

Trace-norm regularization plays an important role in many areas such as machine learning and computer vision. Solving trace-norm regularized Trace-norm regularization plays an important role in many areas such as computer vision and machine learning. When solving general large-scale trace-norm regul…

Cited by 4PDFcodeScholar
2015

Learning Graph Structure for Multi-Label Image Classification via Clique Generation

CVPR 2015poster

Exploiting label dependency for multi-label image classification can significantly improve classification performance. Probabilistic Graphical Models are one of the primary methods for representing such dependencies. The structure of graphical models, however, is either determined heuristically or l…

Cited by 64SourcePDFScholar