Saravan Rajmohan
Author directoryOther people with similar names: Saravan Rajmohan
Unverified author pages with similar names: Saravan Rajmohan
2026
WebXSkill: Skill Learning for Autonomous Web Agents
Zhaoyang Wang | Qianhui Wu | Xuchao Zhang | Chaoyun Zhang | Wenlin Yao | Fazle Elahi Faisal | Baolin Peng | Si Qin | Suman Nath | Qingwei Lin | Chetan Bansal | Dongmei Zhang | Saravan Rajmohan | Jianfeng Gao | Huaxiu Yao
Findings of the Association for Computational Linguistics: EMNLP 2026
Zhaoyang Wang | Qianhui Wu | Xuchao Zhang | Chaoyun Zhang | Wenlin Yao | Fazle Elahi Faisal | Baolin Peng | Si Qin | Suman Nath | Qingwei Lin | Chetan Bansal | Dongmei Zhang | Saravan Rajmohan | Jianfeng Gao | Huaxiu Yao
Findings of the Association for Computational Linguistics: EMNLP 2026
Autonomous web agents powered by large language models (LLMs) remain brittle on long-horizon browser workflows. A key bottleneck is a grounding gap in existing skill formulations: textual workflow skills provide natural language guidance but cannot be directly executed, while code-based skills execute without giving the agent step-level guidance for adaptation or recovery. We introduce WebXSkill, a framework that bridges this gap with executable skills, each pairing a parameterized action program with step-level natural-language guidance. WebXSkill operates in three stages: skill extraction mines reusable action subsequences from readily available synthetic agent trajectories and abstracts them into parameterized skills, skill organization indexes them into a URL-based graph for context-aware retrieval, and skill deployment exposes two complementary modes, grounded mode for fully automated execution and guided mode where skills serve as step-by-step instructions the agent follows with its native planning. WebXSkill demonstrates consistent improvements on WebArena, WebVoyager, and Online-Mind2Web. We further find that better skill deployment mode depends on a model’s plan and execution capability. The code is available at https://github.com/aiming-lab/WebXSkill.
LUPE: Long-History User Profile Evolution for LLM-based Recommendation
Yifei Sun | Lu Wang | Fangkai Yang | Pu Zhao | Liqun Li | Mingzheng Li | Jianjin Zhang | Jimeng Sun | Zhiwei Dai | Nan Hu | Zhengxin Zeng | Yuefeng Zhan | Weihao Han | Hao Sun | Qingwei Lin | Saravan Rajmohan | Weiwei Deng | Feng Sun | Qi Zhang | Dongmei Zhang
Findings of the Association for Computational Linguistics: EMNLP 2026
Yifei Sun | Lu Wang | Fangkai Yang | Pu Zhao | Liqun Li | Mingzheng Li | Jianjin Zhang | Jimeng Sun | Zhiwei Dai | Nan Hu | Zhengxin Zeng | Yuefeng Zhan | Weihao Han | Hao Sun | Qingwei Lin | Saravan Rajmohan | Weiwei Deng | Feng Sun | Qi Zhang | Dongmei Zhang
Findings of the Association for Computational Linguistics: EMNLP 2026
Textual user profiling is an effective paradigm for LLM-based recommendation, but existing methods usually treat profile generation as a one-shot summarization problem. This limitation becomes more severe in long-history recommendation, where early preference signals may be overcompressed and recent behaviors may be overemphasized. We propose LUPE, a Long-history User Profile Evolution framework for LLM-based recommendation. Instead of generating a static profile from the full history, LUPE maintains a persistent textual profile and updates it progressively over temporally ordered interaction chunks. This allows the model to preserve long-term preferences while incorporating new behavioral evidence. To learn how profiles should evolve, LUPE optimizes a profile evolution policy via reinforcement learning using feedback from a frozen downstream LLM-based recommender. This allows the model to directly align profile updates with recommendation utility rather than relying on heuristic or supervised profile targets. Experiments on Yelp, Amazon CDs and Vinyl, and Amazon Books show that LUPE consistently outperforms strong baselines on both rating prediction and ranking tasks. Further analyses show that explicit profile evolution is especially helpful for users with long and heterogeneous interaction histories. The code repository: https://github.com/LongHistoryProfile/long_history_profile.
PEARL: Perturbation Efficient Alignment Group Relative Reinforcement Learning
Yue Huang | ChenZhuo Zhao | Zhi Yang | Pu Zhao | Fangkai Yang | Lu Wang | Mengqiao Liu | Qingwei Lin | Saravan Rajmohan | Dongmei Zhang
Findings of the Association for Computational Linguistics: EMNLP 2026
Yue Huang | ChenZhuo Zhao | Zhi Yang | Pu Zhao | Fangkai Yang | Lu Wang | Mengqiao Liu | Qingwei Lin | Saravan Rajmohan | Dongmei Zhang
Findings of the Association for Computational Linguistics: EMNLP 2026
Reinforcement learning with verifiable rewards (RLVR) has become a widely used approach for improving the mathematical reasoning ability of large language models (LLMs). However, recent studies show that LLMs may solve a math problem in its original form but fail when the same problem is reworded without changing its meaning. This suggests that the model has not fully acquired the underlying reasoning ability and instead relies on the surface form of the problem. In this paper, we propose PEARL, an RLVR framework that uses different versions of the same math problem as structured training signals. PEARL first groups each original problem with its checked variants to build semantic perturbation clusters, and then uses partially solved clusters as the active RL training set. PEARL further performs cluster-level RL optimization by comparing model outputs across different versions of the same mathematical problem. Experiments on mathematical reasoning benchmarks show that PEARL achieves strong results against competitive RLVR baselines while using a small active training set. These results suggest that semantically equivalent variants can serve as useful training units for improving mathematical reasoning in LLMs.
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/.
GUI-360° : A Comprehensive Dataset and Benchmark for Desktop Productivity Computer-Using Agents
Jian Mu | Chaoyun Zhang | Chiming Ni | Lu Wang | Bo Qiao | Kartik Mathur | Qianhui Wu | Yuhang Xie | Xiaojun Ma | Mengyu Zhou | Si Qin | Liqun Li | Yu Kang | Minghua Ma | Qingwei Lin | Saravan Rajmohan | Dongmei Zhang
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Jian Mu | Chaoyun Zhang | Chiming Ni | Lu Wang | Bo Qiao | Kartik Mathur | Qianhui Wu | Yuhang Xie | Xiaojun Ma | Mengyu Zhou | Si Qin | Liqun Li | Yu Kang | Minghua Ma | Qingwei Lin | Saravan Rajmohan | Dongmei Zhang
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
We introduce GUI-360°, a large-scale dataset and benchmark suite for desktop productivity computer-using agents (CUAs). GUI-360° provides an automated pipeline for query sourcing, template-based environment construction, task instantiation, batched execution, and LLM-assisted quality filtering in Windows office applications. The corpus contains 131K validated successful action steps across 17K trajectories for training and evaluation, together with 1.09M failed execution steps for failure analysis and future recovery training. Each step provides screenshots, accessibility metadata, instantiated goals, reasoning traces, and structured GUI/API actions. GUI-360° supports GUI grounding, screen parsing, and action prediction under a hybrid GUI+API action space. Experiments show that current vision–language models struggle with fine-grained grounding and stepwise action prediction, while supervised fine-tuning on GUI-360° brings consistent in-domain and external benchmark gains. Our code and dataset are available at https://github.com/2020-qqtcg/GUI-360 and https://huggingface.co/datasets/vyokky/GUI-360.
Beyond State Consistency: Behavior Consistency in Text-Based World Models
Youling Huang | Guanqiao Chen | Junchi Yao | Lu Wang | Fangkai Yang | Chao Du | ChenZhuo Zhao | Pu Zhao | Qingwei Lin | Saravan Rajmohan | Dongmei Zhang
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Youling Huang | Guanqiao Chen | Junchi Yao | Lu Wang | Fangkai Yang | Chao Du | ChenZhuo Zhao | Pu Zhao | Qingwei Lin | Saravan Rajmohan | Dongmei Zhang
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
World models are emerging as critical components for assessing the consequences of actions taken by interactive agents in online planning and offline evaluation. In text-based environments, world models are typically trained and evaluated with single-step metrics such as Exact Match, aiming to improve the similarity between predicted and real-world states, but such metrics have been shown to be insufficient for capturing actual agent behavior. To address this issue, we introduce a new behavior-consistency training paradigm aimed at improving the functional consistency between the world model and the real environment. This paradigm focuses on optimizing a tractable step-level reward named Behavior Consistency Reward (BehR), which measures how much the likelihood of a logged next action changes between the real state and the world-model-predicted state under a frozen Reference Agent. Experiments on WebShop and TextWorld show that BehR-based training improves task-level functional consistency in several settings, with the clearest gains in WebShop and smaller changes in near-ceiling regimes, while preserving or improving single-step prediction quality in three of four settings. BehR-trained world models also reduce false positives in offline surrogate evaluation and improve inference-time lookahead planning over the corresponding supervised world-model baselines. Code: https://github.com/Ricardo-H/behr-wm.
Contrastive Attribution in the Wild: An Interpretability Analysis of LLM Failures on Realistic Benchmarks
Rongyuan Tan | Jue Zhang | Zhuozhao Li | Qingwei Lin | Saravan Rajmohan | Dongmei Zhang
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Rongyuan Tan | Jue Zhang | Zhuozhao Li | Qingwei Lin | Saravan Rajmohan | Dongmei Zhang
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Interpretability methods are increasingly invoked to diagnose Large Language Model (LLM) failures, but most evidence comes from short prompts or toy templates, leaving their utility on practical benchmarks unclear. We cast failure diagnosis as contrastive attribution over target–contrast token pairs, attributing the logit gap between incorrect and correct output tokens to input tokens and internal states, and introduce a batch-packed multi-target backpropagation scheme that scales cross-layer attribution graphs to long-context inputs. Instantiated with LRP-style attribution, the framework is evaluated on IFEval, GAIA2, MATH, and EvalPlus across multiple model sizes and post-training checkpoints. Along the four axes of coverage, visibility, reliability, and dynamics, contrastive attribution surfaces recurring failure modes such as underweighting task-relevant tokens and overweighting distractors, demonstrating its practical value as a scalable diagnostic tool for realistic LLM failures.
Process Rewarding via Solution-Space Alignment
ChenZhuo Zhao | Yue Huang | Pu Zhao | Fangkai Yang | Lu Wang | Lin Ma | Zhi Yang | Youling Huang | Xinda Wang | Junting Lu | Qingwei Lin | Saravan Rajmohan | Dongmei Zhang
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
ChenZhuo Zhao | Yue Huang | Pu Zhao | Fangkai Yang | Lu Wang | Lin Ma | Zhi Yang | Youling Huang | Xinda Wang | Junting Lu | Qingwei Lin | Saravan Rajmohan | Dongmei Zhang
Proceedings of the 2026 Conference on Empirical Methods in Natural Language Processing
Reinforcement learning with verifiable or outcome-level rewards has recently become an effective post-training paradigm for improving the reasoning abilities of large language models, with GRPO-style optimization achieving strong results on mathematical and other verifiable tasks. However, for multi-step reasoning, outcome-level supervision remains intrinsically sparse: a trajectory that contains partially correct intermediate derivations but makes a late mistake is still treated as a complete failure. As a result, useful reasoning prefixes are penalized together with erroneous continuations, and informative failed trajectories are underutilized. PRMs provide denser step-level feedback, but they often rely on costly process annotation or auxiliary process verifiers, and optimizing local process rewards alone may fail to preserve the global objective of final correctness, leaving room for reward misspecification and reward hacking during reinforcement fine-tuning. We propose PRSA(Process Rewarding via Solution-Space Alignment), a simple framework that bridges outcome supervision and process supervision without requiring human step-level labels. PRSA automatically extracts intermediate-value steps from model outputs and reference solutions, aligns them within the solution space under equivalence and order constraints to estimate reasoning progress, and uses this signal for both reward shaping and token-level prefix-suffix credit assignment. In this way, PRSA preserves the global anchor of final-answer correctness while providing denser and better localized learning signals for partially correct trajectories.
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 18
- Dongmei Zhang 18
- Fangkai Yang 10
- Pu Zhao 10
- Lu Wang 9
- Qianhui Wu 4
- Jue Zhang 4
- ChenZhuo Zhao 4
- Chetan Bansal 3
- Yue Huang 3
- Bo Qiao 3
- Si Qin 3
- Zhaoyang Wang 3
- Huaxiu Yao 3
- Chaoyun Zhang 3
- Xuchao Zhang 3
- Yue Chen 2
- Zhiwei Dai 2
- Weiwei Deng 2
- Jianfeng Gao 2
- Weihao Han 2
- Minghua He 2
- Nan Hu 2
- Youling Huang 2
- Yu Kang 2
- Liqun Li 2
- Junting Lu 2
- Baolin Peng 2
- Xiaoting Qin 2
- Feng Sun 2
- Hao Sun 2
- Yifei Sun 2
- Rujia Wang 2
- Xinda Wang 2
- Zhi Yang 2
- Yuefeng Zhan 2
- Jianjin Zhang 2
- Qi Zhang 2
- Fengshuo Bai 1
- Anson Bastos 1
- Guanqiao Chen 1
- Xu Chen 1
- Sitao Cheng 1
- Zhun Deng 1
- Hang Dong 1
- Yifei Dong 1
- Chao Du 1
- Fazle Elahi Faisal 1
- Wenjie Fu 1
- Haoxiang Gao 1
- Supriyo Ghosh 1
- Gaopeng Gou 1
- Jiawei Han 1
- Siwei Han 1
- Minjie Hong 1
- Shaohan Huang 1
- Shujian Huang (书剑 黄) 1
- Ran Jia 1
- Bowen Jiang 1
- Huiqiang Jiang 1
- Yizhu Jiao 1
- Mingzheng Li 1
- Zhuozhao Li 1
- Yiming Liang 1
- Mengqi Liao 1
- Jia Liu 1
- Mengqiao Liu 1
- Ziqian Liu 1
- Lin Ma 1
- Minghua Ma 1
- Xiaojun Ma 1
- Yubo Ma 1
- Kartik Mathur 1
- Lingrui Mei 1
- Jian Mu 1
- Suman Nath 1
- Chiming Ni 1
- Zhiyuan Peng 1
- Meikang Qiu 1
- Victor Rühle 1
- Shivam Shandilya 1
- Robert Sim 1
- Jimeng Sun 1
- Lihao Sun 1
- Rongyuan Tan 1
- Yuanmin Tang 1
- Huaiyu Wan 1
- Lu Wang 1
- Qibin Wang 1
- Furu Wei 1
- Qi Wu 1
- Lukas Wutschitz 1
- Menglin Xia 1
- Yuhang Xie 1
- Gang Xiong 1
- Junchi Yao 1
- Wenlin Yao 1
- Wenjie Yin 1
- Xin Yin 1
- Jing Yu 1
- Zhengxin Zeng 1
- Qi Zhang 1
- Zhiyang Zhang 1
- Mengyu Zhou 1
- Runchuan Zhu 1
- Ziyuan Zhuang 1