← Search

Geng Yuan

33 accepted papers

2026

Forget-It-All: Multi-Concept Machine Unlearning via Concept-Aware Neuron Masking

ICML 2026poster

The widespread adoption of text-to-image (T2I) diffusion models has raised concerns about their potential to generate copyrighted, inappropriate, or sensitive imagery learned from massive training corpora. As a practical solution, machine unlearning aims to selectively erase unwanted concepts from a…

Cited by 0SourceScholar
2026

Roots Beneath the Cut: Uncovering the Risk of Concept Revival in Pruning-Based Unlearning for Diffusion Models

CVPR 2026

Pruning-based unlearning has recently emerged as a fast, training-free, and data-independent approach to remove undesired concepts from diffusion models. It promises high efficiency and robustness, offering an attractive alternative to traditional fine-tuning or editing-based unlearning. However, in

Cited by 0SourcecodeScholar
2025

FairSMOE: Mitigating Multi-Attribute Fairness Problem with Sparse Mixture-of-Experts

IJCAI 2025

Real‐world datasets usually contain multiple attributes, making it essential to ensure fairness across all of them simultaneously. However, different attributes may vary in difficulty, and no existing approaches have effectively addressed this issue. Consequently, an attribute‐adaptive strategy is n

Cited by 0SourcePDFScholar
2025

Harmony in Divergence: Towards Fast, Accurate, and Memory-efficient Zeroth-order LLM Fine-tuning

NeurIPS 2025poster

Large language models (LLMs) excel across various tasks, but standard first-order (FO) fine-tuning demands considerable memory, significantly limiting real-world deployment. Recently, zeroth-order (ZO) optimization stood out as a promising memory-efficient training paradigm, avoiding backward passes…

Cited by 0SourcecodeScholar
2025

Mutual Effort for Efficiency: A Similarity-based Token Pruning for Vision Transformers in Self-Supervised Learning

ICLR 2025poster

Self-supervised learning (SSL) offers a compelling solution to the challenge of extensive labeled data requirements in traditional supervised learning. With the proven success of Vision Transformers (ViTs) in supervised tasks, there is increasing interest in adapting them for SSL frameworks. However…

Cited by 0SourcePDFScholar
2025

RoRA: Efficient Fine-Tuning of LLM with Reliability Optimization for Rank Adaptation

ICASSP 2025accepted

Fine-tuning helps large language models (LLM) recover degraded information and enhance task performance. Although Low-Rank Adaptation (LoRA) is widely used and effective for fine-tuning, we have observed that its scaling factor can limit or even reduce performance as the rank size increases. To addr…

Cited by 0SourceScholar
2025

Toward Adaptive Large Language Models Structured Pruning via Hybrid-grained Weight Importance Assessment

AAAI 2025technical

Structured pruning for large language models (LLMs) has garnered significant academic interest due to its ability to efficiently compress and accelerate LLMs by eliminating redundant weight groups at a coarse-grained granularity. Current structured pruning methods for LLMs typically depend on a sing…

2024

Advancing Dynamic Sparse Training by Exploring Optimization Opportunities

ICML 2024poster

Dynamic Sparse Training (DST) is an effective approach for addressing the substantial training resource requirements posed by the ever-increasing size of the Deep Neural Networks (DNNs). Characterized by its dynamic "train-prune-grow'' schedule during training, DST implicitly develops a bi-level str…

2024

Fast and Memory-Efficient Video Diffusion Using Streamlined Inference

NeurIPS 2024poster

The rapid progress in artificial intelligence-generated content (AIGC), especially with diffusion models, has significantly advanced development of high-quality video generation. However, current video diffusion models exhibit demanding computational requirements and high peak memory usage, especial…

2024

Waxing-and-Waning: a Generic Similarity-based Framework for Efficient Self-Supervised Learning

ICLR 2024poster

Deep Neural Networks (DNNs), essential for diverse applications such as visual recognition and eldercare, often require a large amount of labeled data for training, making widespread deployment of DNNs a challenging task. Self-supervised learning (SSL) emerges as a promising approach, which leverage…

Cited by 5SourcePDFScholar
2023

