← Search

Azal Ahmad Khan

4 accepted papers

2026

Retrieval-of-Thought: Efficient Reasoning via Reusing Thoughts

ICLR 2026poster

Large reasoning models improve accuracy by producing long reasoning traces, but this inflates latency and cost, motivating inference-time efficiency. We propose Retrieval-of-Thought (RoT), which reuses prior reasoning as composable ``thought" steps to guide new problems. RoT organizes steps into a t…

Cited by 0SourcecodeScholar
2025

Accelerating LLM Reasoning via Early Rejection with Partial Reward Modeling

EMNLP 2025

Large Language Models (LLMs) are increasingly relied upon for solving complex reasoning tasks in domains such as mathematics, logic, and multi-step question answering. A growing line of work seeks to improve reasoning quality by scaling inference time compute particularly through Process Reward Mode

Cited by 0SourcePDFScholar
2025

Beyond Expectations: Quantile-Guided Alignment for Risk-Calibrated Language Models

NeurIPS 2025spotlight

Large language models can generate rare but catastrophic outputs, such as harmful conversations or insecure code. Existing Reinforcement Learning from Human Feedback (RLHF) typically maximizes average reward, leaving high-risk tail events insufficiently controlled. We introduce Quantile‑Guided Align…

Cited by 0SourceScholar
2025

Safety Aware Task Planning via Large Language Models in Robotics

IROS 2025

The integration of large language models (LLMs) into robotic task planning has unlocked better reasoning capabilities for complex, long-horizon workflows. However, ensuring safety in LLM-driven plans remains a critical challenge, as these models often prioritize task completion over risk mitigation.

Cited by 23SourceScholar