Peer Reviewed Conferences and Journals

GWLZ: A Group-wise Learning-based Lossy Compression Framework for Scientific Data

Wenqi Jia, Sian Jin, Jinzhen Wang, Wei Niu, Dingwen Tao, Miao Yin

14TH Workshop on AI and Scientific Computing at Scale using Flexible Computing Infrastructures

MobiTT: Speed-Oriented DNN Compression for Mobile with Tensor Train Decomposition

Miao Yin, Wei Niu

Workshop on Compute Platforms for Autonomous Vehicles (held with ASPLOS'24) [CAV'24]

SmartMem: Layout Transformation Elimination and Adaptation for Efficient DNN Execution on Mobile

Wei Niu, Md Musfiqur Rahman Sanim, Zhihao Shu, Jiexiong Guan, Xipeng Shen, Miao Yin, Gagan Agrawal, Bin Ren

International Conference on Architectural Support for Programming Languages and Operating Systems [ASPLOS'24 Fall]

NeurRev: Train Better Sparse Neural Network Practically via Neuron Revitalization

Gen Li, Lu Yin, Jie Ji, Wei Niu, Minghai Qin, Bin Ren, Linke Guo, Shiwei Liu, Xiaolong Ma

The Twelfth International Conference on Learning Representations [ICLR'24]

A Scalable Real-time Semantic Segmentation Network for Autonomous Driving

Jun Liu, Chao Wu, Geng Yuan, Wei Niu, Wenbin Zhang, Houbing Herbert Song

Advanced Multimedia Computing for Smart Manufacturing and Engineering [AMC-SME'23]

SoD²: Statically Optimizing Dynamic Deep Neural Network Execution

Wei Niu, Gagan Agrawal, Bin Ren

International Conference on Architectural Support for Programming Languages and Operating Systems [ASPLOS'24 Spring]

Achieving Real-time DNN Execution on Mobile Devices with Compiler Optimizations

Wei Niu

Thesis Thesis

Decentralized Application-Level Adaptive Scheduling for Multi-Instance DNNs on Open Mobile Devices

Hsin-Hsuan Sung, Jiexiong Guan, Wei Niu, Jou-An Chen, Bin Ren, Xipeng Shen

2023 USENIX Annual Technical Conference [USENIX ATC'23]

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

Gen Li, Jie Ji, Minghai Qin, Wei Niu, Bin Ren, Fatemeh Afghah, Linke Guo, Xiaolong Ma

IEEE/CVF Conference on Computer Vision and Pattern Recognition, 2023 (acceptance rate: 2360/9155) [CVPR'23]

Pruning Parameterization with Bi-level Optimization for Efficient Semantic Segmentation on the Edge

Changdi Yang, Pu Zhao, Yanyu Li, Wei Niu, Jiexiong Guan, Hao Tang, Minghai Qin, Bin Ren, Xue Lin, Yanzhi Wang

IEEE/CVF Conference on Computer Vision and Pattern Recognition, 2023 (acceptance rate: 2360/9155) [CVPR'23]

Towards Real-Time Segmentation on the Edge

Yanyu Li, Changdi Yang, Pu Zhao, Geng Yuan, Wei Niu, Jiexiong Guan, Hao Tang, Minghai Qin, Qing Jin, Bin Ren, Xue Lin, Yanzhi Wang

Thirty-Seventh AAAI Conference on Artificial Intelligence [AAAI'23]

SparCL: Sparse Continual Learning on the Edge

Zifeng Wang, Zheng Zhan, Yifan Gong, Geng Yuan, Wei Niu, Tong Jian, Bin Ren, Stratis Ioannidis, Yanzhi Wang, Jennifer Dy

Thirty-sixth Conference on Neural Information Processing Systems [NeurIPS'22]

Survey: Exploiting Data Redundancy for Optimization of Deep Learning

Jou-an Chen, Wei Niu, Bin Ren, Yanzhi Wang, Xipeng Shen

ACM Computing Surveys [CSUR]

GCD²: A Globally Optimizing Compiler for Mapping DNNs to Mobile DSPs

Wei Niu, Jiexiong Guan, Xipeng Shen, Yanzhi Wang, Gagan Agrawal, Bin Ren

55th IEEE/ACM International Symposium on Microarchitecture (Acceptance Rate: 83/369 = 22.5%, to appear) [MICRO'22]

SPViT: Enabling Faster Vision Transformers via Soft Token Pruning

Zhenglun Kong, Peiyan Dong, Xiaolong Ma, Xin Meng, Wei Niu, Mengshu Sun, Xuan Shen, Geng Yuan, Bin Ren, Minghai Qin, Hao Tang, Yanzhi Wang

Proceeding of European Conference on Computer Vision, 2022 (to appear) [ECCV'22]

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

Yushu Wu, Yifan Gong, Pu Zhao, Yanyu Li, Zheng Zhan, Wei Niu, Hao Tang, Minghai Qin, Bin Ren, Yanzhi Wang

Proceeding of European Conference on Computer Vision, 2022 (to appear) [ECCV'22]

Automatic Mapping of the Best-Suited DNN Pruning Schemes for Real-Time Mobile Acceleration

Yifan Gong, Geng Yuan, Zheng Zhan, Wei Niu, Zhengang Li, Pu Zhao, Yuxuan Cai, Sijia Liu, Bin Ren, Xue Lin, Xulong Tang, Yanzhi Wang

ACM Transactions on Design Automation of Electronic Systems [TDAES'22]

Mobile or FPGA? A Comprehensive Evaluation on Energy Efficiency and a Unified Optimization Framework

Geng Yuan, Mengshu Sun, Wei Niu, Zhengang Li, Yuxuan Cai, Yanyu Li, Jun Liu, Weiwen Jiang, Xue Lin, Bin Ren, Xulong Tang, Yanzhi Wang

ACM Transactions on Embedded Computing Systems (TECS) [TECS'22]

Brief Industry Paper: Enabling Level-4 Autonomous Driving on a Single $1k Off-the-Shelf Card

Hsin-Hsuan Sung, Yuanchao Xu, Jiexiong Guan, Wei Niu, Yanzhi Wang, Bin Ren, Xipeng Shen

The 28th IEEE Real-Time and Embedded Technology and Applications Symposium - Brief Presentation Track [RTAS'22 BP]

BLCR: Towards Real-time DNN Execution with Block-based Reweighted Pruning

Xiaolong Ma, Geng Yuan, Zhengang Li, Yifan Gong, Tianyun Zhan, Wei Niu, Zheng Zhan, Pu Zhao, Ning Liu, Jian Tang, Xue Lin, Bin Ren, Yanzhi Wang

23rd International Symposium on Quality Electronic Design, 2022 [ISQED'22]

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

Yuan, G., Ma, X., Niu, W., Li, Z., Kong, Z., Liu, N., Gong, Y., Zhan, Z., He, C., Jin, Q., Wang, S., Qin, M., Ren, B., Wang, Y., Liu, S., Lin, X.

Thirty-fifth Conference on Neural Information Processing Systems [NeurIPS'21]

HEALS: A Parallel eALS Recommendation System on CPU/GPU Heterogeneous Platforms

Qihan Wang, Wei Niu, Li Chen, Ruoming Jin, Bin Ren

28th IEEE International Conference on High Performance Computing, Data, & Analytics, 2021 (Acceptance Rate: 23%) [HiPC'21]

Achieving on-Mobile Real-Time Super-Resolution with Neural Architecture and Pruning Search

Zheng Zhan, Yifan Gong, Pu Zhao, Geng Yuan, Wei Niu, Yushu Wu, Tianyun Zhang, Malith Jayaweera, David Kaeli, Bin Ren, Xue Lin, Yanzhi Wang

International Conference on Computer Vision, 2021 [ICCV'21]

CoCoPIE: Enabling Real-Time AI on Off-the-Shelf Mobile Devices via Compression-Compilation Co-Design

Hui Guan, Shaoshan Liu, Xiaolong Ma, Wei Niu, Bin Ren, Xipeng Shen, Yanzhi Wang, Pu Zhao (author in alphabet order)

Communications of the ACM, 2021 (Featured Article) [CACM'21]

GRIM: A General, Real-Time Deep Learning Inference Framework for Mobile Devices based on Fine-Grained Structured Weight Sparsity

Wei Niu*, Zhengang Li*, Xiaolong Ma, Peiyang Dong, Gang Zhou, Xuehai Qian, Xue Lin, Yanzhi Wang, Bin Ren

IEEE Transactions on Pattern Analysis and Machine Intelligence, 2021 [TPAMI'21]

ClickTrain: Efficient and Accurate End-to-End Deep LearningTraining via Fine-Grained Architecture-Preserving Pruning

Chengming Zhang, Geng Yuan, Wei Niu, Jiannan Tian, Sian Jin, Donglin Zhuang, Zhe Jiang, Yanzhi Wang, Bin Ren, Shuaiwen Song, Dingwen Tao

The 35th ACM International Conference on Supercomputing (acceptance rate: 39/157) [ICS'21]

DNNFusion: Accelerating Deep Neural Networks Execution with Advanced Operator Fusion

Wei Niu, Jiexiong Guan, Yanzhi Wang, Gagan Agrawal, Bin Ren

42nd ACM SIGPLAN Conference on Programming Language Design and Implementation, 2021 (acceptance rate: 87/320) [PLDI'21]

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

Li, Z.*, Yuan, G.*, Niu, W.*, Zhao, P.*, Li, Y., Cai, Y., Shen, X., Zhan, Z., Kong, Z., Jin, Q., Chen, Z., Liu, S., Yang, K., Ren, B., Wang, Y., Lin, X.

IEEE/CVF Conference on Computer Vision and Pattern Recognition, 2021 (oral paper: top 5%, acceptance rate: 1593/7500) [CVPR'21]

Neural Pruning Search for Real-Time Object Detection of Autonomous Vehicles

Pu Zhao, Geng Yuan, Yuxuan Cai, Wei Niu, Qi Liu, Wujie Wen, Bin Ren, Yanzhi Wang, Xue Lin

58th Annual Design Automation Conference, 2021 [DAC'21]

Real-Time Mobile Acceleration of DNNs: From Computer Vision to Medical Applications

Hongjia Li, Geng Yuan, Wei Niu, Yuxuan Cai, Mengshu Sun, Zhengang Li, Bin Ren, Xue Lin, and Yanzhi Wang

Proceeding of Asia and South Pacific Design Automation Conference, 2021 [ASP-DAC'21]

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

Yuxuan Cai, Hongjia Li, Geng Yuan, Wei Niu, Yanyu Li, Xulong Tang, Bin Ren, Yanzhi Wang

Thirty-Fifth AAAI Conference on Artificial Intelligence, 2021 (Acceptance Rate: 20.9%) [AAAI'21]

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

Gen Li, Zhihao Shu

The European Conference on Computer Vision [ECCV'24]

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

Wei Niu*, Mengshu Sun*, Zhengang Li*, Jou-An Chen, Jiexiong Guan, Xipeng Shen, Yanzhi Wang, Xue Lin, Bin Ren

Thirty-Fifth AAAI Conference on Artificial Intelligence, 2021 (Acceptance Rate: 20.9%) [AAAI'21]

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

Xiaolong Ma*, Wei Niu*, Tianyun Zhang, Sijia Liu, Sheng Lin, Hongjia Li, Xiang Chen, Jian Tang, Kaisheng Ma, Bin Ren, and Yanzhi Wang

Proceeding of European Conference on Computer Vision, 2020 [ECCV'20]

A Privacy-Preserving-Oriented DNN Pruning and Mobile Acceleration Framework

Yifan Gong, Zheng Zhan, Zhengang Li, Wei Niu, Xiaolong Ma, Wenhao Wang, Bin Ren, Caiwen Ding, Xue Lin, Xiaolin Xu, Yanzhi Wang

Proceedings of the 2020 on Great Lakes Symposium on VLSI, 2020 [GLSVLSI'20]

RTMobile: Beyond Real-Time Mobile Acceleration of RNNs for Speech Recognition

Peiyan Dong, Siyue Wang, Wei Niu, Chengming Zhang, Sheng Lin, Zhengang Li, Yifan Gong, Bin Ren, Xue Lin, and Dingwen Tao

The 57th Annual Design Automation Conference, July, 2020 [DAC'20]

PCONV: The Missing but Desirable Sparsity in DNN Weight Pruning for Real-time Execution on Mobile Devices

Xiaolong Ma, Fu-Ming Guo, Wei Niu, Xue Lin, Jian Tang, Kaisheng Ma, Bin Ren, Yanzhi Wang

Thirty-four AAAI Conference on Artificial Intelligence, February, 2020 (Acceptance Rate: 1,591/7,737 = 20.6%) [AAAI'20]

PatDNN: Achieving Real-Time DNN Execution on Mobile Devices with Pattern-based Weight Pruning

Wei Niu, Xiaolong Ma, Sheng Lin, Shihao Wang, Xuehai Qian, Xue Lin, Yanzhi Wang, Bin Ren

The 25th International Conference on Architectural Support for Programming Languages and Operating Systems, March, 2020 [ASPLOS'20]

26ms Inference Time for ResNet-50: Towards Real-Time Execution of all DNNs on Smartphone

Wei Niu, Xiaolong Ma, Yanzhi Wang, Bin Ren

ICML Workshop, 2019 [ICML'19 Workshop]