← Search

Zhenglun Kong

23 accepted papers

2026

SPATIA: Multimodal Generation and Prediction of Spatial Cell Phenotypes

ICML 2026poster

Understanding how cellular morphology, gene expression, and spatial context jointly shape tissue function is a central challenge in biology. Image-based spatial transcriptomics technologies now provide high-resolution measurements of cell images and gene expression profiles, but existing methods typ…

Cited by 0SourceScholar
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

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

Agile-Quant: Activation-Guided Quantization for Faster Inference of LLMs on the Edge

AAAI 2024technical

Large Language Models (LLMs) stand out for their impressive performance in intricate language modeling tasks. However, their demanding computational and memory needs pose obstacles for broad use on edge devices. Quantization is then introduced to boost LLMs' on-device efficiency. Recent works show t…

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

FasterVD: On Acceleration of Video Diffusion Models

IJCAI 2024poster

Equipped with Denoising Diffusion Probabilistic Models, video content generation has gained significant research interest recently. However, diffusion pipelines call for intensive computation and model storage, which poses challenges for their wide and efficient deployment. In this work, we address…

Cited by 0SourcePDFScholar
2024

Pruning Foundation Models for High Accuracy without Retraining

EMNLP 2024finding

Despite the superior performance, it is challenging to deploy large language models (LLMs) due to their massive parameters and computations. While pruning is a promising technique to reduce model size and accelerate the inference, the traditional pruning techniques can hardly be applied for LLMs as…

2024

Rethinking Token Reduction for State Space Models

EMNLP 2024main

Recent advancements in State Space Models (SSMs) have attracted significant interest, particularly in models optimized for parallel training and handling long-range dependencies. Architectures like Mamba have scaled to billions of parameters with selective SSM. To facilitate broader applications usi…

2024

Search for Efficient Large Language Models

NeurIPS 2024poster

Large Language Models (LLMs) have long held sway in the realms of artificial intelligence research. Numerous efficient techniques, including weight pruning, quantization, and distillation, have been embraced to compress LLMs, targeting memory reduction and inference acceleration, which underscore th…

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

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

SpeedDETR: Speed-aware Transformers for End-to-end Object Detection

ICML 2023poster

Vision Transformers (ViTs) have continuously achieved new milestones in object detection. However, the considerable computation and memory burden compromise their efficiency and generalization of deployment on resource-constraint devices. Besides, efficient transformer-based detectors designed by ex…

Cited by 3SourcePDFScholar
2023

You Need Multiple Exiting: Dynamic Early Exiting for Accelerating Unified Vision Language Model

CVPR 2023poster

Large-scale transformer models bring significant improvements for various downstream vision language tasks with a unified architecture. The performance improvements come with increasing model size, resulting in slow inference speed and increased cost for severing. While some certain predictions bene…

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

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

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

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