02 / Research

Research

I'm a PhD researcher in Computer Science at Arizona State University. My core research is on LLM agents — how they're trained, how they reason and act at test time, and how we evaluate and keep them safe. My work spans these stages, with papers at ICLR, ACL, NAACL, EMNLP and NeurIPS workshops.

Core research · LLM agents

The agent lifecycle

My work spans the full stack of building capable, reliable LLM agents — from generating the data that probes them, to training, to test-time methods, to evaluation & safety. Select a stage to explore the papers.

Themes

Research areas

LLM Agents

Multi-agent query rewriting for e-commerce (OptAgent), goal-driven hypothesis generation for materials discovery, and agents that compile real-world open-source software (BuildBench) — spanning how agents are trained, act, and are evaluated.

Reasoning & Planning

Reasoning about actions and change (ActionReasoningBench) and guiding models toward more diverse, higher-quality solutions at inference time (GuidedSampling).

Evaluation & Benchmarks

Benchmarks for temporal reasoning (UnSeenTimeQA), reasoning about actions (ActionReasoningBench) and agentic software building (BuildBench).

Safety & Robustness

Jailbreaking LLMs with novel complex ciphers (ACE / LACE) and studying how models fail when the common assumptions behind a context no longer hold.

Training & Self-Improvement

Teaching models to reflect without distillation (ThinkTuning) and improving RL with self-generated mid-training data.

Output

Recent publications

All publications →
[001]

COGTRL: Training LLMs for Scientific Discovery Assistance using Cognitive Traces via Reinforcement Learning

Shrinidhi Kumbhar, Santosh Mashetty, Divij Handa, Kevin Coutinho, Siddharth Sambhaji Ghule, Chitta Baral

✓ EMNLP 2026 Conference

TL;DR Trains LLMs to think like scientists: RL rewards the cognitive trace behind each method step, lifting 3B models by +7.85 points to rival 70B ones.

Large Language Models (LLMs) trained on extensive scientific research are increasingly integrated as assistants for scientific discovery. However, most research papers omit the fine-grained cognitive process of examining constraints, failed alternatives, and iterative decisions required to achieve the desired goal. Such cognitive processes are vital for real-world scientists working toward specific goals under constraints. In this paper, we show that LLMs, when trained to produce such cognitive traces, perform better as scientific discovery assistants than when trained solely on scientific literature. We propose COGTRL, a trajectory-level reinforcement learning framework that trains LLMs to emulate cognitively grounded reasoning by jointly optimizing cognitive traces and the scientific steps produced in an interleaved manner. Across two 3B-parameter models and two scientific domains (AI and Materials Science), COGTRL improves method quality by an average of 7.85 points over comparable 3B model baselines and achieves competitive performance relative to 70B parameter models. Moreover, analysis by domain experts shows a preference for methods generated by COGTRL over the baselines.
[002]

No-Box Vulnerability Analysis: Description-only Detection of Indirect Prompt Injection Vulnerabilities in MCP Servers

Zehua Zhang, Jie Hu, Pratham Hegde, Aditya Maheshbhai Gabani, Souradip Nath, Yibo Liu, Siyu Liu, Hongkai Chen, Hulin Wang, Zhuoer Lyu, Chang Zhu, Divij Handa, Yan Shoshitaishvili, Tiffany Bao, Ruoyu Wang, Adam Doupé

Preprint arXiv

TL;DR Finds indirect prompt-injection vulnerabilities in MCP servers from tool descriptions alone, no code or access needed, recovering 94 of 95 confirmed bugs.

Conventional vulnerability analysis relies on source-level or binary-level access, or dynamic interaction, all of which may be unavailable to third-party analysts auditing closed-source, remotely hosted, or commercially gated software. Therefore, we propose a new paradigm of no-box vulnerability analysis in which neither source code nor runtime interaction is accessible, and only functionality metadata is available. Such metadata defines the intended behavior of the system, including its inputs, outputs, and side effects, while constraining the space of implementations consistent with that behavior. The intended behavior implies irreducible data flows, the minimal source-to-sink data flow skeletons shared by all conforming implementations. By reasoning over irreducible data flows, an analyst can formulate vulnerability hypotheses without observing or interacting with the target system. The analyst can later validate these hypotheses when additional access becomes available. We showcase the feasibility of no-box vulnerability analysis through implementing a pipeline called MCPSEC, which audits Model Context Protocol (MCP) servers for indirect prompt injection vulnerabilities using only the tool metadata exposed at server registration time. We evaluate MCPSEC on 20 widely deployed MCP servers comprising 177 tools, among which human evaluators confirm 95 vulnerable tools. MCPSEC identified 143 tools as vulnerable, and for each vulnerable tool, it produces Theory of Concepts (ToCs), a hypothesized attack scenario for later analyst validation. Using metadata alone, MCPSEC recovers 94 (98.9% recall) real verified vulnerabilities, compared with 80 (84.2% recall) for the LLM baseline. Overall, our results introduce no-box vulnerability analysis as a new analysis paradigm and demonstrate its practical feasibility in realistic systems.
[004]

GuidedSampling: Steering LLMs Towards Diverse Candidate Solutions at Inference-Time

Divij Handa, Mihir Parmar, Aswin RRV, Md Nayem Uddin, Hamid Palangi, Chitta Baral

✓ ICLR 2026 Conference

TL;DR Makes inference-time sampling produce genuinely diverse solutions — +21.6% pass@50.

Decouples exploration from generation at inference time so repeated sampling produces genuinely diverse candidate solutions, improving pass@50 by ~21.6% over standard repeated sampling.
[005]

BuildBench: Benchmarking LLM Agents on Compiling Real-World Open-Source Software

Zehua Zhang, A. P. Bajaj, Divij Handa, S. Liu, A. S. Raj, H. Chen, H. Wang, Y. Liu, et al.

✓ DL4C @ NeurIPS 2025 Workshop

TL;DR A realistic benchmark for LLM agents that compile real-world open-source software.

A challenging, realistic benchmark of diverse real-world open-source software for evaluating LLM agents on automatically compiling projects, plus OSS-Build-Agent, a strong baseline with an enhanced build-instruction retrieval module.