Publications

Yefei He, Luoming Zhang, Weijia Wu, Jing Liu, Hong Zhou, Bohan Zhuang (2024). ZipCache: Accurate and Efficient KV Cache Quantization with Salient Token Identification. arXiv preprint arXiv:2405.14256.

Yefei He, Jing Liu, Weijia Wu, Hong Zhou, Bohan Zhuang (2024). EfficientDM: Efficient Quantization-Aware Fine-Tuning of Low-Bit Diffusion Models. ICLR 2024 (spotlight).

Yefei He, Luping Liu, Jing Liu, Weijia Wu, Hong Zhou, Bohan Zhuang (2023). Ptqd: Accurate post-training quantization for diffusion models. NeurIPS 2023.

Weijia Wu, Zhuang Li, Yefei He, Mike Zheng Shou, Chunhua Shen, Lele Cheng, Yan Li, Tingting Gao, Di Zhang, Zhongyuan Wang (2023). Paragraph-to-image generation with information-enriched diffusion model. arXiv preprint arXiv:2311.14284.

Weijia Wu, Yuzhong Zhao, Hao Chen, Yuchao Gu, Rui Zhao, Yefei He, Hong Zhou, Mike Zheng Shou, Chunhua Shen (2023). Datasetdm: Synthesizing data with perception annotations using diffusion models. NeurIPS 2023.

Yefei He, Luoming Zhang, Weijia Wu, Hong Zhou (2023). Data-Free Quantization with Accurate Activation Clipping and Adaptive Batch Normalization. Neural Processing Letters.

Yefei He, Zhenyu Lou, Luoming Zhang, Jing Liu, Weijia Wu, Hong Zhou, Bohan Zhuang (2023). BiViT: Extremely Compressed Binary Vision Transformers. ICCV 2023.

Yefei He, Luoming Zhang, Weijia Wu, Hong Zhou (2023). Binarizing by classification: Is soft function really necessary?. IEEE Transactions on Circuits and Systems for Video Technology.