Generalization in Online Reinforcement Learning for Mobile Agents
Authors: Li Gu, Zihuan Jiang, Zhixiang Chi, Huan Liu, Ziqiang Wang, Yuanhao Yu, Glen Berseth, Yang Wang
Organizations: Mila – Québec AI Institute · Concordia University · University of Toronto · McMaster University · Université de Montréal · CIFAR AI Chair
Graphical user interface (GUI)-based mobile agents automate digital tasks on mobile devices by interpreting natural-language instructions and interacting with the screen. While recent methods apply reinforcement learning (RL) to train vision-language-model(VLM) agents in interactive environments with a primary focus on performance, generalization remains underexplored due to the lack of standardized benchmarks and open-source RL systems. In this work, we formalize the problem as a Contextual Markov Decision Process (CMDP) and introduce AndroidWorld-Generalization, a benchmark with three increasingly challenging regimes for evaluating zero-shot generalization to unseen task instances, templates, and applications. We further propose an RL training system that integrates Group Relative Policy Optimization (GRPO) with a scalable rollout collection system, consisting of containerized infrastructure and asynchronous execution % , and error recovery to support reliable and efficient training. Experiments on AndroidWorld-Generalization show that RL enables a 7B-parameter VLM agent to surpass supervised fine-tuning baselines, yielding a 26.1% improvement on unseen instances but only limited gains on unseen templates (15.7%) and apps (8.3%), underscoring the challenges of generalization. As a preliminary step, we demonstrate that few-shot adaptation at test-time improves performance on unseen apps, motivating future research in this direction. To support reproducibility and fair comparison, we open-source the full RL training system, including the environment, task suite, models, prompt configurations, and the underlying infrastructure \footnote{https://github.com/zihuanjiang/AndroidWorld-Generalization}.
Figures & tables
Figure 1: Sample task instructions with corresponding screenshots from the train and test sets of the three unseen regimes in the AndroidWorld-Generalization benchmark. Red highlights the unseen scenarios: Instance, Template, and Application.
Regime
Overlap
Train
Test
Instance
Template
App
I/T/A
Diff.
I/T/A
Diff.
Unseen Instance
✗
✓
✓
1149 / 78 / 17
1.68
234 / 78 / 17
1.68
Unseen Template
✗
✗
✓
836 / 57 / 14
1.70
54 / 18 / 14
1.72
Unseen App
✗
✗
✗
905 / 62 / 12
1.68
48 / 16 / 5
1.69
Table 1: Statistics of the AndroidWorld-Generalization benchmark . “I”, “T”, and “A” denote Instances, Templates, and Applications; “Diff.” denotes average difficulty. Bold numbers indicate the number of train/test examples in each regime, while gray numbers indicate the templates and applications from which those examples are generated.
Figure 2: RL training system for mobile agent. We integrate GRPO with a scalable rollout collection system that parallelizes multiple environments. Docker containerization provides resource isolation and decouples trainer and environments through HTTP communication for reliability. Asynchronous rollouts eliminate synchronization bottlenecks, enabling more agent steps per unit time. Together, these three techniques facilitate reliable and efficient large-scale training.
Models
API-Based
Open-Source
Reproducible
Average (SR)
Proprietary model
GPT-4o ( Hurst et al., 2024 )
✓
✓
34.5
Claude Computer Use ( Anthropic, 2024 )
✓
✓
27.9
UGround+GPT-4o ( Gou et al., 2024 )
✓
✓
✓
44.0
Aria-UI+GPT-4o ( Yang et al., 2024 )
✓
✓
✓
44.8
Agent S2 ( Agashe et al., 2025 )
✓
54.3
Table 2: Comparison of methods on AndroidWorld. “API-Based” denotes use of proprietary inference APIs. “Open-Source” denotes availability of model weights. “Reproducible” denotes availability of full codebase.
Figure 3: Training dynamics on Unseen Instance with curriculum learning. Colored areas denote curriculum stages: blue (Easy), red (Easy + Medium), green (All). (Left) Average training and evaluation success rates. (Right) Average evaluation success rates by task type (Task Completion, Information Retrieval) and difficulty (Easy, Medium, Hard).
Figure 4: Training dynamics of GRPO across the three unseen regimes. We report training success rates and evaluation success rates with standard deviations.
Figure 5: Training dynamics of PPO across the three unseen regimes.
Figure 6: Few-shot adaptation vs. non-adaptation, averaged. over Unseen App test-set.
Figure 7: (Left) Time profiling of a policy update vs. the collection of 16 rollouts per training step. (Right) Performance of async vs. sync rollouts with varying environment counts.
Appendix figures & tables9 assets
Supplementary material from the paper’s appendix.
Appendix
AndroidWorld
AndroidWorld-Generalization
Environment
116 templates across 20 apps
Same 20 apps and templates (subset splits)
Task Suite
116 eval instances (1 seed / template)
Multi-seed eval instances; thousands of non-overlapping train instances
Generalization
None
Three unseen regimes (instance, template, app)
RL Support
No; evaluation-only
Yes; standardized train set for RL
Infrastructure
Sequential rollouts
Parallel rollout collection for scalable evaluation & RL training
Appendix
Table 3: Comparison between AndroidWorld and AndroidWorld-Generalization.
Models
Easy (SR)
Medium (SR)
Hard (SR)
Average (SR)
Close-source Models
GPT-4o ( Hurst et al., 2024 )
-
-
-
34.5
Claude Computer Use ( Anthropic, 2024 )
-
-
-
27.9
UGround+GPT-4o ( Gou et al., 2024 )
-
-
-
44.0
Aria-UI+GPT-4o ( Yang et al., 2024 )
-
-
-
44.8
Agent S2 ( Agashe et al., 2025 )
-
-
-
54.3
Appendix
Table 4: AndroidWorld evaluation performance for Mobile Agents. 3 3 3 Prior methds marked with ∗ are evaluated on a sub-set of AndroidWorld.
Figure 8: Case study 1.
Figure 9: Case study 2.
Figure 10: Case study 3.
Figure 11: Case study 4.
Figure 12: Case study 5.
Figure 17
Action Type
Description
Click (x,y)
Tap at screen coordinate (x,y)
Long press (x,y)
Long press at screen coordinate (x,y)
Type (text)
Enter variable-length natural-language text
Scroll (d)
Scroll in direction d∈{up, down, left, right}
Navigate home ()
Navigate Back to home screen
Navigate back ()
Navigate Back to previous screen
Appendix
Table 5: Action space of mobile use agents in AndroidWorld.