Yujun Lin

yujunlin [at] mit.edu

Google Scholar /  LinkedIn /  GitHub

I am currently a Ph.D. student at MIT, advised by Prof.Song Han.
My research area is the intersection of computer architecture and machine learning, especially software and hardware co-design for deep learning and its applications..

Before coming to MIT, I received my B.Eng. in Electronic Engineering from Tsinghua University. During my undergrad, I was fortunate to work with Prof.Bill Dally and Prof.Yu Wang.


Selected Publications [Full List]

* indicates equal contribution

PointAcc: Efficient Point Cloud Accelerator
Yujun Lin, Zhekai Zhang, Haotian Tang, Hanrui Wang, Song Han
MICRO 2021 / arXiv / Project Page / Video / Slides
NAAS: Neural Accelerator Architecture Search
Yujun Lin*, Mengtian Yang*, Song Han
DAC 2021 / arXiv / Project Page / Slides
Searching Efficient 3D Architectures with Sparse Point-Voxel Convolution
Haotian Tang*, Zhijian Liu*, Shengyu Zhao, Yujun Lin, Ji Lin, Hanrui Wang, Song Han
ECCV 2020 / arXiv / Project Page / Code
MCUNet: Tiny Deep Learning on IoT Devices
Ji Lin, Wei-Ming Chen, Yujun Lin, John Cohn, Chuang Gan, Song Han
NeurIPS 2020 / arXiv / Project Page / Code
Press: MIT News (homepage spotlight) / WIRED / MIT TR-China / IBM / Morning Brew / Stacey on IoT / Analytics Insight / Techable / Tendencias
Spotlight Presentation
APQ: Joint Search for Network Architecture, Pruning and Quantization Policy
Tianzhe Wang*, Kuan Wang*, Han Cai, Ji Lin, Zhijian Liu, Hanrui Wang, Yujun Lin, Song Han
CVPR 2020 / Paper
Lite Transformer with Long-Short Range Attention
Zhanghao Wu*, Zhijian Liu*, Ji Lin, Yujun Lin, Song Han
ICLR 2020
AutoML for Architecting Efficient and Specialized Neural Networks
Han Cai*, Ji Lin*, Yujun Lin*, Zhijian Liu*, Kuan Wang*, Tianzhe Wang*, Ligeng Zhu*, Song Han
IEEE Micro 2020
Point-Voxel CNN for Efficient 3D Deep Learning
Zhijian Liu*, Haotian Tang*, Yujun Lin, Song Han
NeurIPS 2019 / arXiv / Project Page / Code
Spotlight Presentation
A configurable multi-precision CNN computing framework based on single bit RRAM
Zhenhua Zhu, Hanbo Sun, Yujun Lin, Guohao Dai, Lixue Xia, Song Han, Yu Wang, Huazhong Yang
DAC 2019
HAQ: Hardware-Aware Automated Quantization with Mixed Precision
Kuan Wang*, Zhijian Liu*, Yujun Lin*, Ji Lin, Song Han
CVPR 2019 / arXiv / Project Page / Code
Oral Presentation
Hardware-Centric AutoML for Mixed-Precision Quantization
Kuan Wang*, Zhijian Liu*, Yujun Lin*, Ji Lin, Song Han
IJCV 2020
Long live TIME: improving lifetime for training-in-memory engines by structured gradient sparsification
Yi Cai*, Yujun Lin*, Lixue Xia, Xiaoming Chen, Song Han, Yu Wang, Huazhong Yang
DAC 2018
Long live TIME: Improving lifetime and security for NVM-based training-in-memory systems
Yi Cai, Yujun Lin, Lixue Xia, Xiaoming Chen, Song Han, Yu Wang, Huazhong Yang
TCAD 2020
Deep Gradient Compression: Reducing the communication bandwidth for distributed training
Yujun Lin, Song Han, Huizi Mao, Yu Wang, William J. Dally
ICLR 2018 / arXiv / Code

Hornors and Awards

Academic Services