Reinforcement learning · Tool use
GRASP: Granularity-Aware Search Policy for Agentic RAG
Learning when to search, read, and answer. GRASP coordinates semantic search, keyword search, and paragraph reading to acquire evidence for multi-hop questions.
Published work across language models, human-centered NLP, and efficient computing.
01
Reinforcement learning · Tool use
Learning when to search, read, and answer. GRASP coordinates semantic search, keyword search, and paragraph reading to acquire evidence for multi-hop questions.
02
User simulation · Model evaluation
Evaluating whether simulated students reproduce the linguistic, behavioral, and cognitive properties of real learners in tutoring dialogues.
ACL 2026
Read paperPreference optimization · Activation steering
Learning tutor-persona steering vectors from human tutor–student dialogues to control model behavior through observed conversational preferences.
BEA 2026
Read paperPost-training · Tutoring dialogues
Training language-model tutors with the goal of improving student learning across a dialogue.
AIED 2025
Read paper03
Personalization · Human–AI interaction
Designing adaptive kanji mnemonics around learner preferences through an interactive prototype and user feedback.
CHI EA 2026
Read paperPreference modeling · Personalized generation
Inferring community-level and user-specific preferences to personalize mnemonic generation with limited feedback.
EMNLP 2025
Read paperCross-lingual NLP · Mnemonic generation
Using phonological similarity across languages to find keywords for memorable vocabulary-learning cues.
EMNLP 2025
Read paperApplied generation · Educational content
Generating vector-based mathematical diagrams from text with large language models.
AIED 2025
Read paperMnemonic generation · Human evaluation
Generating and ranking candidate mnemonics, with evaluation of their quality and usefulness as learning aids.
Findings of EMNLP 2024
Read paperAssessment generation · Applied NLP
Using large language models to generate distractors for mathematics multiple-choice questions.
Findings of NAACL 2024
Read paperLanguage learning · Multimodal mnemonics
Exploring automatically generated verbal and visual cues for keyword-based vocabulary learning.
AIED 2023
Read paper04
Model partitioning · Processing-in-memory
Partitioning deep-learning inference across heterogeneous hardware using low-overhead profiling and runtime scheduling.
ACM TODAES
Read paperML infrastructure · Runtime integration
Connecting ONNX models to processing-in-memory operations and evaluating BERT inference on an FPGA-emulated PIM platform.
ICEIC 2022
Read paperCompilers · Memory management
Extending LLVM/Polly to generate tiled execution, memory transfers, and address calculations for scratchpad memory.
ICEIC 2021
Read paperNatural language processing and ML systems, from model behavior to efficient inference.
01 / Natural language processing
NLP × Systems · Agentic retrieval
GRASP trains language-model agents to coordinate semantic search, keyword search, and paragraph reading. The policy learns how to acquire evidence and when to expand its context.
An illustration of the available actions. The learned policy chooses their order and when to answer.
Evaluation · User simulation
When do LLM-simulated students behave like real learners? We evaluate linguistic, behavioral, and cognitive realism, showing why surface-level plausibility can hide failures in learning dynamics.
Post-training · Activation steering
Tutor-persona steering vectors learned from human tutor–student dialogues offer interpretable control over model behavior. Preference optimization connects observed conversations to how a model responds.
Personalization · Human-centered AI
From interpretable preference modeling to an interactive iOS prototype, this work explores how language models can generate kanji mnemonics that adapt to individual learners, even with limited feedback.
A closer look / GRASP
Multi-hop questions require more than retrieving a relevant document. An agent must identify the next missing piece of evidence, choose a useful tool, and decide how much context to expose.
GRASP uses GRPO to learn these decisions. Search returns sentences; a separate read action expands their parent paragraphs. The training signal combines answer accuracy, grounded reading, complementary search, and turn efficiency.
The updated study evaluates both 3B and 7B models on HotpotQA, 2WikiMultihopQA, and MuSiQue.
Explore the project72 F1
HotpotQA · 7B model · 500 sampled validation questions
Token F1 × 100. Strongest listed baseline on this metric: AutoRefine. Updated manuscript, Table 3. GRASP ties AutoRefine on MuSiQue F1.
What remains open
The training rewards rely on gold supporting-document annotations. Learning grounded tool use with weaker supervision is an important next step.
02 / ML Systems
The model is one part of the system. Retrieval, data movement, and where computation runs shape its practical performance.
Hardware–software co-design · Inference
Model partitioning and low-overhead profiling for deep-learning inference on a processing-in-memory platform. This work connects runtime decisions to the capabilities of heterogeneous hardware.
ML infrastructure · Runtime integration
An execution provider connects ONNX models to processing-in-memory operations. The framework offloads supported operators and evaluates end-to-end BERT inference on an FPGA-emulated PIM platform.
Compilers · Memory management
Extending LLVM/Polly to generate tiled execution, memory transfers, and address calculations for scratchpad memory. The study examines how tile shapes and reuse affect data movement.
Connecting NLP and systems
GRASP studies retrieval decisions through learned policies. My ongoing RAG systems work also examines the latency–accuracy trade-offs of approximate versus exact retrieval, including time-to-first-token and downstream answer quality.
Current research in the CV