Android in the Wild: A Large-Scale Dataset for Android Device Control

TL;DR

The Android in the Wild (AITW) dataset offers 715k device interaction examples across multiple Android versions and device types.

cs.LG 🔴 Advanced 2023-07-19 5 views
Christopher Rawles Alice Li Daniel Rodriguez Oriana Riva Timothy Lillicrap
dataset device control natural language processing user interface gesture recognition

Key Findings

Methodology

The AITW dataset encompasses 715k device interaction episodes, covering 30k unique instructions, supporting Android versions 10 to 13 and 8 device types. It records multi-step tasks requiring semantic understanding of language and visual context. Two agent models were developed for performance evaluation.

Key Results

  • The diversity of the AITW dataset includes 357 apps and websites, with an average task step of 6.5.
  • Behavioral cloning agents achieved a complete match rate of 65% on the standard test set.
  • The LLM-based model excelled in unseen instruction templates, demonstrating strong generalization capabilities.

Significance

The release of the AITW dataset provides a large-scale, diverse benchmark for device control system research, promoting robustness analysis under new task descriptions, applications, or platform versions. This is significant for developing automated systems capable of understanding and executing natural language instructions.

Technical Contribution

The technical contribution of the AITW dataset lies in its scale and diversity, surpassing existing datasets. It offers a rich resource for training and evaluating device control systems, supporting semantic understanding of multi-step tasks and precise gesture operations.

Novelty

AITW is the first large-scale, multi-version, multi-device Android device control dataset, overcoming limitations in task diversity and scale of existing datasets.

Limitations

  • The accuracy of gesture recognition is limited by differences in screen resolution and device types.
  • OCR text recognition may have errors affecting model accuracy.

Future Work

Future work could include developing more accurate gesture recognition models and improving OCR text recognition to enhance device control system performance.

AI Executive Summary

With the proliferation of mobile devices, there is an increasing demand for systems that can understand natural language instructions and perform device operations. However, existing datasets are limited in scale and diversity, hindering research progress.

To address this issue, researchers have released the Android in the Wild (AITW) dataset. This dataset includes 715k device interaction examples, covering 30k unique instructions, supporting Android versions 10 to 13 and 8 device types. AITW not only surpasses existing datasets in scale but also offers a rich variety of tasks and device types.

The release of the AITW dataset provides a large-scale, diverse benchmark for device control system research, promoting robustness analysis under new task descriptions, applications, or platform versions. This is significant for developing automated systems capable of understanding and executing natural language instructions. While the dataset has certain limitations in gesture recognition and OCR text recognition, it offers valuable resources and directions for future research.

Deep Analysis

Background

In recent years, research on device control systems has been on the rise. These systems can manipulate device user interfaces through natural language instructions. However, existing datasets are limited in scale and diversity, restricting research progress. The release of the AITW dataset fills this gap, providing a large-scale, diverse benchmark for research.

Core Problem

Existing device control datasets are limited in task diversity and scale, unable to support semantic understanding of multi-step tasks and precise gesture operations. The AITW dataset aims to address these issues and advance research in device control systems.

Innovation

The innovation of the AITW dataset lies in its scale and diversity, covering multiple Android versions and device types. It records multi-step tasks requiring semantic understanding of language and visual context, supporting precise gesture operations.

Methodology

  • �� Data Collection: The AITW dataset includes 715k device interaction episodes, covering 30k unique instructions.
  • �� Data Processing: Screenshots are processed using OCR and icon detection techniques.
  • �� Model Development: Two agent models were developed for performance evaluation.

Experiments

The experimental design includes evaluating two agent models using the AITW dataset. Performance tests were conducted using standard test sets and OOD settings to assess model generalization on unseen task descriptions and applications.

Results

Experimental results show that behavioral cloning agents achieved a complete match rate of 65% on the standard test set. The LLM-based model excelled in unseen instruction templates, demonstrating strong generalization capabilities.

Applications

The AITW dataset can be used for training and evaluating device control systems, supporting semantic understanding of multi-step tasks and precise gesture operations. It provides valuable resources for developing automated systems capable of understanding and executing natural language instructions.

Limitations & Outlook

The AITW dataset has certain limitations in gesture recognition and OCR text recognition, affecting model accuracy. Future research can improve these technologies to enhance device control system performance.

Plain Language Accessible to non-experts

Imagine you're using your phone to complete a task, like opening your calendar to check next week's schedule. The AITW dataset is like a huge guide that teaches computers how to complete these tasks through natural language instructions. It records how people operate on different phones and apps, like swiping the screen or tapping buttons. By learning these operations, computers can better understand and execute our instructions, just like a smart assistant.

ELI14 Explained like you're 14

Hey, imagine you're playing a super complex mobile game with lots of levels and tasks. The AITW dataset is like a game guide that helps computers learn how to complete these tasks through natural language instructions. It records how people operate on different phones and apps, like swiping the screen or tapping buttons. This way, computers can become as smart as you in completing tasks!

Glossary

Behavioral Cloning

A machine learning method that trains models by imitating human behavior.

Used for training agent models to evaluate the AITW dataset.

Natural Language Processing

Technology for computers to understand and generate human language.

Used to parse and execute device control instructions.

OCR (Optical Character Recognition)

Technology that recognizes text in images and converts it to editable text.

Used to process screenshots in the AITW dataset.

Gesture Recognition

Technology for recognizing and interpreting human gestures.

Used for precise operations in device control systems.

OOD (Out-of-Distribution)

Refers to a model's performance on unseen data or tasks.

Used to evaluate agent models' generalization capabilities on the AITW dataset.

Open Questions Unanswered questions from this research

  • 1 How to improve gesture recognition accuracy to adapt to different devices and screen resolutions?
  • 2 How to enhance OCR technology to reduce text recognition errors?

Applications

Immediate Applications

Device Automation

Develop device control systems that understand and execute natural language instructions, applicable to smartphones and tablets.

Long-term Vision

Accessibility Technology

Develop more user-friendly device control systems for people with disabilities, improving their quality of life.

Abstract

There is a growing interest in device-control systems that can interpret human natural language instructions and execute them on a digital device by directly controlling its user interface. We present a dataset for device-control research, Android in the Wild (AITW), which is orders of magnitude larger than current datasets. The dataset contains human demonstrations of device interactions, including the screens and actions, and corresponding natural language instructions. It consists of 715k episodes spanning 30k unique instructions, four versions of Android (v10-13),and eight device types (Pixel 2 XL to Pixel 6) with varying screen resolutions. It contains multi-step tasks that require semantic understanding of language and visual context. This dataset poses a new challenge: actions available through the user interface must be inferred from their visual appearance. And, instead of simple UI element-based actions, the action space consists of precise gestures (e.g., horizontal scrolls to operate carousel widgets). We organize our dataset to encourage robustness analysis of device-control systems, i.e., how well a system performs in the presence of new task descriptions, new applications, or new platform versions. We develop two agents and report performance across the dataset. The dataset is available at https://github.com/google-research/google-research/tree/master/android_in_the_wild.

cs.LG cs.CL cs.HC