A Context Engineering Framework for Improving Enterprise AI Agents based on Digital-Twin MDP
Organizations: IBM Software Innovation Lab, Yorktown Heights, New York, USA
Abstract
Despite rapid progress in AI agents for enterprise automation and decision-making, their real-world deployment and further performance gains remain constrained by limited data quality and quantity, complex real-world reasoning demands, difficulties with self-play, and the lack of reliable feedback signals. To address these challenges, we propose a lightweight, model-agnostic framework for improving LLM-based enterprise agents via offline reinforcement learning (RL). The proposed Context Engineering via DT-MDP (DT-MDP-CE) framework comprises three key components: (1) A Digital-Twin Markov Decision Process (DT-MDP), which abstracts the agent's reasoning behavior as a finite MDP; (2) A robust contrastive inverse RL, which, armed with the DT-MDP, to efficiently estimate a well-founded reward function and induces policies from mixed-quality offline trajectories; and (3) RL-guided context engineering, which uses the policy obtained from the integrated process of (1) and (2), to improve the agent's decision-making behavior. As a case study, we apply the framework to a representative task in the enterprise-oriented domain of IT automation. Extensive experimental results demonstrate consistent and significant improvements over baseline agents across a wide range of evaluation settings, suggesting that the framework can generalize to other agents sharing similar characteristics in enterprise environments.
Figures & tables
| Baseline | Name | Name-Type | Topology | |||||||
|---|---|---|---|---|---|---|---|---|---|---|
| Cost | I | II | III | I | II | III | I | II | III | |
| In Tok.(K) | 439 | 525 | 412 | 554 | 601 | 451 | 539 | 571 | 305 | 541 |
| Out Tok.(K) | 4.7 | 4.6 | 3.1 | 5.7 | 4.8 | 3.0 | 6.1 | 4.9 | 3.2 | 5.1 |
| Time (s) | 841 | 886 | 818 | 924 | 968 | 869 | 888 | 986 | 778 | 894 |