Publications
You can also find my articles on my Google Scholar profile.
- Chaoyi Jiang*, Lei Gao*, Hossein Entezari Zarch, Murali Annavaram “KVPR: Efficient LLM Inference with I/O-Aware Partial KV Cache Recomputation.” AAAI 2025 Scalable and Efficient Artificial Intelligence Systems (SEAS) Workshop. [Link]
- Lei Gao*, Amir Ziashahabi*, Yue Niu, Salman Avestimehr, Murali Annavaram “Enabling Resource-Efficient On-Device Fine-Tuning of LLMs Using Only Inference Engines” NeurIPS 2024 Efficient Natural Language and Speech Processing Workshop. [Link]
- Lei Gao*, Yue Niu*, Tingting Tang, Salman Avestimehr, Murali Annavaram “Ethos: Rectifying Language Models in Orthogonal Parameter Space” NAACL Findings 2024. [Link]
- Lei Gao*, Yue Niu*, Tingting Tang, Salman Avestimehr, Murali Annavaram “Ethos: Rectifying Language Models in Orthogonal Parameter Space” AAAI 2024 Responsible Language Models Workshop (spotlight). [Link]
- Tuo Zhang, Lei Gao, Sunwoo Lee, Mi Zhang, Chaoyang He, Salman Avestimehr “TimelyFL: Heterogeneity aware Asynchronous Federated Learning with Adaptive Partial Training” CVPR 2023 Federated Learning for CV workshop. [Link]
- Tuo Zhang, Lei Gao, Chaoyang He, Mi Zhang, Bhaskar Krishnamachari, Salman Avestimehr “Federated Learning for Internet of Things: Applications, Challenges, and Opportunities” 2022 IEEE Internet of Things Magazine. [Link]
- Tuo Zhang, Chaoyang He, Tianhao Ma, Lei Gao, Mark Ma, Salman Avestimehr “Federated Learning for Internet of Things: A Federated Learning Framework for On-device Anomaly Data Detection” ACM SenSys 2021 AIChallengeIoT Workshop. [Link]
- Feng Yun, Yunkun Lin, Lou Yunfei, Lei Gao, Vaibhav Gera, Boxuan Li, Vennela Chowdary Nekkanti, Aditya Rajendra Pharande, Kunal Sheth, Meghana Thommondru, Guizhong Ye, Sandeep Gupta “Fault-coverage Maximizing March Tests for Memory Testing” 2022 IEEE International Test Conference. [Link]