Publications

Yefei He, Feng Chen, Yuanyu He, Shaoxuan He, Hong Zhou, Kaipeng Zhang, Bohan Zhuang (2024). ZipAR: Accelerating Autoregressive Image Generation through Spatial Locality. arXiv preprint arXiv:2412.04062.

PDF Cite Code Project

Yefei He, Feng Chen, Jing Liu, Wenqi Shao, Hong Zhou, Kaipeng Zhang, Bohan Zhuang (2024). ZipVL: Efficient Large Vision-Language Models with Dynamic Token Sparsification and KV Cache Compression. arXiv preprint arXiv:2410.08584.

PDF Cite Project

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

Akide Liu, Jing Liu, Zizheng Pan, Yefei He, Gholamreza Haffari, Bohan Zhuang (2024). MiniCache: KV Cache Compression in Depth Dimension for Large Language Models. NeurIPS 2024.

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

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.

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, 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, 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.