147 citations · 741 across the 35 of their papers we have counts for
Showing cs.LGShow all
3 papers · 1 filter
cs.LG2025
CUDA-L2: Surpassing cuBLAS Performance for Matrix Multiplication through Reinforcement Learning
Songqiao Su, Xiaoya Li, Albert Wang +3
In this paper, we propose CUDA-L2, a system that combines large language models (LLMs) and reinforcement learning (RL) to automatically optimize Half-precision General Matrix Multi…
cs.LG2025
CRINN: Contrastive Reinforcement Learning for Approximate Nearest Neighbor Search
Xiaoya Li, Albert Wang, Guoyin Wang +2
Approximate nearest-neighbor search (ANNS) algorithms have become increasingly critical for recent AI applications, particularly in retrieval-augmented generation (RAG) and agent-b…
cs.LG2021★ 3 cited
A General Framework for Defending Against Backdoor Attacks via Influence Graph
Xiaofei Sun, Jiwei Li, Xiaoya Li +5
In this work, we propose a new and general framework to defend against backdoor attacks, inspired by the fact that attack triggers usually follow a \textsc{specific} type of attack…