Data Level Lottery Ticket Hypothesis for Vision Transformers

IJCAI 2023poster

The conventional lottery ticket hypothesis (LTH) claims that there exists a sparse subnetwork within a dense neural network and a proper random initialization method, called the winning ticket, such that it can be trained from scratch to almost as good as the dense counterpart. Meanwhile, the resear…

2023

HotBEV: Hardware-oriented Transformer-based Multi-View 3D Detector for BEV Perception

NeurIPS 2023poster

The bird's-eye-view (BEV) perception plays a critical role in autonomous driving systems, involving the accurate and efficient detection and tracking of objects from a top-down perspective. To achieve real-time decision-making in self-driving scenarios, low-latency computation is essential. While re…

Cited by 5SourcePDFScholar
2023

PackQViT: Faster Sub-8-bit Vision Transformers via Full and Packed Quantization on the Mobile

NeurIPS 2023poster

While Vision Transformers (ViTs) have undoubtedly made impressive strides in computer vision (CV), their intricate network structures necessitate substantial computation and memory resources. A decision-making process for CV tasks typically entails performing computations with low latency, which is…

Cited by 21SourcePDFScholar
2023

Peeling the Onion: Hierarchical Reduction of Data Redundancy for Efficient Vision Transformer Training

AAAI 2023technical

Vision transformers (ViTs) have recently obtained success in many applications, but their intensive computation and heavy memory usage at both training and inference time limit their generalization. Previous compression algorithms usually start from the pre-trained dense models and only focus on eff…

2023

Self-Ensemble Protection: Training Checkpoints Are Good Data Protectors

ICLR 2023poster

As data becomes increasingly vital, a company would be very cautious about releasing data, because the competitors could use it to train high-performance models, thereby posing a tremendous threat to the company's commercial competence. To prevent training good models on the data, we could add imper…

2023

SmartFRZ: An Efficient Training Framework using Attention-Based Layer Freezing

ICLR 2023top-25%

There has been a proliferation of artificial intelligence applications, where model training is key to promising high-quality services for these applications. However, the model training process is both time-intensive and energy-intensive, inevitably affecting the user's demand for application effic…

Cited by 21SourcePDFScholar
2023

Towards Real-Time Segmentation on the Edge

AAAI 2023technical

The research in real-time segmentation mainly focuses on desktop GPUs. However, autonomous driving and many other applications rely on real-time segmentation on the edge, and current arts are far from the goal. In addition, recent advances in vision transformers also inspire us to re-design the ne…

Cited by 14SourcePDFScholar
2022

EfficientFormer: Vision Transformers at MobileNet Speed

NeurIPS 2022accept

Vision Transformers (ViT) have shown rapid progress in computer vision tasks, achieving promising results on various benchmarks. However, due to the massive number of parameters and model design, e.g., attention mechanism, ViT-based models are generally times slower than lightweight convolutional n…

Cited by 453SourcePDFScholar
2022

Layer Freezing & Data Sieving: Missing Pieces of a Generic Framework for Sparse Training

NeurIPS 2022accept

Recently, sparse training has emerged as a promising paradigm for efficient deep learning on edge devices. The current research mainly devotes the efforts to reducing training costs by further increasing model sparsity. However, increasing sparsity is not always ideal since it will inevitably introd…

2022

Pruning-as-Search: Efficient Neural Architecture Search via Channel Pruning and Structural Reparameterization

IJCAI 2022poster

Neural architecture search (NAS) and network pruning are widely studied efficient AI techniques, but not yet perfect. NAS performs exhaustive candidate architecture search, incurring tremendous search cost. Though (structured) pruning can simply shrink model dimension, it remains unclear how to de…

Cited by 48SourcePDFScholar
2022

SPViT: Enabling Faster Vision Transformers via Latency-Aware Soft Token Pruning

ECCV 2022poster

"Recently, Vision Transformer (ViT) has continuously established new milestones in the computer vision field, while the high computation and memory cost makes its propagation in industrial production difficult. Considering the computation complexity, the internal data pattern of ViTs, and the edge d…

2022

SparCL: Sparse Continual Learning on the Edge

NeurIPS 2022accept

