1 citations · 1 across the 3 of their papers we have counts for
7 papers
PROBE: Learning to Audit Policy Compliance in Tool-Using LLM Agents
Kshitij Mishra, Abhijith Sharma, Nils Lukas +1
Modern agentic systems extend language models with external tools, enabling them to complete complex tasks through multi-turn interactions. As these agents are deployed in conseque…
Selection Is Not Learning: Choosing Model Futures
Haoyi Yang, Kshitij Mishra, Nils Lukas +1
Self-evolving language models train on tasks they generate, so every round must decide which updated model the next round inherits. Current loops such as R-Zero, Agent0, and Skill…
TRACER: Trace-Regularized Training Against Indirect Prompt Injection
Kshitij Mishra, Salem Lahlou, Nils Lukas
Indirect prompt injection uses untrusted retrieved content, tool outputs, or agent messages to override an LLM's intended behavior. We show that defense robustness is channel-condi…
AdvSim2Real : Training Web Agents Against Adaptive Prompt Injection in a Web World Model
Sarim Hashmi, Mukul Ranjan, Kshitij Mishra +3
Web agents complete user requests by reading and acting on pages that third parties write, so an instruction planted on a page can redirect the agent away from the user's goal. The…
SD-E: Semantic Exploration for Reasoning Under Token Budgets
Kshitij Mishra, Nils Lukas, Salem Lahlou
Small language models (SLMs) struggle with complex reasoning because exploration is expensive under tight compute budgets. We introduce Semantic Diversity-Exploration-Exploitation…
CORE: Collaborative Reasoning via Cross Teaching
Kshitij Mishra, Mirat Aubakirov, Martin Takac +2
Large language models exhibit complementary reasoning errors: on the same instance, one model may succeed with a particular decomposition while another fails. We propose Collaborat…
Tabular and Deep Reinforcement Learning for Gittins Index
Harshit Dhankhar, Kshitij Mishra, Tejas Bodas
In the realm of multi-arm bandit problems, the Gittins index policy is known to be optimal in maximizing the expected total discounted reward obtained from pulling the Markovian ar…