← Search

Shuai Zheng

20 accepted papers

2026

Subspace-Aware Feature Reshaping for Open-Set Graph Class-Incremental Learning

ICML 2026poster

Graph class-incremental learning (GCIL) has emerged to address the challenge of learning from dynamically evolving graphs, which continuously learns new classes over a sequence of tasks while retaining performance on previously seen classes. However, existing GCIL methods assume a closed-set test di…

Cited by 0SourceScholar
2025

A Pioneering Neural Network Method for Efficient and Robust Fuel Sloshing Simulation in Aircraft

AAAI 2025technical

Simulating fuel sloshing within aircraft tanks during flight is crucial for aircraft safety research. Traditional methods based on Navier-Stokes equations are computationally expensive. In this paper, we treat fluid motion as point cloud transformation and propose the first neural network method spe…

Cited by 1SourcePDFScholar
2025

Towards Pre-trained Graph Condensation via Optimal Transport

NeurIPS 2025poster

Graph condensation (GC) aims to distill the original graph into a small-scale graph, mitigating redundancy and accelerating GNN training. However, conventional GC approaches heavily rely on rigid GNNs and task-specific supervision. Such a dependency severely restricts their reusability and generaliz…

Cited by 0SourceScholar
2024

Endow SAM with Keen Eyes: Temporal-spatial Prompt Learning for Video Camouflaged Object Detection

CVPR 2024poster

The Segment Anything Model (SAM) a prompt-driven foundational model has demonstrated remarkable performance in natural image segmentation. However its application in video camouflaged object detection (VCOD) encounters challenges chiefly stemming from the overlooked temporal-spatial associations and…

Cited by 11SourcePDFScholar
2023

Disentangling Orthogonal Planes for Indoor Panoramic Room Layout Estimation With Cross-Scale Distortion Awareness

CVPR 2023poster

Based on the Manhattan World assumption, most existing indoor layout estimation schemes focus on recovering layouts from vertically compressed 1D sequences. However, the compression procedure confuses the semantics of different planes, yielding inferior performance with ambiguous interpretability. T…

2023

Prompt Pre-Training with Twenty-Thousand Classes for Open-Vocabulary Visual Recognition

NeurIPS 2023poster

This work proposes POMP, a prompt pre-training method for vision-language models. Being memory and computation efficient, POMP enables the learned prompt to condense semantic information for a rich set of visual concepts with over twenty-thousand classes. Once pre-trained, the prompt with a strong t…

2023

VCC: Scaling Transformers to 128K Tokens or More by Prioritizing Important Tokens

NeurIPS 2023poster

Transformers are central in modern natural language processing and computer vision applications. Despite recent works devoted to reducing the quadratic cost of such models with respect to sequence length, dealing with ultra long sequences (e.g., $>$16K tokens) remains challenging. Applications such…

2022

Partial and Asymmetric Contrastive Learning for Out-of-Distribution Detection in Long-Tailed Recognition

ICML 2022oral

Existing out-of-distribution (OOD) detection methods are typically benchmarked on training sets with balanced class distributions. However, in real-world applications, it is common for the training sets to have long-tailed distributions. In this work, we first demonstrate that existing OOD detection…

2022

Removing Batch Normalization Boosts Adversarial Training

ICML 2022spotlight

Adversarial training (AT) defends deep neural networks against adversarial attacks. One challenge that limits its practical application is the performance degradation on clean samples. A major bottleneck identified by previous works is the widely used batch normalization (BN), which struggles to mod…

2020

A New Multihypothesis Prediction Scheme for Compressed Video Sensing Reconstruction

ICASSP 2020accepted

For multihypothesis-based compressed video sensing schemes, the low accuracy of weight prediction and degradation of recovery quality for high-motion videos are open challenges. To solve this problem, this paper proposes a new multihypothesis prediction scheme. To efficiently get high-quality hypoth…

Cited by 0SourceScholar
2020

CSER: Communication-efficient SGD with Error Reset

NeurIPS 2020poster

The scalability of Distributed Stochastic Gradient Descent (SGD) is today limited by communication bottlenecks. We propose a novel SGD variant: \underline{C}ommunication-efficient \underline{S}GD with \underline{E}rror \underline{R}eset, or \underline{CSER}. The key idea in CSER is first a new techn…

Cited by 50SourcePDFScholar
2020

Discriminant Generative Adversarial Networks with its Application to Equipment Health Classification

ICASSP 2020accepted

In equipment health classification, machines in normal, degradation and critical stages are classified based on domain experts KPI (Remaining Useful Life). Higher KPI values indicate healthier machines. GANs can be used to generate sensor data for machines in different health stages. There are chall…

Cited by 0SourceScholar
2020

Distribution-Induced Bidirectional Generative Adversarial Network for Graph Representation Learning

CVPR 2020poster

Graph representation learning aims to encode all nodes of a graph into low-dimensional vectors that will serve as input of many computer vision tasks. However, most existing algorithms ignore the existence of inherent data distribution and even noises. This may significantly increase the phenomenon…

Cited by 48PDFcodeScholar
2019

Communication-Efficient Distributed Blockwise Momentum SGD with Error-Feedback

NeurIPS 2019poster

Communication overhead is a major bottleneck hampering the scalability of distributed machine learning systems. Recently, there has been a surge of interest in using gradient compression to improve the communication efficiency of distributed neural network training. Using 1-bit quantization, signSG…

2018

Conditional Image-Text Embedding Networks

ECCV 2018poster

This paper presents an approach for grounding phrases in images which jointly learns multiple text-conditioned embeddings in a single end-to-end model. In order to differentiate text phrases into semantically distinct subspaces, we propose a concept weight branch that automatically assigns phrases t…

2015

Conditional Random Fields as Recurrent Neural Networks

ICCV 2015poster

Pixel-level labelling tasks, such as semantic segmentation, play a central role in image understanding. Recent approaches have attempted to harness the capabilities of deep learning techniques for image recognition to tackle pixel-level labelling tasks. One central issue in this methodology is the l…

Cited by 3378PDFcodeScholar