4 papers
MARS: A Monte Carlo Tree Search-based Adaptive and Responsive Scheduler
Yash Kurkure, Yihe Zhang, Zhiling Lan +1
Modern High Performance Computing systems depend on static heuristics and manual administration for job scheduling and reservation management. Deep Reinforcement Learning (DRL) has…
Ranking Before Serving: Low-Latency LLM Serving via Pairwise Learning-to-Rank
Yiheng Tao, Yihe Zhang, Matthew Dearing +4
Efficient scheduling of large language model (LLM) inference tasks is critical for achieving low latency and high throughput, a challenge that is becoming increasingly acute with t…
A Real-Time Digital Twin for Adaptive Scheduling
Yihe Zhang, Yash Kurkure, Yiheng Tao +2
High-performance computing (HPC) workloads are becoming increasingly diverse, exhibiting wide variability in job characteristics, yet cluster scheduling has long relied on static,…
Deep Learning Aided Software Vulnerability Detection: A Survey
Md Nizam Uddin, Yihe Zhang, Xiali Hei
The pervasive nature of software vulnerabilities has emerged as a primary factor for the surge in cyberattacks. Traditional vulnerability detection methods, including rule-based, s…