Weiheng Bai

According to our database, Weiheng Bai authored at least 10 papers between 2023 and 2026.

Collaborative distances:

• Dijkstra number of four.
• Erdős number of four.

2026

How to Compress KV Cache in RL Post-Training? Shadow Mask Distillation for Memory-Efficient Alignment.

Rui Zhu, Weiheng Bai, Qiushi Wu, Yang Ren, Haixu Tang, Yuchu Liu

CoRR, May, 2026

GenDetect: Generalizing Reactive Detection for Resilience Against Imitative DeFi Attack Cascade.

Bowen Cai, Weiheng Bai, Youshui Lu, Haoran Xu, Yuannan Yang, Yajin Zhou, Kangjie Lu

CoRR, April, 2026

Capturing Monetarily Exploitable Vulnerability in Smart Contracts via Auditor Knowledge-Learning Fuzzing.

Bowen Cai, Weiheng Bai, Hangyun Tang, Youshui Lu, Kangjie Lu

CoRR, April, 2026

Compatibility at a Cost: Systematic Discovery and Exploitation of MCP Clause-Compliance Vulnerabilities.

Nanzi Yang, Weiheng Bai, Kangjie Lu

CoRR, March, 2026

2025

AFLGopher: Accelerating Directed Fuzzing via Feasibility-Aware Guidance.

Weiheng Bai, Kefu Wu, Qiushi Wu, Kangjie Lu

CoRR, November, 2025

BULKHEAD: Secure, Scalable, and Efficient Kernel Compartmentalization with PKS.

Yinggang Guo, Zicheng Wang, Weiheng Bai, Qingkai Zeng, Kangjie Lu

Proceedings of the 32nd Annual Network and Distributed System Security Symposium, 2025

APILOT: Improving the Security and Usability of LLM Code Suggestions via Outdated API Mitigation.

Weiheng Bai, Keyang Xuan, Pengxiang Huang, Qiushi Wu, Jianing Wen, Jingjing Wu, Kangjie Lu

Proceedings of the IEEE Annual Computer Security Applications Conference, 2025

2024

APILOT: Navigating Large Language Models to Generate Secure Code by Sidestepping Outdated API Pitfalls.

Weiheng Bai, Keyang Xuan, Pengxiang Huang, Qiushi Wu, Jianing Wen, Jingjing Wu, Kangjie Lu

CoRR, 2024

2023

ACTOR: Action-Guided Kernel Fuzzing.

Marius Fleischer, Dipanjan Das, Priyanka Bose, Weiheng Bai, Kangjie Lu, Mathias Payer, Christopher Kruegel, Giovanni Vigna

Proceedings of the 32nd USENIX Security Symposium, 2023

Guiding Directed Fuzzing with Feasibility.

Weiheng Bai, Kefu Wu, Qiushi Wu, Kangjie Lu

Proceedings of the IEEE European Symposium on Security and Privacy, 2023