EDBT 2026 Demo / reviewers in the wild / expert
Shaokai Ye
dblp:215/4368
· DBLP profile ↗
14ranked-venue papers
3as first author
7since 2021 · last 2023
0000-0003-4250-2220ORCID · corroborated
Domains — the database's venue-derived domains; a paper can count in several
Artificial intelligence and machine learning · 10 · 3 first-author · 7 since 2021Graphics, computer vision, multimedia, augmented reality and games · 6 · 2 first-author · 3 since 2021Systems, architecture and hardware · 4Software engineering, systems software and programming languages · 1
Expertise — from the expertise taxonomy: the topics of the expert's papers under the CCF categories. A weight counts papers with recency: 1 for a paper about the topic, 0.3 when the topic is its context, halved every five years.
| Artificial intelligence
9 papers |
Efficient and distributed learning · 36% Trustworthy machine learning · 27% Image recognition and object detection · 12% | |
| Computer architecture, parallel and distributed computing, and storage systems
3 papers |
Hardware accelerators and domain-specific architectures · 83% Memory systems · 17% | |
| Network and information security
2 papers |
Security and privacy of machine learning · 100% | |
| Databases, data mining, and information retrieval
1 paper |
Information retrieval · 100% | |
| Interdisciplinary, comprehensive, and emerging computing
1 paper |
Bioinformatics and computational biology · 100% | |
| Human-computer interaction and pervasive computing
1 paper |
Human-AI interaction · 100% |
Topics — the 27 heaviest of 28, each with the papers that count most for it
| Topic | Weight | Papers | Last | Evidence papers |
|---|---|---|---|---|
Machine learning › Efficient and distributed learning
model compression |
1.5 | 4 | 2020 | PCNN: Pattern-based Fine-Grained Regular Pruning Towards Optimizing CNN Accelerators · DAC 2020 Adversarial Robustness vs. Model Compression, or Both? · ICCV 2019 ADMM-NN: An Algorithm-Hardware Co-Design Framework of DNNs Using Alternating Direction Methods of Multipliers · ASPLOS 2019 |
Machine learning › Efficient and distributed learning › model compression › pruning
weight pruning |
1.5 | 4 | 2020 | PCNN: Pattern-based Fine-Grained Regular Pruning Towards Optimizing CNN Accelerators · DAC 2020 Adversarial Robustness vs. Model Compression, or Both? · ICCV 2019 ADMM-NN: An Algorithm-Hardware Co-Design Framework of DNNs Using Alternating Direction Methods of Multipliers · ASPLOS 2019 |
Machine learning › Trustworthy machine learning
robustness |
1.1 | 2 | 2022 | Enhance the Visual Representation via Discrete Adversarial Training · NeurIPS 2022 Adversarial Laser Beam: Effective Physical-World Attack to DNNs in a Blink · CVPR 2021 |
Security and privacy of machine learning
adversarial attack |
1.0 | 2 | 2021 | QAIR: Practical Query-Efficient Black-Box Attacks for Image Retrieval · CVPR 2021 Adversarial Laser Beam: Effective Physical-World Attack to DNNs in a Blink · CVPR 2021 |
Machine learning › Trustworthy machine learning › robustness
adversarial robustness |
1.0 | 2 | 2022 | Towards Robust Vision Transformer · CVPR 2022 Adversarial Robustness vs. Model Compression, or Both? · ICCV 2019 |
Hardware accelerators and domain-specific architectures
machine learning accelerator |
0.8 | 2 | 2020 | PIM-Prune: Fine-Grain DCNN Pruning for Crossbar-Based Process-In-Memory Architecture · DAC 2020 ADMM-NN: An Algorithm-Hardware Co-Design Framework of DNNs Using Alternating Direction Methods of Multipliers · ASPLOS 2019 |
Bioinformatics and computational biology
behavioral analysis |
0.7 | 1 | 2023 | AmadeusGPT: a natural language interface for interactive animal behavioral analysis · NeurIPS 2023 |
Human-AI interaction › large language model interaction › language-based interaction
natural language interface |
0.7 | 1 | 2023 | AmadeusGPT: a natural language interface for interactive animal behavioral analysis · NeurIPS 2023 |
Machine learning › Trustworthy machine learning › robustness › adversarial robustness
adversarial training |
0.6 | 1 | 2022 | Enhance the Visual Representation via Discrete Adversarial Training · NeurIPS 2022 |
Computer vision › Image recognition and object detection
image classification |
0.6 | 1 | 2022 | Enhance the Visual Representation via Discrete Adversarial Training · NeurIPS 2022 |
Computer vision › Image recognition and object detection › image classification
robust image classification |
0.6 | 1 | 2022 | Enhance the Visual Representation via Discrete Adversarial Training · NeurIPS 2022 |
Machine learning › Deep learning architectures and training › transformer
vision transformer |
0.6 | 1 | 2022 | Towards Robust Vision Transformer · CVPR 2022 |
Machine learning › Representation and self-supervised learning › representation learning
visual representation learning |
0.6 | 1 | 2022 | Enhance the Visual Representation via Discrete Adversarial Training · NeurIPS 2022 |
Information retrieval
adversarial retrieval |
0.5 | 1 | 2021 | QAIR: Practical Query-Efficient Black-Box Attacks for Image Retrieval · CVPR 2021 |
Information retrieval › trustworthy information retrieval
retrieval robustness |
0.5 | 1 | 2021 | QAIR: Practical Query-Efficient Black-Box Attacks for Image Retrieval · CVPR 2021 |
Security and privacy of machine learning › adversarial attack
physical adversarial attack |
0.5 | 1 | 2021 | Adversarial Laser Beam: Effective Physical-World Attack to DNNs in a Blink · CVPR 2021 |
Security and privacy of machine learning › adversarial attack › query-efficient attack
query-efficient black-box attack |
0.5 | 1 | 2021 | QAIR: Practical Query-Efficient Black-Box Attacks for Image Retrieval · CVPR 2021 |
Machine learning › Transfer learning and domain adaptation
domain adaptation |
0.4 | 1 | 2020 | Light-weight Calibrator: A Separable Component for Unsupervised Domain Adaptation · CVPR 2020 |
Machine learning › Transfer learning and domain adaptation › domain adaptation
unsupervised domain adaptation |
0.4 | 1 | 2020 | Light-weight Calibrator: A Separable Component for Unsupervised Domain Adaptation · CVPR 2020 |
Hardware accelerators and domain-specific architectures › machine learning accelerator
in-memory computing accelerator |
0.4 | 1 | 2020 | PIM-Prune: Fine-Grain DCNN Pruning for Crossbar-Based Process-In-Memory Architecture · DAC 2020 |
Hardware accelerators and domain-specific architectures › machine learning accelerator
neural network accelerator |
0.4 | 1 | 2020 | PCNN: Pattern-based Fine-Grained Regular Pruning Towards Optimizing CNN Accelerators · DAC 2020 |
Memory systems
processing-in-memory |
0.4 | 1 | 2020 | PIM-Prune: Fine-Grain DCNN Pruning for Crossbar-Based Process-In-Memory Architecture · DAC 2020 |
Hardware accelerators and domain-specific architectures › machine learning accelerator › CNN accelerator
sparse CNN accelerator |
0.4 | 1 | 2020 | PCNN: Pattern-based Fine-Grained Regular Pruning Towards Optimizing CNN Accelerators · DAC 2020 |
Machine learning › Efficient and distributed learning › model compression › quantization
weight quantization |
0.4 | 1 | 2019 | ADMM-NN: An Algorithm-Hardware Co-Design Framework of DNNs Using Alternating Direction Methods of Multipliers · ASPLOS 2019 |
Computer vision › Vision and language
multimodal understanding |
0.2 | 1 | 2023 | AmadeusGPT: a natural language interface for interactive animal behavioral analysis · NeurIPS 2023 |
Machine learning › Optimization for machine learning
non-convex optimization |
0.1 | 1 | 2019 | ADMM-NN: An Algorithm-Hardware Co-Design Framework of DNNs Using Alternating Direction Methods of Multipliers · ASPLOS 2019 |
Machine learning › Optimization for machine learning
alternating direction method of multipliers |
0.1 | 1 | 2018 | A Systematic DNN Weight Pruning Framework Using Alternating Direction Method of Multipliers · ECCV (8) 2018 |
Methods — techniques the papers use, named apart from their topics
large language model · 2.0dual-memory mechanism · 2.0alternating direction method of multipliers · 1.1relevance-based loss · 1.0recursive model stealing · 1.0gradient optimization · 1.0adversarial perturbation · 1.0self-attention · 0.6position-aware attention scaling · 0.6patch-wise augmentation · 0.6masked autoencoding · 0.6discrete adversarial training · 0.6VQGAN · 0.6laser beam manipulation · 0.5structural pruning · 0.4sparsity pattern mask · 0.4pattern-based pruning · 0.4fine-grain pruning · 0.4
| Year | Publication | Venue | Position |
|---|---|---|---|
| 2023 | AmadeusGPT: a natural language interface for interactive animal behavioral analysisabstractThe process of quantifying and analyzing animal behavior involves translating the naturally occurring descriptive language of their actions into machine-readable code. Yet, codifying behavior analysis is often challenging without deep understanding of animal behavior and technical machine learning knowledge. To limit this gap, we introduce AmadeusGPT: a natural language interface that turns natural language descriptions of behaviors into machine-executable code. Large-language models (LLMs) such as GPT3.5 and GPT4 allow for interactive language-based queries that are potentially well suited for making interactive behavior analysis. However, the comprehension capability of these LLMs is limited by the context window size, which prevents it from remembering distant conversations. To overcome the context window limitation, we implement a novel dual-memory mechanism to allow communication between short-term and long-term memory using symbols as context pointers for retrieval and saving. Concretely, users directly use language-based definitions of behavior and our augmented GPT develops code based on the core AmadeusGPT API, which contains machine learning, computer vision, spatio-temporal reasoning, and visualization modules. Users then can interactively refine results, and seamlessly add new behavioral modules as needed. We used the MABe 2022 behavior challenge tasks to benchmark AmadeusGPT and show excellent performance. Note, an end-user would not need to write any code to achieve this. Thus, collectively AmadeusGPT presents a novel way to merge deep biological knowledge, large-language models, and core computer vision modules into a more naturally intelligent system. Code and demos can be found at: https://github.com/AdaptiveMotorControlLab/AmadeusGPT Shaokai Ye, Jessy Lauer, Mu Zhou, Alexander Mathis, Mackenzie W. Mathis |
NeurIPS | 1 |
| 2022 | Towards Robust Vision TransformerabstractRecent advances on Vision Transformer (ViT) and its improved variants have shown that self-attention-based networks surpass traditional Convolutional Neural Networks (CNNs) in most vision tasks. However, existing ViTs focus on the standard accuracy and computation cost, lacking the investigation of the intrinsic influence on model robustness and generalization. In this work, we conduct systematic evaluation on components of ViTs in terms of their impact on robustness to adversarial examples, common corruptions and distribution shifts. We find some components can be harmful to robustness. By leveraging robust components as building blocks of ViTs, we propose Robust Vision Transformer (RVT), which is a new vision transformer and has superior performance with strong robustness. Inspired by the findings during the evaluation, we further propose two new plug-and-play techniques called position-aware attention scaling and patch-wise augmentation to augment our RVT, which we abbreviate as RVT*. The experimental results of RVT on ImageNet and six robustness benchmarks demonstrate its advanced robustness and generalization ability compared with previous ViTs and state-of-the-art CNNs. Furthermore, RVT-S* achieves Top-1 rank on multiple robustness leaderboards including ImageNet-C, ImageNet-Sketch and ImageNet-R. Xiaofeng Mao, Gege Qi, Yuefeng Chen, Ranjie Duan, Shaokai Ye, Yuan He 0011, Hui Xue 0001 |
CVPR | 6 |
| 2022 | Enhance the Visual Representation via Discrete Adversarial TrainingabstractAdversarial Training (AT), which is commonly accepted as one of the most effective approaches defending against adversarial examples, can largely harm the standard performance, thus has limited usefulness on industrial-scale production and applications. Surprisingly, this phenomenon is totally opposite in Natural Language Processing (NLP) task, where AT can even benefit for generalization. We notice the merit of AT in NLP tasks could derive from the discrete and symbolic input space. For borrowing the advantage from NLP-style AT, we propose Discrete Adversarial Training (DAT). DAT leverages VQGAN to reform the image data to discrete text-like inputs, i.e. visual words. Then it minimizes the maximal risk on such discrete images with symbolic adversarial perturbations. We further give an explanation from the perspective of distribution to demonstrate the effectiveness of DAT. As a plug-and-play technique for enhancing the visual representation, DAT achieves significant improvement on multiple tasks including image classification, object detection and self-supervised learning. Especially, the model pre-trained with Masked Auto-Encoding (MAE) and fine-tuned by our DAT without extra data can get 31.40 mCE on ImageNet-C and 32.77% top-1 accuracy on Stylized-ImageNet, building the new state-of-the-art. The code will be available at https://github.com/alibaba/easyrobust. Xiaofeng Mao, Yuefeng Chen, Ranjie Duan, Yao Zhu 0003, Gege Qi, Shaokai Ye, Rong Zhang 0006, Hui Xue 0001 |
NeurIPS | 6 |
| 2022 | Non-Structured DNN Weight Pruning - Is It Beneficial in Any Platform?abstractLarge deep neural network (DNN) models pose the key challenge to energy efficiency due to the significantly higher energy consumption of off-chip DRAM accesses than arithmetic or SRAM operations. It motivates the intensive research on model compression with two main approaches. Weight pruning leverages the redundancy in the number of weights and can be performed in a non-structured, which has higher flexibility and pruning rate but incurs index accesses due to irregular weights, or structured manner, which preserves the full matrix structure with a lower pruning rate. Weight quantization leverages the redundancy in the number of bits in weights. Compared to pruning, quantization is much more hardware-friendly and has become a "must-do" step for FPGA and ASIC implementations. Thus, any evaluation of the effectiveness of pruning should be on top of quantization. The key open question is, with quantization, what kind of pruning (non-structured versus structured) is most beneficial? This question is fundamental because the answer will determine the design aspects that we should really focus on to avoid the diminishing return of certain optimizations. This article provides a definitive answer to the question for the first time. First, we build ADMM-NN-S by extending and enhancing ADMM-NN, a recently proposed joint weight pruning and quantization framework, with the algorithmic supports for structured pruning, dynamic ADMM regulation, and masked mapping and retraining. Second, we develop a methodology for fair and fundamental comparison of non-structured and structured pruning in terms of both storage and computation efficiency. Our results show that ADMM-NN-S consistently outperforms the prior art: 1) it achieves 348× , 36× , and 8× overall weight pruning on LeNet-5, AlexNet, and ResNet-50, respectively, with (almost) zero accuracy loss and 2) we demonstrate the first fully binarized (for all layers) DNNs can be lossless in accuracy in many cases. These results provide a strong baseline and credibility of our study. Based on the proposed comparison framework, with the same accuracy and quantization, the results show that non-structured pruning is not competitive in terms of both storage and computation efficiency. Thus, we conclude that structured pruning has a greater potential compared to non-structured pruning. We encourage the community to focus on studying the DNN inference acceleration with structured sparsity. Sheng Lin 0001, Shaokai Ye, Zhezhi He, Linfeng Zhang 0001, Geng Yuan, Sia Huat Tan, Zhengang Li 0001, Deliang Fan, Xuehai Qian, Xue Lin 0001, Kaisheng Ma, Yanzhi Wang 0001 |
IEEE Trans. Neural Networks Learn. Syst. | 3 |
| 2022 | StructADMM: Achieving Ultrahigh Efficiency in Structured Pruning for DNNsabstractWeight pruning methods of deep neural networks (DNNs) have been demonstrated to achieve a good model pruning rate without loss of accuracy, thereby alleviating the significant computation/storage requirements of large-scale DNNs. Structured weight pruning methods have been proposed to overcome the limitation of irregular network structure and demonstrated actual GPU acceleration. However, in prior work, the pruning rate (degree of sparsity) and GPU acceleration are limited (to less than 50%) when accuracy needs to be maintained. In this work, we overcome these limitations by proposing a unified, systematic framework of structured weight pruning for DNNs. It is a framework that can be used to induce different types of structured sparsity, such as filterwise, channelwise, and shapewise sparsity, as well as nonstructured sparsity. The proposed framework incorporates stochastic gradient descent (SGD; or ADAM) with alternating direction method of multipliers (ADMM) and can be understood as a dynamic regularization method in which the regularization target is analytically updated in each iteration. Leveraging special characteristics of ADMM, we further propose a progressive, multistep weight pruning framework and a network purification and unused path removal procedure, in order to achieve higher pruning rate without accuracy loss. Without loss of accuracy on the AlexNet model, we achieve 2.58× and 3.65× average measured speedup on two GPUs, clearly outperforming the prior work. The average speedups reach 3.15× and 8.52× when allowing a moderate accuracy loss of 2%. In this case, the model compression for convolutional layers is 15.0× , corresponding to 11.93× measured CPU speedup. As another example, for the ResNet-18 model on the CIFAR-10 data set, we achieve an unprecedented 54.2× structured pruning rate on CONV layers. This is 32× higher pruning rate compared with recent work and can further translate into 7.6× inference time speedup on the Adreno 640 mobile GPU compared with the original, unpruned DNN model. We share our codes and models at the link http://bit.ly/2M0V7DO. Tianyun Zhang, Shaokai Ye, Xiaoyu Feng, Kaiqi Zhang 0003, Zhengang Li 0001, Jian Tang 0008, Sijia Liu 0001, Xue Lin 0001, Yongpan Liu, Makan Fardad, Yanzhi Wang 0001 |
IEEE Trans. Neural Networks Learn. Syst. | 2 |
| 2021 | Adversarial Laser Beam: Effective Physical-World Attack to DNNs in a BlinkabstractThough it is well known that the performance of deep neural networks (DNNs) degrades under certain light conditions, there exists no study on the threats of light beams emitted from some physical source as adversarial attacker on DNNs in a real-world scenario. In this work, we show by simply using a laser beam that DNNs are easily fooled. To this end, we propose a novel attack method called Adversarial Laser Beam (AdvLB), which enables manipulation of laser beam’s physical parameters to perform adversarial attack. Experiments demonstrate the effectiveness of our proposed approach in both digital- and physical-settings. We further empirically analyze the evaluation results and reveal that the proposed laser beam attack may lead to some interesting prediction errors of the state-of-the-art DNNs. We envisage that the proposed AdvLB method enriches the current family of adversarial attacks and builds the foundation for future robustness studies for light. Ranjie Duan, Xiaofeng Mao, A. K. Qin 0001, Yuefeng Chen, Shaokai Ye, Yuan He 0011, Yun Yang 0001 |
CVPR | 5 |
| 2021 | QAIR: Practical Query-Efficient Black-Box Attacks for Image RetrievalabstractWe study the query-based attack against image retrieval to evaluate its robustness against adversarial examples under the black-box setting, where the adversary only has query access to the top-k ranked unlabeled images from the database. Compared with query attacks in image classification, which produce adversaries according to the returned labels or confidence score, the challenge becomes even more prominent due to the difficulty in quantifying the attack effectiveness on the partial retrieved list. In this paper, we make the first attempt in Query-based Attack against Image Retrieval (QAIR), to completely subvert the top-k retrieval results. Specifically, a new relevance-based loss is designed to quantify the attack effects by measuring the set similarity on the top-k retrieval results before and after attacks and guide the gradient optimization. To further boost the attack efficiency, a recursive model stealing method is proposed to acquire transferable priors on the target model and generate the prior-guided gradients. Comprehensive experiments show that the proposed attack achieves a high attack success rate with few queries against the image retrieval systems under the black-box setting. The attack evaluations on the real-world visual search engine show that it successfully deceives a commercial system such as Bing Visual Search with 98% attack success rate by only 33 queries on average. Yuefeng Chen, Shaokai Ye, Yuan He 0011, Shuhui Wang, Hang Su 0006, Hui Xue 0001 |
CVPR | 4 |
| 2020 | Light-weight Calibrator: A Separable Component for Unsupervised Domain AdaptationabstractExisting domain adaptation methods aim at learning features that can be generalized among domains. These methods commonly require to update source classifier to adapt to the target domain and do not properly handle the trade-off between the source domain and the target domain. In this work, instead of training a classifier to adapt to the target domain, we use a separable component called data calibrator to help the fixed source classifier recover discrimination power in the target domain, while preserving the source domain's performance. When the difference between two domains is small, the source classifier's representation is sufficient to perform well in the target domain and outperforms GAN-based methods in digits. Otherwise, the proposed method can leverage synthetic images generated by GANs to boost performance and achieve state-of-the-art performance in digits datasets and driving scene semantic segmentation. Our method also empirically suggests the potential connection between domain adaptation and adversarial attacks. Shaokai Ye, Kailu Wu, Mu Zhou, Sia Huat Tan, Kaidi Xu, Jiebo Song, Chenglong Bao, Kaisheng Ma |
CVPR | 1 |
| 2020 | PIM-Prune: Fine-Grain DCNN Pruning for Crossbar-Based Process-In-Memory ArchitectureabstractDeep Convolution Neural network (DCNN) pruning is an efficient way to reduce the resource and power consumption in a DCNN accelerator. Exploiting the sparsity in the weight matrices of DCNNs, however, is nontrivial if we deploy these DC-NNs in a crossbar-based Process-In-Memory (PIM) architecture, because of the crossbar structure. Structural pruning-exploiting a coarse-grained sparsity, such as filter/channel-level pruning-can result in a compressed weight matrix that fits the crossbar structure. However, this pruning method inevitably degrades the model accuracy. To solve this problem, in this paper, we propose PIM-PRUNE to exploit the finer-grained sparsity in PIM-architecture, and the resulting compressed weight matrices can significantly reduce the demand of crossbars with negligible accuracy loss. Further, we explore the design space of the crossbar, such as the crossbar size and aspect-ratio, from a new point-of-view of resource-oriented pruning. We find a trade-off existing between the pruning algorithm and the hardware overhead: a PIM with smaller crossbars is more friendly for pruning methods; however, the resulting peripheral circuit cause higher power consumption. Given a specific DCNN, we can suggest a sweet-spot of crossbar design to the optimal overall energy efficiency. Experimental results show that the proposed pruning method applied on Resnet18 can achieve up to 24.85× and 3.56× higher compression rate of occupied crossbars on CifarlO and Imagenet, respectively; while the accuracy loss is negligible, which is 4.56× and 1.99× better than the state-of-art methods. Chaoqun Chu, Yanzhi Wang 0001, Yilong Zhao 0004, Shaokai Ye, Yunyan Hong, Xiaoyao Liang, Yinhe Han 0001, Li Jiang 0002 |
DAC | 5 |
| 2020 | PCNN: Pattern-based Fine-Grained Regular Pruning Towards Optimizing CNN AcceleratorsabstractWeight pruning is a powerful technique to realize model compression. We propose PCNN, a fine-grained regular 1D pruning method. A novel index format called Sparsity Pattern Mask (SPM) is presented to encode the sparsity in PCNN. Leveraging SPM with limited pruning patterns and non-zero sequences with equal length, PCNN can be efficiently employed in hardware. Evaluated on VGG-16 and ResNet-18, our PCNN achieves the compression rate up to 8.4× with only 0.2% accuracy loss. We also implement a pattern-aware architecture in 55nm process, achieving up to 9.0× speedup and 28.39 TOPS/W efficiency with only 3.1% on-chip memory overhead of indices. Zhanhong Tan, Jiebo Song, Sia Huat Tan, Hongyang Chen 0003, Yuanqing Miao, Yifu Wu, Shaokai Ye, Yanzhi Wang 0001, Dehui Li, Kaisheng Ma |
DAC | 8 |
| 2019 | ADMM-NN: An Algorithm-Hardware Co-Design Framework of DNNs Using Alternating Direction Methods of MultipliersabstractModel compression is an important technique to facilitate efficient embedded and hardware implementations of deep neural networks (DNNs), a number of prior works are dedicated to model compression techniques. The target is to simultaneously reduce the model storage size and accelerate the computation, with minor effect on accuracy. Two important categories of DNN model compression techniques are weight pruning and weight quantization. The former leverages the redundancy in the number of weights, whereas the latter leverages the redundancy in bit representation of weights. These two sources of redundancy can be combined, thereby leading to a higher degree of DNN model compression. However, a systematic framework of joint weight pruning and quantization of DNNs is lacking, thereby limiting the available model compression ratio. Moreover, the computation reduction, energy efficiency improvement, and hardware performance overhead need to be accounted besides simply model size reduction, and the hardware performance overhead resulted from weight pruning method needs to be taken into consideration. To address these limitations, we present ADMM-NN, the first algorithm-hardware co-optimization framework of DNNs using Alternating Direction Method of Multipliers (ADMM), a powerful technique to solve non-convex optimization problems with possibly combinatorial constraints. The first part of ADMM-NN is a systematic, joint framework of DNN weight pruning and quantization using ADMM. It can be understood as a smart regularization technique with regularization target dynamically updated in each ADMM iteration, thereby resulting in higher performance in model compression than the state-of-the-art. The second part is hardware-aware DNN optimizations to facilitate hardware-level implementations. We perform ADMM-based weight pruning and quantization considering (i) the computation reduction and energy efficiency improvement, and (ii) the hardware performance overhead due to irregular sparsity. The first requirement prioritizes the convolutional layer compression over fully-connected layers, while the latter requires a concept of the break-even pruning ratio, defined as the minimum pruning ratio of a specific layer that results in no hardware performance degradation. Without accuracy loss, ADMM-NN achieves 85× and 24× pruning on LeNet-5 and AlexNet models, respectively, --- significantly higher than the state-of-the-art. The improvements become more significant when focusing on computation reduction. Combining weight pruning and quantization, we achieve 1,910× and 231× reductions in overall model size on these two benchmarks, when focusing on data storage. Highly promising results are also observed on other representative DNNs such as VGGNet and ResNet-50. We release codes and models at https://github.com/yeshaokai/admm-nn. Ao Ren, Tianyun Zhang, Shaokai Ye, Wenyao Xu, Xuehai Qian, Xue Lin 0001, Yanzhi Wang 0001 |
ASPLOS | 3 |
| 2019 | ADMM-based Weight Pruning for Real-Time Deep Learning Acceleration on Mobile DevicesabstractDeep learning solutions are being increasingly deployed in mobile applications, at least for the inference phase. Due to the large model size and computational requirements, model compression for deep neural networks (DNNs) becomes necessary, especially considering the real-time requirement in embedded systems. In this paper, we extend the prior work on systematic DNN weight pruning using ADMM (Alternating Direction Method of Multipliers). We integrate ADMM regularization with masked mapping/retraining, thereby guaranteeing solution feasibility and providing high solution quality. Besides superior performance on representative DNN benchmarks (e.g., AlexNet, ResNet), we focus on two new applications facial emotion detection and eye tracking, and develop a top-down framework of DNN training, model compression, and acceleration in mobile devices. Experimental results show that with negligible accuracy degradation, the proposed method can achieve significant storage/memory reduction and speedup in mobile devices. Hongjia Li 0003, Ning Liu 0007, Sheng Lin 0001, Shaokai Ye, Tianyun Zhang, Xue Lin 0001, Wenyao Xu, Yanzhi Wang 0001 |
ACM Great Lakes Symposium on VLSI | 5 |
| 2019 | Adversarial Robustness vs. Model Compression, or Both?abstractIt is well known that deep neural networks (DNNs) are vulnerable to adversarial attacks, which are implemented by adding crafted perturbations onto benign examples. Min-max robust optimization based adversarial training can provide a notion of security against adversarial attacks. However, adversarial robustness requires a significantly larger capacity of the network than that for the natural training with only benign examples. This paper proposes a framework of concurrent adversarial training and weight pruning that enables model compression while still preserving the adversarial robustness and essentially tackles the dilemma of adversarial training. Furthermore, this work studies two hypotheses about weight pruning in the conventional setting and finds that weight pruning is essential for reducing the network model size in the adversarial setting; training a small model from scratch even with inherited initialization from the large model cannot achieve neither adversarial robustness nor high standard accuracy. Code is available at https://github.com/yeshaokai/Robustness-Aware-Pruning-ADMM. Shaokai Ye, Xue Lin 0001, Kaidi Xu, Sijia Liu 0001, Hao Cheng 0015, Jan-Henrik Lambrechts, Huan Zhang 0001, Aojun Zhou, Kaisheng Ma, Yanzhi Wang 0001 |
ICCV | 1 |
| 2018 | A Systematic DNN Weight Pruning Framework Using Alternating Direction Method of Multipliers
Tianyun Zhang, Shaokai Ye, Kaiqi Zhang 0003, Jian Tang 0008, Wujie Wen, Makan Fardad, Yanzhi Wang 0001 |
ECCV (8) | 2 |