Saravan Rajmohan
Author directoryOther people with similar names: Saravan Rajmohan
Unverified author pages with similar names: Saravan Rajmohan
2026
Zipage: Maintain High Request Concurrency for LLM Reasoning through Compressed PagedAttention
Mengqi Liao | Lu Wang | Chaoyun Zhang | Bo Qiao | Si Qin | Qingwei Lin | Saravan Rajmohan | Dongmei Zhang | Huaiyu Wan
Findings of the Association for Computational Linguistics: ACL 2026
Mengqi Liao | Lu Wang | Chaoyun Zhang | Bo Qiao | Si Qin | Qingwei Lin | Saravan Rajmohan | Dongmei Zhang | Huaiyu Wan
Findings of the Association for Computational Linguistics: ACL 2026
With reasoning becoming the generative paradigm for large language models, the memory bottleneck caused by KV cache during the inference phase has become a critical factor limiting high-concurrency service capabilities. Although existing KV cache eviction methods address the memory issue, most of them are impractical for industrial-grade applications. This paper introduces Compressed PagedAttention, a method that combines token-wise KV cache eviction with PagedAttention. We propose a comprehensive scheduling strategy and support prefix caching and asynchronous compression for Compressed PagedAttention. Based on this, we have developed a high-concurrency inference engine, Zipage. On large-scale mathematical reasoning tasks, Zipage achieves around 95% of the performance of Full KV inference engines while delivering over 2.1 speedup.
EgoMemory: Memory-Augmented Personalized Retrieval for Long-Context Egocentric Video
Yuanmin Tang | Jue Zhang | Xiaoting Qin | Jing Yu | Meikang Qiu | Gaopeng Gou | Gang Xiong | Qingwei Lin | Saravan Rajmohan | Dongmei Zhang | Qi Wu
Findings of the Association for Computational Linguistics: ACL 2026
Yuanmin Tang | Jue Zhang | Xiaoting Qin | Jing Yu | Meikang Qiu | Gaopeng Gou | Gang Xiong | Qingwei Lin | Saravan Rajmohan | Dongmei Zhang | Qi Wu
Findings of the Association for Computational Linguistics: ACL 2026
Recent advances in AI and wearable devices, such as augmented-reality glasses, have made it possible to augment human memory by retrieving personal experiences in response to natural language queries. However, existing egocentric video datasets fall short in supporting the personalization and long-context reasoning required for episodic memory retrieval. To address these limitations, we introduce EgoMemory, a benchmark derived from Ego4D, enriched with 165,795 user-specific object annotations over 245 videos from 45 participants, yielding 639 distinct, human-curated, and evaluated queries for rich and individualized episodic memory retrieval. Leveraging this resource, we present EgoRetriever, a novel, training-free retrieval framework that combines Multimodal Large Language Models with reflective Chain-of-Thought prompting. Our approach enables interpretive inference of user intent and generates detailed target video descriptions by leveraging contextualized personal memory for video retrieval. Extensive experiments on three benchmarks, including EgoMemory, EgoCVR, and EgoLife, demonstrate that EgoRetriever consistently and substantially outperforms state-of-the-art baselines, highlighting its strong generalizability and practical potential for personalized, long-context egocentric video retrieval.
Learning to Refine: Self-Refinement of Parallel Reasoning in LLMs
Qibin Wang | Pu Zhao | Shaohan Huang | Fangkai Yang | Lu Wang | Furu Wei | Qingwei Lin | Saravan Rajmohan | Dongmei Zhang
Findings of the Association for Computational Linguistics: ACL 2026
Qibin Wang | Pu Zhao | Shaohan Huang | Fangkai Yang | Lu Wang | Furu Wei | Qingwei Lin | Saravan Rajmohan | Dongmei Zhang
Findings of the Association for Computational Linguistics: ACL 2026
Test-time scaling (TTS) has gained widespread attention for enhancing LLM reasoning. Existing approaches such as Best-of-N and majority voting are limited as their performance depends on the quality of candidate responses, making them unable to produce a correct solution when all candidates are incorrect. Parallel self-refinement, generating multiple candidates and synthesizing a refined answer conditioned on them, offers a promising alternative, but the underlying mechanism driving its effectiveness remains obscure. To bridge this gap in understanding, we introduce a new metric, the Refinement Gap, designed to quantify the relative improvement of self-refinement beyond majority voting. We show that the Refinement Gap exhibits a clear scaling trend with model size and is only weakly correlated with the base capability. Based on this discovery, we propose Generative Self-Refinement (GSR), a parallel test-time scaling framework that transfers the refinement policy from larger teacher models with higher refinement gap into smaller students. Crucially, GSR jointly trains a single model to generate strong candidates and refine a better final answer based on these candidates. Experimental results demonstrate that our method achieves state-of-the-art performance across five mathematical benchmarks over other parallel aggregation methods, while the learned refinement skill transfers across multiple model scales and families and exhibits robust generalization to an out-of-distribution domain.
DUET: Joint Exploration of User–Item Profiles in Recommendation System
Yue Chen | Yifei Sun | Lu Wang | Fangkai Yang | Pu Zhao | Minjie Hong | Yifei Dong | Minghua He | Nan Hu | Jianjin Zhang | Zhiwei Dai | Yuefeng Zhan | Weihao Han | Hao Sun | Qingwei Lin | Weiwei Deng | Feng Sun | Qi Zhang | Saravan Rajmohan | Dongmei Zhang
Findings of the Association for Computational Linguistics: ACL 2026
Yue Chen | Yifei Sun | Lu Wang | Fangkai Yang | Pu Zhao | Minjie Hong | Yifei Dong | Minghua He | Nan Hu | Jianjin Zhang | Zhiwei Dai | Yuefeng Zhan | Weihao Han | Hao Sun | Qingwei Lin | Weiwei Deng | Feng Sun | Qi Zhang | Saravan Rajmohan | Dongmei Zhang
Findings of the Association for Computational Linguistics: ACL 2026
Traditional recommendation systems represent users and items as dense vectors and learn to align them in a shared latent space for relevance estimation. Recent LLM-based recommenders instead leverage natural-language representations that are easier to interpret and integrate with downstream reasoning modules. This paper studies how to construct effective textual profiles for users and items, and how to align them for recommendation.A central difficulty is that the best profile format is not known a priori: manually designed templates can be brittle and misaligned with task objectives. Moreover, generating user and item profiles independently may produce descriptions that are individually plausible yet semantically inconsistent for a specific user–item pair. We propose Duet, an interaction-aware profile generator that jointly produces user and item profiles conditioned on both user history and item evidence. Duet follows a three-stage procedure: it first turns raw histories and metadata into compact cues, then expands these cues into paired profile prompts and then generate profiles, and finally optimizes the generation policy with reinforcement learning using downstream recommendation performance as feedback. Experiments on three real-world datasets show that Duet consistently outperforms strong baselines, demonstrating the benefits of template-free profile exploration and joint user–item textual alignment. Project page: https://duet-rec.github.io/.
SynthAgent: Adapting Web Agents with Synthetic Supervision
Zhaoyang Wang | Yiming Liang | Xuchao Zhang | Qianhui Wu | Siwei Han | Anson Bastos | Rujia Wang | Chetan Bansal | Baolin Peng | Jianfeng Gao | Saravan Rajmohan | Huaxiu Yao
Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
Zhaoyang Wang | Yiming Liang | Xuchao Zhang | Qianhui Wu | Siwei Han | Anson Bastos | Rujia Wang | Chetan Bansal | Baolin Peng | Jianfeng Gao | Saravan Rajmohan | Huaxiu Yao
Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
Web agents struggle to adapt to new websites due to the scarcity of environment specific tasks and demonstrations. Recent works have explored synthetic data generation to address this challenge, however, they suffer from data quality issues where synthesized tasks contain hallucinations that cannot be executed, and collected trajectories are noisy with redundant or misaligned actions. In this paper, we propose SynthAgent, a fully synthetic supervision framework that aims at improving synthetic data quality via dual refinement of both tasks and trajectories. Our approach begins by synthesizing diverse tasks through categorized exploration of web elements, ensuring efficient coverage of the target environment. During trajectory collection, tasks are refined only when conflicts with observations are detected, which mitigates hallucinations while preserving task consistency. After collection, we conduct trajectory refinement with global context to mitigate potential noise or misalignments. Finally, we fine-tune open-source web agents on the refined synthetic data to adapt them to the target environment. Experimental results demonstrate that SynthAgent outperforms existing synthetic data methods, validating the importance of high-quality synthetic supervision. The code is publicly available at https://github.com/aiming-lab/SynthAgent.
RepoGenesis: Benchmarking End-to-End Microservice Generation from Readme to Repository
Zhiyuan Peng | Xin Yin | Pu Zhao | Fangkai Yang | Lu Wang | Ran Jia | Xu Chen | Qingwei Lin | Saravan Rajmohan | Dongmei Zhang
Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
Zhiyuan Peng | Xin Yin | Pu Zhao | Fangkai Yang | Lu Wang | Ran Jia | Xu Chen | Qingwei Lin | Saravan Rajmohan | Dongmei Zhang
Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
Large language models and agents have achieved remarkable progress in code generation. However, existing benchmarks focus on isolated function/class-level generation (e.g., ClassEval) or modifications to existing codebases (e.g., SWE-Bench), neglecting complete microservice repository generation that reflects real-world 0-to-1 development workflows. To bridge this gap, we introduce RepoGenesis, the first multilingual benchmark for repository-level end-to-end web microservice generation, comprising 106 repositories (60 Python, 46 Java) across 18 domains and 11 frameworks, with 1,258 API endpoints and 2,335 test cases verified through a “review-rebuttal” quality assurance process. We evaluate open-source agents (e.g., DeepCode) and commercial IDEs (e.g., Cursor) using Pass@1, API Coverage (AC), and Deployment Success Rate (DSR). Results reveal that despite high AC (up to 73.91%) and DSR (up to 100%), the best-performing system achieves only 23.67% Pass@1 on Python and 21.45% on Java, exposing deficiencies in architectural coherence, dependency management, and cross-file consistency. Notably, RepoGenesis-8B, fine-tuned on RepoGenesis (train), achieves performance comparable to GPT-5 mini, demonstrating the quality of RepoGenesis for advancing microservice generation. We release our benchmark at https://github.com/pzy2000/RepoGenesis.
LLM Reasoning as Trajectories: Step-Specific Representation Geometry and Correctness Signals
Lihao Sun | Hang Dong | Bo Qiao | Qingwei Lin | Dongmei Zhang | Saravan Rajmohan
Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
Lihao Sun | Hang Dong | Bo Qiao | Qingwei Lin | Dongmei Zhang | Saravan Rajmohan
Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
This work characterizes large language models’ chain-of-thought generation as a structured trajectory through representation space. We show that mathematical reasoning traverses functionally ordered, step-specific subspaces that become increasingly separable with layer depth. This structure already exists in base models, while reasoning training primarily accelerates convergence toward termination-related subspaces rather than introducing new representational organization. While early reasoning steps follow similar trajectories, correct and incorrect solutions diverge systematically at late stages. This late-stage divergence enables mid-reasoning prediction of final-answer correctness with ROC–AUC up to 0.87. Furthermore, we introduce trajectory-based steering, an inference-time intervention framework that enables reasoning correction and length control based on derived ideal trajectories. Together, these results establish reasoning trajectories as a geometric lens for interpreting, predicting, and controlling LLM reasoning behavior.
Gradient-Guided Multi-Judge Prompt Optimization
ChenZhuo Zhao | Xinda Wang | Pu Zhao | Yue Huang | Junting Lu | Ziqian Liu | Qingwei Lin | Saravan Rajmohan | Dongmei Zhang
Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
ChenZhuo Zhao | Xinda Wang | Pu Zhao | Yue Huang | Junting Lu | Ziqian Liu | Qingwei Lin | Saravan Rajmohan | Dongmei Zhang
Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
Automatic prompt optimization is a practical alternative to fine-tuning for adapting large language models (LLMs), yet existing approaches often trade off signal quality against computational cost. Methods that rely on generative feedback can be informative but expensive to scale, while sampling-based optimization typically requires many evaluations and exhibits high variance. Even loss-driven prompt optimization remains limited by costly segment attribution that scales with prompt length and by overfitting to a single evaluator, which weakens transfer across model families and domains. We propose Gradient-guided Multi-judge Prompt Optimization (GMPO), a scalable framework that improves both efficiency and robustness. GMPO uses a first-order gradient approximation to score segment importance in a continuous masking direction, requiring only one forward and one backward pass. GMPO further employs a generate multi-judge design in which candidate prompt edits are proposed by a generator and selected using cross-entropy losses aggregated from multiple lightweight judge models, reducing evaluator bias and improving generalization. Experiments across math, reasoning, instruction-following evaluation, and safety robustness benchmarks demonstrate consistent gains with substantially lower optimization overhead.
CI-Work: Benchmarking Contextual Integrity in Enterprise LLM Agents
Wenjie Fu | Xiaoting Qin | Jue Zhang | Qingwei Lin | Lukas Wutschitz | Robert Sim | Saravan Rajmohan | Dongmei Zhang
Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 6: Industry Track)
Wenjie Fu | Xiaoting Qin | Jue Zhang | Qingwei Lin | Lukas Wutschitz | Robert Sim | Saravan Rajmohan | Dongmei Zhang
Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 6: Industry Track)
Enterprise LLM agents can dramatically improve workplace productivity, but their core capability, retrieving and using internal context to act on a user’s behalf, also creates new risks for sensitive information leakage. We introduce CI-Work, a Contextual Integrity (CI)-grounded benchmark that simulates enterprise workflows across five information-flow directions and evaluates whether agents can convey essential content while withholding sensitive context in dense retrieval settings.Our evaluation of frontier models reveals that privacy failures are prevalent (violation rates range from 15.8%-50.9%, with leakage reaching up to 26.7%) and uncovers a counterintuitive trade-off critical for industrial deployment: higher task utility often correlates with increased privacy violations.Moreover, the massive scale of enterprise data and potential user behavior further amplify this vulnerability. Simply increasing model size or reasoning depth fails to address the problem. We conclude that safeguarding enterprise workflows requires a paradigm shift, moving beyond model-centric scaling toward context-centric architectures.
2025
AdaptFlow: Adaptive Workflow Optimization via Meta-Learning
Runchuan Zhu | Bowen Jiang | Lingrui Mei | Fangkai Yang | Lu Wang | Haoxiang Gao | Fengshuo Bai | Pu Zhao | Qingwei Lin | Saravan Rajmohan | Dongmei Zhang
Findings of the Association for Computational Linguistics: EMNLP 2025
Runchuan Zhu | Bowen Jiang | Lingrui Mei | Fangkai Yang | Lu Wang | Haoxiang Gao | Fengshuo Bai | Pu Zhao | Qingwei Lin | Saravan Rajmohan | Dongmei Zhang
Findings of the Association for Computational Linguistics: EMNLP 2025
Recent advances in large language models (LLMs) have sparked growing interest in agentic workflows—structured sequences of LLM invocations designed to solve complex tasks. However, existing approaches often rely on static templates or manually designed workflows, which limit adaptability to diverse tasks and hinder scalability. We propose AdaptFlow, a natural language-based meta-learning framework inspired by model-agnostic meta-learning (MAML). AdaptFlow uses a bi-level optimization process: the inner loop performs task-specific adaptation via LLM-generated feedback, while the outer loop consolidates these refinements into a shared, generalizable initialization. Evaluated across question answering, code generation, and mathematical reasoning benchmarks, AdaptFlow consistently outperforms both manually crafted and automatically searched baselines, achieving state-of-the-art results with strong generalization across tasks and models.
TACO-RL: Task Aware Prompt Compression Optimization with Reinforcement Learning
Shivam Shandilya | Menglin Xia | Supriyo Ghosh | Huiqiang Jiang | Jue Zhang | Qianhui Wu | Victor Rühle | Saravan Rajmohan
Findings of the Association for Computational Linguistics: ACL 2025
Shivam Shandilya | Menglin Xia | Supriyo Ghosh | Huiqiang Jiang | Jue Zhang | Qianhui Wu | Victor Rühle | Saravan Rajmohan
Findings of the Association for Computational Linguistics: ACL 2025
The increasing prevalence of large language models (LLMs) such as GPT-4 in various applications has led to a surge in the size of prompts required for optimal performance, leading to challenges in computational efficiency. Prompt compression aims to reduce the inference cost by minimizing input tokens without compromising on the task performance. However, existing prompt compression techniques either rely on sub-optimal metrics such as information entropy or model it as a task-agnostic token classification problem that fails to capture task-specific information.To address these issues, we propose a novel and efficient reinforcement learning (RL) based task-aware prompt compression method. To ensure low latency requirements, we leverage existing Transformer encoder-based token classification model while guiding the learning process with task-specific reward signals using lightweight REINFORCE algorithm. We evaluate the performance of our method on three diverse and challenging tasks including text summarization, question answering and code summarization. We demonstrate that our RL-guided compression method improves the task performance by 8% - 189% across these three scenarios over state-of-the-art compression techniques while satisfying the same compression rate and latency requirements.
ExeCoder: Empowering Large Language Models with Executability Representation for Code Translation
Minghua He | Yue Chen | Fangkai Yang | Pu Zhao | Wenjie Yin | Yu Kang | Qingwei Lin | Saravan Rajmohan | Dongmei Zhang
Proceedings of the 2025 Conference on Empirical Methods in Natural Language Processing
Minghua He | Yue Chen | Fangkai Yang | Pu Zhao | Wenjie Yin | Yu Kang | Qingwei Lin | Saravan Rajmohan | Dongmei Zhang
Proceedings of the 2025 Conference on Empirical Methods in Natural Language Processing
Code translation is a crucial activity in the software development and maintenance process, and researchers have recently begun to focus on using pre-trained large language models (LLMs) for code translation. However, existing LLMs only learn the contextual semantics of code during pre-training, neglecting executability information closely related to the execution state of the code, which results in unguaranteed code executability and unreliable automated code translation. To address this issue, we propose ExeCoder, an LLM specifically designed for code translation, aimed at utilizing executability representations such as functional semantics, syntax structures, and variable dependencies to enhance the capabilities of LLMs in code translation. To evaluate the effectiveness of ExeCoder, we manually enhanced the widely used benchmark TransCoder-test, resulting in a benchmark called TransCoder-test-X that serves LLMs. Evaluation of TransCoder-test-X indicates that ExeCoder achieves state-of-the-art performance in code translation, surpassing existing open-source code LLMs by over 10.88% to 38.78% and over 27.44% to 42.97% on two metrics, and even outperforms the renowned closed-source LLM GPT-4o. Code is available at https://aka.ms/execoder
Synergistic Weak-Strong Collaboration by Aligning Preferences
Yizhu Jiao | Xuchao Zhang | Zhaoyang Wang | Yubo Ma | Zhun Deng | Rujia Wang | Chetan Bansal | Saravan Rajmohan | Jiawei Han | Huaxiu Yao
Proceedings of the 63rd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
Yizhu Jiao | Xuchao Zhang | Zhaoyang Wang | Yubo Ma | Zhun Deng | Rujia Wang | Chetan Bansal | Saravan Rajmohan | Jiawei Han | Huaxiu Yao
Proceedings of the 63rd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers)
Current Large Language Models excel in general reasoning yet struggle with specialized tasks requiring proprietary or domain-specific knowledge. Fine-tuning large models for every niche application is often infeasible due to black-box constraints and high computational overhead. To address this, we propose a collaborative framework that pairs a specialized weak model with a general strong model. The weak model, tailored to specific domains, produces initial drafts and background information, while the strong model leverages its advanced reasoning to refine these drafts, extending LLMs’ capabilities to critical yet specialized tasks. To optimize this collaboration, we introduce a collaborative feedback to fine-tunes the weak model, which quantifies the influence of the weak model’s contributions in the collaboration procedure and establishes preference pairs to guide preference tuning of the weak model. We validate our framework through experiments on three domains. We find that the collaboration significantly outperforms each model alone by leveraging complementary strengths. Moreover, aligning the weak model with the collaborative preference further enhances overall performance.
2024
EfficientRAG: Efficient Retriever for Multi-Hop Question Answering
Ziyuan Zhuang | Zhiyang Zhang | Sitao Cheng | Fangkai Yang | Jia Liu | Shujian Huang | Qingwei Lin | Saravan Rajmohan | Dongmei Zhang | Qi Zhang
Proceedings of the 2024 Conference on Empirical Methods in Natural Language Processing
Ziyuan Zhuang | Zhiyang Zhang | Sitao Cheng | Fangkai Yang | Jia Liu | Shujian Huang | Qingwei Lin | Saravan Rajmohan | Dongmei Zhang | Qi Zhang
Proceedings of the 2024 Conference on Empirical Methods in Natural Language Processing
Retrieval-augmented generation (RAG) methods encounter difficulties when addressing complex questions like multi-hop queries.While iterative retrieval methods improve performance by gathering additional information, current approaches often rely on multiple calls of large language models (LLMs).In this paper, we introduce EfficientRAG, an efficient retriever for multi-hop question answering.EfficientRAG iteratively generates new queries without the need for LLM calls at each iteration and filters out irrelevant information.Experimental results demonstrate that EfficientRAG surpasses existing RAG methods on three open-domain multi-hop question-answering datasets.The code is available in aka.ms/efficientrag.
Search
Fix author
Co-authors
- Qingwei Lin 11
- Dongmei Zhang 11
- Fangkai Yang 6
- Pu Zhao 6
- Lu Wang 4
- Jue Zhang 3
- Chetan Bansal 2
- Yue Chen 2
- Minghua He 2
- Bo Qiao 2
- Xiaoting Qin 2
- Rujia Wang 2
- Zhaoyang Wang 2
- Qianhui Wu 2
- Huaxiu Yao 2
- Xuchao Zhang 2
- Fengshuo Bai 1
- Anson Bastos 1
- Xu Chen 1
- Sitao Cheng 1
- Zhiwei Dai 1
- Weiwei Deng 1
- Zhun Deng 1
- Hang Dong 1
- Yifei Dong 1
- Wenjie Fu 1
- Haoxiang Gao 1
- Jianfeng Gao 1
- Supriyo Ghosh 1
- Gaopeng Gou 1
- Jiawei Han 1
- Siwei Han 1
- Weihao Han 1
- Minjie Hong 1
- Nan Hu 1
- Shaohan Huang 1
- Shujian Huang (书剑 黄) 1
- Yue Huang 1
- Ran Jia 1
- Bowen Jiang 1
- Huiqiang Jiang 1
- Yizhu Jiao 1
- Yu Kang 1
- Yiming Liang 1
- Mengqi Liao 1
- Jia Liu 1
- Ziqian Liu 1
- Junting Lu 1
- Yubo Ma 1
- Lingrui Mei 1
- Baolin Peng 1
- Zhiyuan Peng 1
- Si Qin 1
- Meikang Qiu 1
- Victor Rühle 1
- Shivam Shandilya 1
- Robert Sim 1
- Feng Sun 1
- Hao Sun 1
- Lihao Sun 1
- Yifei Sun 1
- Yuanmin Tang 1
- Huaiyu Wan 1
- Lu Wang 1
- Qibin Wang 1
- Xinda Wang 1
- Furu Wei 1
- Qi Wu 1
- Lukas Wutschitz 1
- Menglin Xia 1
- Gang Xiong 1
- Wenjie Yin 1
- Xin Yin 1
- Jing Yu 1
- Yuefeng Zhan 1
- Chaoyun Zhang 1
- Jianjin Zhang 1
- Qi Zhang 1
- Qi Zhang 1
- Zhiyang Zhang 1
- ChenZhuo Zhao 1
- Runchuan Zhu 1
- Ziyuan Zhuang 1