Existing work in continual learning (CL) focuses on mitigating catastrophic forgetting, i.e., model performance deterioration on past tasks when learning a new task. However, the training efficiency of a CL system is under-investigated, which limits the real-world application of CL systems under res…

2022

You Already Have It: A Generator-Free Low-Precision DNN Training Framework Using Stochastic Rounding

ECCV 2022poster

"Stochastic rounding is a critical technique used in low-precision deep neural networks (DNNs) training to ensure good model accuracy. However, it requires a large number of random numbers generated on the fly. This is not a trivial task on the hardware platforms such as FPGA and ASIC. The widely us…

Cited by 5SourcePDFScholar
2021

A Compression-Compilation Framework for On-mobile Real-time BERT Applications

IJCAI 2021poster

Transformer-based deep learning models have increasingly demonstrated high accuracy on many natural language processing (NLP) tasks. In this paper, we propose a compression-compilation co-design framework that can guarantee the identified model meets both resource and real-time specifications of mob…

Cited by 4SourcePDFScholar
2021

Achieving On-Mobile Real-Time Super-Resolution With Neural Architecture and Pruning Search

ICCV 2021poster

Though recent years have witnessed remarkable progress in single image super-resolution (SISR) tasks with the prosperous development of deep neural networks (DNNs), the deep learning methods are confronted with the computation and memory consumption issues in practice, especially for resource-limite…

Cited by 63PDFcodeScholar
2021

Lottery Ticket Preserves Weight Correlation: Is It Desirable or Not?

ICML 2021spotlight

In deep model compression, the recent finding "Lottery Ticket Hypothesis" (LTH) pointed out that there could exist a winning ticket (i.e., a properly pruned sub-network together with original weight initialization) that can achieve competitive performance than the original dense network. However, it…

Cited by 38SourcePDFScholar
2021

MEST: Accurate and Fast Memory-Economic Sparse Training Framework on the Edge

NeurIPS 2021spotlight

Recently, a new trend of exploring sparsity for accelerating neural network training has emerged, embracing the paradigm of training on the edge. This paper proposes a novel Memory-Economic Sparse Training (MEST) framework targeting for accurate and fast execution on edge devices. The proposed MEST…

2021

NPAS: A Compiler-Aware Framework of Unified Network Pruning and Architecture Search for Beyond Real-Time Mobile Acceleration

CVPR 2021poster

With the increasing demand to efficiently deploy DNNs on mobile edge devices, it becomes much more important to reduce unnecessary computation and increase the execution speed. Prior methods towards this goal, including model compression and network architecture search (NAS), are largely performed i…

Cited by 34PDFcodeScholar
2021

Sanity Checks for Lottery Tickets: Does Your Winning Ticket Really Win the Jackpot?

NeurIPS 2021poster

There have been long-standing controversies and inconsistencies over the experiment setup and criteria for identifying the "winning ticket" in literature. To reconcile such, we revisit the definition of lottery ticket hypothesis, with comprehensive and more rigorous conditions. Under our new definit…

2021

Teachers Do More Than Teach: Compressing Image-to-Image Models

CVPR 2021poster

Generative Adversarial Networks (GANs) have achieved huge success in generating high-fidelity images, however, they suffer from low efficiency due to tremendous computational cost and bulky memory usage. Recent efforts on compression GANs show noticeable progress in obtaining smaller generators by s…

Cited by 73PDFcodeScholar
2021

Towards Fast and Accurate Multi-Person Pose Estimation on Mobile Devices

IJCAI 2021poster

The rapid development of autonomous driving, abnormal behavior detection, and behavior recognition makes an increasing demand for multi-person pose estimation-based applications, especially on mobile platforms. However, to achieve high accuracy, state-of-the-art methods tend to have a large model si…

Cited by 11SourcePDFScholar
2021

YOLObile: Real-Time Object Detection on Mobile Devices via Compression-Compilation Co-Design

AAAI 2021technical

The rapid development and wide utilization of object detection techniques have aroused attention on both accuracy and speed of object detectors. However, the current state-of-the-art object detection works are either accuracy-oriented using a large model but leading to high latency or speed-oriented…