StepX-Edge: An On-Device UI Vision-Language Model via Architecture-Training-Deployment Co-Design
On the map: Multimodal LLMs / vision-language models → Embodied and 3D vision-language models
Abstract (arXiv)
Deploying a vision-language model with full UI understanding on end devices has long been trapped between accuracy and efficiency: on one side is the accuracy bar for OCR, screen understanding, visual question answering, and element grounding; on the other is the strict compute, memory, and power budget of mobile chips. Existing work either trades one for the other, or stops at simulation without real-device validation. We present StepX-Edge, a 0.9B-parameter on-device UI vision-language model that resolves this tension through three-layer co-design of architecture, training, and deployment. Architecturally, UI-aware Layered Visual Encoding (ULVE) and a Progressive Dimensionality Projection (PDP) connector target the extreme aspect ratios and fine-grained perception of screens, while standard full attention throughout ensures native compatibility with mainstream mobile NPU operators. For training, the five-stage StepX-Curriculum framework is designed around our observation of mutual-promotion effects among UI subtasks, so that all four capabilities grow synergistically under a tight parameter budget rather than interfering. For deployment, a module-wise differentiated two-stage PTQ-to-QAT quantization scheme keeps the post-quantization accuracy loss within 1%. StepX-Edge achieves the strongest overall UI understanding among <=1B models, surpassing all 2B-2.3B baselines on ScreenQA (88.76 F1) and Chinese OCRBench v2 (57.25), and matching 1.3B-2.3B general VLMs on RefCOCO (92.0%) and OCRBench v1 (831) with far fewer parameters. After W4A16+KV8 quantization, the model runs stably on Snapdragon 8 Gen5 devices with ~0.84 s TTFT, 98 tok/s decode, and 1.4 GB peak memory. We will open-source the training data, the full training recipe, and the quantization deployment pipeline.
Key facts. Cited by 1 paper in the map (data through 7 Oct 2026). Placed in Embodied and 3D vision-language models, one of 16 directions in Multimodal LLMs / vision-language models. Builds on 49 of its 74 references that are in the map. First posted on arXiv 20 Jul 2026.
Builds on
49 of 74 references.
References · 49
- 2017Attention Is All You Needcited by 59,932
- 2021Learning Transferable Visual Models From Natural Language Supervisioncited by 33,456
- 2024DeepSeekMath: Pushing the Limits of Mathematical Reasoning in Open Language Modelscited by 8,851
- 2023Direct Preference Optimization: Your Language Model is Secretly a Reward Modelcited by 7,916
- 2023Visual Instruction Tuningcited by 7,792
- 2023Efficient Memory Management for Large Language Model Serving with PagedAttentioncited by 5,767
- 2025Qwen2.5-VL Technical Reportcited by 5,418
- 2016SGDR: Stochastic Gradient Descent with Warm Restartscited by 4,993
- 2022Flamingo: a Visual Language Model for Few-Shot Learningcited by 4,337
- 2022LAION-5B: An open large-scale dataset for training next generation image-text modelscited by 4,131
- 2023Mistral 7Bcited by 3,065
- 2025Qwen3-VL Technical Reportcited by 2,534
37 more
- 2022Learn to Explain: Multimodal Reasoning via Thought Chains for Science Question Answeringcited by 1,881
- 2019Towards VQA Models That Can Readcited by 1,696
- 2023MathVista: Evaluating Mathematical Reasoning of Foundation Models in Visual Contextscited by 1,582
- 2019Generating Long Sequences with Sparse Transformerscited by 1,523
- 2022ChartQA: A Benchmark for Question Answering about Charts with Visual and Logical Reasoningcited by 1,449
- 2022SmoothQuant: Accurate and Efficient Post-Training Quantization for Large Language Modelscited by 1,338
- 2020DocVQA: A Dataset for VQA on Document Imagescited by 1,288
- 2021Conceptual 12M: Pushing Web-Scale Image-Text Pre-Training To Recognize Long-Tail Visual Conceptscited by 1,209
- 2023ShareGPT4V: Improving Large Multi-Modal Models with Better Captionscited by 1,000
- 2024SeeClick: Harnessing GUI Grounding for Advanced Visual GUI Agentscited by 553
- 2023OCRBench: On the Hidden Mystery of OCR in Large Multimodal Modelscited by 466
- 2024AndroidWorld: A Dynamic Benchmarking Environment for Autonomous Agentscited by 450
- 2021InfographicVQAcited by 430
- 2016Synthetic Data for Text Localisation in Natural Imagescited by 396
- 2019FUNSD: A Dataset for Form Understanding in Noisy Scanned Documentscited by 345
- 2023Android in the Wild: A Large-Scale Dataset for Android Device Controlcited by 314
- 2024Massive Activations in Large Language Modelscited by 277
- 2016COCO-Text: Dataset and Benchmark for Text Detection and Recognition in Natural Imagescited by 235
- 2019Image-based table recognition: data, model, and evaluationcited by 233
- 2024ShowUI: One Vision-Language-Action Model for GUI Visual Agentcited by 226
- 2014Synthetic Data and Artificial Neural Networks for Natural Scene Text Recognitioncited by 214
- 2024ALLaVA: Harnessing GPT4V-Synthesized Data for Lite Vision-Language Modelscited by 204
- 2024LLaVA-UHD: an LMM Perceiving Any Aspect Ratio and High-Resolution Imagescited by 193
- 2024On the Effects of Data Scale on UI Control Agentscited by 187
- 2025MiniCPM-V 4.5: Cooking Efficient MLLMs via Architecture, Data, and Training Recipecited by 174
- 2021TextOCR: Towards large-scale end-to-end reasoning for arbitrary-shaped scene textcited by 170
- 2024OCRBench v2: An Improved Benchmark for Evaluating Large Multimodal Models on Visual Text Localization and Reasoningcited by 165
- 2023SVIT: Scaling up Visual Instruction Tuningcited by 161
- 2026Qwen3.5-Omni Technical Reportcited by 146
- 2024RLAIF-V: Open-Source AI Feedback Leads to Super GPT-4V Trustworthinesscited by 129
- 2021Screen2Words: Automatic Mobile UI Summarization with Multimodal Learningcited by 119
- 2022Wukong: A 100 Million Large-scale Chinese Cross-modal Pre-training Benchmarkcited by 114
- 2024AMEX: Android Multi-annotation Expo Dataset for Mobile GUI Agentscited by 110
- 2025RLPR: Extrapolating RLVR to General Domains without Verifierscited by 83
- 2018COCO-CN for Cross-Lingual Image Tagging, Captioning and Retrievalcited by 80
- 2022ScreenQA: Large-Scale Question-Answer Pairs over Mobile App Screenshotscited by 65
- 2021Spatial Dual-Modality Graph Reasoning for Key Information Extractioncited by 43
Built on by
Citing papers per year: 2026: 1.
1 paper.
Similar, not cited
Close in wording, no citation either way
- 2026VisionPsy-Nano: Improving Accuracy, Efficiency, and Reliability in On-Device Vision-Language Models
- 2022Spotlight: Mobile UI Understanding using Vision-Language Models with a Focus
- 2025ZonUI-3B: A Lightweight Vision-Language Model for Cross-Resolution GUI Grounding
- 2026Mobile-O: Unified Multimodal Understanding and Generation on Mobile Device
- 2024Training a Vision Language Model as Smartphone Assistant
- 2026GUITrans2Act: Understanding User Operational Behaviors from Mobile GUI Interactions with Vision-Language Models
- 2025CANVAS: A Benchmark for Vision-Language Models on Tool-Based User Interface Design
- 2023ILuvUI: Instruction-tuned LangUage-Vision modeling of UIs from Machine Conversations