← Search

Wei Niu

30 accepted papers

2025

AdaCM^2: On Understanding Extremely Long-Term Video with Adaptive Cross-Modality Memory Reduction

CVPR 2025highlight

The advancements in large language models (LLMs) have propelled the improvement of video understanding tasks by incorporating LLMs with visual models. However, most existing LLM-based models (e.g., VideoLLaMA, VideoChat) are constrained to processing short-duration videos. Recent attempts to underst…

Cited by 1SourcePDFScholar
2025

GaussianSpa: An "Optimizing-Sparsifying" Simplification Framework for Compact and High-Quality 3D Gaussian Splatting

CVPR 2025poster

3D Gaussian Splatting (3DGS) has emerged as a mainstream for novel view synthesis, leveraging continuous aggregations of Gaussian functions to model scene geometry. However, 3DGS suffers from substantial memory requirements to store the large amount of Gaussians, hindering its efficiency and practic…

Cited by 2SourcePDFScholar
2025

LazyDiT: Lazy Learning for the Acceleration of Diffusion Transformers

AAAI 2025technical

Diffusion Transformers have emerged as the preeminent models for a wide array of generative tasks, demonstrating superior performance and efficacy across various applications. The promising results come at the cost of slow inference, as each denoising step requires running the whole transformer mode…

2025

Learning LLM Preference over Intra-Dialogue Pairs: A Framework for Utterance-level Understandings

NAACL 2025industry

Large language models (LLMs) have demonstrated remarkable capabilities in handling complex dialogue tasks without requiring use case-specific fine-tuning. However, analyzing live dialogues in real-time necessitates low-latency processing systems, making it impractical to deploy models with billions…

Cited by 0SourcePDFScholar
2025

QuartDepth: Post-Training Quantization for Real-Time Depth Estimation on the Edge

CVPR 2025poster

Monocular Depth Estimation (MDE) has emerged as a pivotal task in computer vision, supporting numerous real-world applications. However, deploying accurate depth estimation models on resource-limited edge devices, especially Application-Specific Integrated Circuits (ASICs), is challenging due to the…

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

Sparse Learning for State Space Models on Mobile

ICLR 2025poster

Transformer models have been widely investigated in different domains by providing long-range dependency handling and global contextual awareness, driving the development of popular AI applications such as ChatGPT, Gemini, and Alexa. State Space Models (SSMs) have emerged as strong contenders in the…

Cited by 1SourcePDFScholar
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

Data Overfitting for On-Device Super-Resolution with Dynamic Algorithm and Compiler Co-Design

ECCV 2024poster

"Deep neural networks (DNNs) are frequently employed in a variety of computer vision applications. Nowadays, an emerging trend in the current video distribution system is to take advantage of DNN’s overfitting properties to perform video resolution upscaling. By splitting videos into chunks and appl…

2024

Exploring Token Pruning in Vision State Space Models

NeurIPS 2024poster

State Space Models (SSMs) have the advantage of keeping linear computational complexity compared to attention modules in transformers, and have been applied to vision tasks as a new type of powerful vision foundation model. Inspired by the observations that the final prediction in vision transformer…

Cited by 6SourcePDFScholar
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

NeurRev: Train Better Sparse Neural Network Practically via Neuron Revitalization

ICLR 2024poster

Dynamic Sparse Training (DST) employs a greedy search mechanism to identify an optimal sparse subnetwork by periodically pruning and growing network connections during training. To guarantee effectiveness, DST algorithms rely on high search frequency, which consequently, requires large learning rate…

Cited by 3SourcePDFScholar
2024

Real-time Core-Periphery Guided ViT with Smart Data Layout Selection on Mobile Devices

NeurIPS 2024poster

Mobile devices have become essential enablers for AI applications, particularly in scenarios that require real-time performance. Vision Transformer (ViT) has become a fundamental cornerstone in this regard due to its high accuracy. Recent efforts have been dedicated to developing various transformer…

Cited by 0SourcePDFScholar
2023

Pruning Parameterization With Bi-Level Optimization for Efficient Semantic Segmentation on the Edge

CVPR 2023poster

With the ever-increasing popularity of edge devices, it is necessary to implement real-time segmentation on the edge for autonomous driving and many other applications. Vision Transformers (ViTs) have shown considerably stronger results for many vision tasks. However, ViTs with the full-attention me…

Cited by 28SourcePDFScholar
2023

Towards High-Quality and Efficient Video Super-Resolution via Spatial-Temporal Data Overfitting

CVPR 2023highlight

As deep convolutional neural networks (DNNs) are widely used in various fields of computer vision, leveraging the overfitting ability of the DNN to achieve video resolution upscaling has become a new trend in the modern video delivery system. By dividing videos into chunks and overfitting each chunk…

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

Compiler-Aware Neural Architecture Search for On-Mobile Real-Time Super-Resolution

ECCV 2022poster

"Deep learning-based super-resolution (SR) has gained tremendous popularity in recent years because of its high image quality performance and wide application scenarios. However, prior methods typically suffer from large amounts of computations and huge power consumption, causing difficulties for re…

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…

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

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

RT3D: Achieving Real-Time Execution of 3D Convolutional Neural Networks on Mobile Devices

AAAI 2021technical

Mobile devices are becoming an important carrier for deep learning tasks, as they are being equipped with powerful, high-end mobile CPUs and GPUs. However, it is still a challenging task to execute 3D Convolutional Neural Networks (CNNs) targeting for real-time performance, besides high inference ac…

Cited by 14SourcePDFScholar
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…

2020

An Image Enhancing Pattern-based Sparsity for Real-time Inference on Mobile Devices

ECCV 2020poster

Weight pruning has been widely acknowledged as a straightforward and effective method to eliminate redundancy in Deep Neural Networks (DNN), thereby achieving acceleration on various platforms. However, most of the pruning techniques are essentially trade-offs between model accuracy and regularity w…

2020

Towards Real-Time DNN Inference on Mobile Platforms with Model Pruning and Compiler Optimization

IJCAI 2020poster

High-end mobile platforms rapidly serve as primary computing devices for a wide range of Deep Neural Network (DNN) applications. However, the constrained computation and storage resources on these devices still pose significant challenges for real-time DNN inference executions. To address this probl…