Title: From Pixels to UI Actions: Learning to Follow Instructions via Graphical User Interfaces
Venue: NeurIPS 2023
Accepted: yes
Keywords: instruction following

web tasks

user interface tasks

vision and language

representation learning

reinforcement learning

imitation learning

tree search

language grounding

web agents

computer control
Abstract: Much of the previous work towards digital agents for graphical user interfaces (GUIs) has relied on text-based representations (derived from HTML or other structured data sources), which are not always readily available. These input representations have been often coupled with custom, task-specific action spaces.  This paper focuses on creating agents that interact with the digital world using the same conceptual interface that humans commonly use — via pixel-based screenshots and a generic action space corresponding to keyboard and mouse actions. Building upon recent progress in pixel-based pretraining, we show, for the first time, that it is possible for such agents to outperform human crowdworkers on the MiniWob++ benchmark of GUI-based instruction following tasks.

From Pixels to UI Actions: Learning to Follow Instructions via Graphical User Interfaces

Abstract

Much of the previous work towards digital agents for graphical user interfaces (GUIs) has relied on text-based representations (derived from HTML or other structured data sources), which are not always readily available. These input representations have been often coupled with custom, task-specific action spaces. This paper focuses on creating agents that interact with the digital world using the same conceptual interface that humans commonly use - via pixel-based screenshots and a generic action space corresponding to keyboard and mouse actions. Building upon recent progress in pixel-based pretraining, we show, for the first time, that it is possible for such agents to outperform human crowdworkers on the MiniWob++ benchmark of GUI-based instruction following tasks.

1 Introduction

Systems that can follow instructions to complete tasks through graphical user interfaces (GUIs) can help automate tedious tasks, improve accessibility, and expand the usefulness of digital assistants by allowing them to interact with tools and services. Despite the visual nature of GUIs, prior work has primarily focused on utilizing structured representations of the user interfaces (such as HTML sources, Document Object Model (DOM) trees, and Android view hierarchies) as well as custom, task-specific representations of high-level actions based on these structured representations (see §6). Recent efforts have achieved positive outcomes thanks to the advances of powerful language models (Gur et al., 2022; Kim et al., 2023; Yao et al., 2022).

While structured and task-specific representations may be useful, they are not always available some examples are web applications that use extensive scripting, sandboxed environments where access to DOM is limited, and mobile applications which often do not expose the underlying structure to external modules. Even when structured application source data is available, it may be hard to interpret due to obfuscation and misalignment with what actually appears on the GUIs. Finally, aligning human demonstrations with task-dependent actions is often challenging.

In contrast, people interact with GUIs by perceiving the visual input and using generic mouse and keyboard actions, without needing to inspect the application's source code for cues on its functionality. They can quickly learn to interact with new applications that offer familiar visual interfaces, regardless of differences in implementation technologies. In this paper we ask: Can we build an agent that can

Chrome-based

Environment

Pix2Act Agent

click X Y

begin_drag X Y

end_drag X Y

key [M] K

scroll Z

Possible Actions

Example Observations

Select 10/03/2016as the date

Select allthe shadesof green

Dragthesmallerboxsothatitis

nd theemailbyCourtneyand

andhit submit.

genuine and clickSubmit.

andpressSubmit.

ohplately inside the larger box.

lkthe staricon tomarkitas

important.

Date:

chop

Primary

November 2016

WoTh

genuine

miserable

In vel. Rhoncus..

Velamet.

☆

21

15

23

9

24

7

10

口

fearless

Submit

Risus auctor ad..

Orci.

☆

30

Submit

Anette

☆言

Submit

Tempor.

Feugiat. Sit eg..

iwant togetan8fluidouncepackof24sweet andsourmargarita anddaquiril

iwant togetan8fluidouncepackof24sweetandsourmargaritaanddaquiril

mixpacketsmadewithonlynatural ingredients,andpricelowerthan140.00

dollars

dollars

Ba

MAF

MARC

MARGARITA

XIW

WebShop

61g

Instruction:

i want to get an 8fluid ounce pack of 24 sweet and sour margarita and daquiri mix

10

packetsmade with only natural ingredlients,and price lower than 140.00 dollars

8 fluid ounce pack of 24 sweet and sourargarta and daquiri mix packets

QSearch

OriginalMargaritaMixSingles(8Single-Serve

Packets Per Box)-The Original Zero Sugar,Low

Calorie,LowCarb,KetoFriendly,SkinnyCocktail

Mixer.(3-PackMakes24IndividualCocktails)

Price: $22.92

Rating: N.A.

Description

Features

Reviews

BuyNow

complete tasks for users while relying solely on pixel-level visual representations of the GUI state, and generic low-level actions?

Learning based on pixel-only inputs proved effective for game playing environments such as Atari (Mnih et al., 2015). However, for GUI-based instruction following tasks, learning from pixel-only inputs coupled with general low-level actions leads to several challenges. Interpreting GUIs visually requires understanding the interface layout, recognizing and interpreting visually-situated natural language, identifying visual elements, and predicting their functions and methods of interaction. A generic action space also poses the challenge of a more complex mapping between high-level textual instructions and corresponding sequences of low-level actions. As an example of the increased difficulty in this setting, on the MiniWob++ benchmark (Shi et al., 2017; Liu et al., 2018) of web GUI interaction, CC-Net (Humphreys et al., 2022) demonstrates human-level accuracy when accessing both screenshots and DOM structure, but its performance drops by 75% when the DOM information is removed from the agent's observations.

Here we present PIX2ACT, a model that relies solely on pixel-based screenshots as input and selects actions corresponding to basic mouse and keyboard functionalities. 2 We build on PIX2STRUCT (Lee et al., 2022), a Transformer-based (Vaswani et al., 2017) image-to-text model pre-trained to map

screenshots to structured representations derived from HTML on web-scale data. PIX2ACT tunes this model using a combination of human demonstrations and environment interactions, applying tree search to iteratively generate new expert trajectories for training. We develop a general browser-based environment framework, and adapt two benchmark datasets, MiniWob++ and WebShop (Yao et al., 2022), to our setting with a unified, general purpose observation and action format.

On MiniWob++, PIX2ACT outperforms human crowdworkers and improves task score nearly 4x compared to the best prior results in our proposed setting (CC-Net without DOM). Ablations show that a key ingredient for PIX2ACT's performance is the pixel-based pre-training of PIX2STRUCT.

Our contributions are as follows:

We show, for the first time, that an agent using pixel-only inputs and a generic action space can outperform human crowdworkers on the MiniWob++ benchmark, significantly improving over prior work on this setting, and reaching performance comparable to that of state-of-the-art agents that access DOM information and use a comparable number of human demonstrations.

We adapt the WebShop benchmark to our setting, using pixel-based observations and general lowlevel actions. We establish the first baseline on this setting, although there is still a performance gap relative to larger language models using HTML-based inputs and task-specific actions.

We show that PIX2STRUCT's pre-training via screenshot parsing is effective for GUI-based instruction following with pixel-based inputs. In the behavioral cloning setting, pre-training improves task scores from 17.1 to 66.5 on MiniWob++ and from 1.1 to 46.7 on WebShop.

We demonstrate the successful application of tree search as a relatively simple method for policy improvement for MiniWob++.

2 Environment

Following the reinforcement learning literature, we model GUI interaction as a Markov Decision Process (MDP): at each time step, our agent receives an observation and selects an action. We develop a common environment framework with shared observation and action formats for browserbased tasks. Similarly to prior work on web-based agents (Liu et al., 2018), we use Selenium to programmatically interact with the Google Chrome browser.

Observations To form an observation, we first take a screenshot of the current browser window using Selenium and then augment it with additional information. First, if not already present, we render the natural language instruction on the top of the screenshot, following Lee et al. (2022). Second, as Selenium screenshots do not include cursors (which are typically rendered by the operating system), we draw a cursor on the screenshot to indicate the mouse pointer position. Finally, we render an indicator of whether the mouse button is currently pressed down, which is useful for dragging actions.

Actions Our action space consists of raw mouse and keyboard actions, as shown in Figure 1, where X and Y refer to discrete coordinate bins, K is one or more keys, M is an optional modifier key such as 'shift', and Z refers to a vertical scroll amount, also represented as a discrete bin. 3 The begin_drag and end_drag actions can be used to execute 'click and drag' actions. We use a configurable number of coordinate buckets per vertical and horizontal axis. Importantly, the DOM information is not provided by the environment and is therefore not used in any way to define observations or actions.

Episodes and Rewards Episodes continue until a terminal state or a configurable number of maximum steps is reached. For the environments we consider, the agent only receives a reward at a terminal state. This can be a binary reward based on whether the task was completed successfully or a partial reward based on how well the task was completed.

Step

1

2

3

4

5

Observation

Action

click

23

12

click

12

20

click

29

17

click

30

12

click

14

19

with

Select the following color  with

Select the following color

with

Select the following color

with

with

th color picker and hit Submit.

the color picker and hit Submit.

the color picker and hit Submit.

the color picker and hit Submit.

the color picker and hit Submit.

Color.

Color.

Color.

Color.

Color.

AB2567

AB2567

47AB69

5CDE88

5CDE88

Submit

Submit

Sglect the following color with

Select the following color  with

Select the following color  with

Select the following color with

Select the following color with

tcolor picker and hit Submit.

the color picker and hit Submit.

the color picker and hit Submit.

the color picker and hit Submit.

the color picker and hit Submit.

Color.

Color.

Color.

Color.

Color.

AB2567

AB2567

47AB69

5CDE88

5CDE88

Submit

Subnt

3 Proposed Agent

Our agent, PIX2ACT, is based on the PIX2STRUCT model (Lee et al., 2022), which uses an image Transformer encoder and a text Transformer decoder. The architecture is based on Vision Transformer (Dosovitskiy et al., 2021) and T5 (Raffel et al., 2020). PIX2STRUCT is pre-trained on a screenshot parsing task: predicting simplified HTMLs from screenshots with visually-masked regions. Such pre-training was proven effective for tasks related to understanding user interfaces in a non-interactive setting, such as screen summarization and widget captioning (Wang et al., 2021; Li et al., 2020b). We use the PIX2STRUCT base variant with 282M parameters (12 encoder and 12 decoder layers; hidden size 768) for all our experiments. The model is called once per time step.

Input The only input to the model is pixel-based observation from the environment. We can also condition on multiple previous observations by concatenating multiple frames. In preliminary experiments, we did not observe significant gains from conditioning on past observations for MiniWob++, and thus we only use the screenshot of the current step in our experiments. We reuse PIX2STRUCT's image processing by scaling input images up or down so as to extract the maximal number of fixed-size patches that still fit within the sequence length limit. We use resolutions of 160 × 210 and 800 × 600 for MiniWoB++ and WebShop, respectively.

Output We encode actions as text tokens, which are predicted autoregressively by the Transformer decoder. We use beam search over tokens to output the k -best actions (see Appendix B.1 for details).

Greedy Policy For interacting with the environment, we adopt a standard greedy policy, selecting the highest scoring action at each step, with one modification. To help prevent the agent from getting stuck in cycles, we track which actions have been taken for a given observation, and select the highest probability action in the beam that has not previously been taken given the current observation, which provides a modest increase in performance.

3.1 Training

We explore two methods for training models to follow instructions via GUIs. First, similarly to prior work, we use Behavioral Cloning (BC), where we train our model using standard supervised learning to predict the given action for each observation in a set of human demonstrations. Second, given access to environments with reward signals, prior work has also explored Reinforcement Learning (RL) to further improve agent performance. As an alternative to common reinforcement learning algorithms such as REINFORCE (Williams, 2004) and PPO (Schulman et al., 2017), we apply tree search as a simple method for policy improvement.

Tree Search For a given set of model parameters, tree search leverages the deterministic nature of the environment to look ahead at the consequences of possible actions to determine a more optimal policy than greedily selecting actions.

We adopt Monte Carlo Tree Search (MCTS) (Coulom, 2006), which outperformed more naive search algorithms in initial experiments, and has been successfully integrated with neural network policies in prior work (Silver et al., 2017; Anthony et al., 2017). Similarly to this prior work, we train a model to estimate a value function , which predicts the value (i.e., estimated future rewards) of a given state. We use a surrogate reward which penalizes the number of steps taken to encourage concise trajectories without unnecessary actions. We implement this value function approximator using the same PIX2STRUCT architecture used for our agent. 4 However, instead of predicting actions, this model predicts state-values mapped to discrete buckets. To estimate the value of leaf states during MCTS, we use a combination of this value function approximator and rollouts using our greedy policy, similarly to Silver et al. (2017). See Appendix B for additional technical details.

We can then use successful episodes found with this stronger tree search policy to improve our model. As this stronger model then yields a more effective tree search policy, we can continue to iteratively improve our model using this method. Notably, this approach requires no modifications to the fine-tuning procedure of PIX2ACT, as, for simplicity, we tune on episodes from the tree search policy using standard supervised learning.

4 Benchmarks and Demonstrations

We adapt two benchmarks, MiniWob++ and WebShop, to our environment framework (§2) which consists of pixel-based observations and generic low-level actions. We also map previously collected human demonstrations for these benchmarks to our observation and action spaces.

4.1 MiniWob++

MiniWob++ (Liu et al., 2018) is a set of over a hundred web-browser based tasks. See Figures 1 and 2 for task examples. Each task consists of an algorithm for generating variations of the task and an instruction template, controlled by a random seed, with up to billions of possible configurations per task. The task instruction is given as (mostly) natural language text in the top yellow part, which in our framework can only be accessed visually. An automatic reward is given at the end of the task.

Human Demonstrations We use the human demonstrations collected by Humphreys et al. (2022). However, their demonstrations were collected using an X11-based environment, which is different from our Selenium-based environment. This results in different renderings of the same underlying environment state, introducing a shift between the screenshots seen during training and those observed at test time. Additionally, we need to map from their real-time X11-based action sequences to our action space. We were able to perform this mapping with a reasonable degree of success for 59 tasks. Notably, not all behaviors in the human demonstrations are supported in our Selenium-based environment. For example, Selenium does not implement the ability to highlight text and drag it into a text field, and such an action is widely used in the human demonstrations for tasks where text is copied and pasted. Additionally, while our environment framework intends to cover the basic functionality of most web interfaces, aspects of some MiniWob++ tasks, such as capturing real-time observations for animated elements, are not supported. See Appendix A for additional details. 5

Starting with approximately 1.3 million demonstrations across the 59 supported tasks, we filtered demonstrations with a reward of < 0 . 8 , or approximately 6% of demonstrations. We were able to successfully convert 81% of the remaining demonstrations to our action space. We reserve 10% of the data for a development set. Demonstrations contain approximately 3 steps per task on average, although this varies considerably across tasks.

MiniWob++ Mean Scores

Ours

Ours

(BC only)

Ours

(BC only

no pretrain.)

CC-Net

(no DOM)

Human

SoTA w/

DOM or

HTML

0

25

50

75

100

96

.

2

66

.

5

17

.

1

24

.

1

94

.

4

96

.

3

WebShop Scores

Ours

Ours

(no pretrain.)

SoTA w/

DOM or

HTML

0

25

50

75

100

46

.

7

1

.

1

67

.

5

Evaluation We report the mean score across seeds and tasks. The score is the MiniWob++ raw reward (without time decay) mapped from the original range [ -1 , 1] to the range [0 , 100] . The score is equivalent to the success rate ( i.e . the proportion of episodes in which the agent receives a positive reward) for tasks with binary rewards. For episodes that do not complete due to reaching a maximum number of allowed steps, we assume a score of 0 . For each task, we compute the mean over 100 random seeds, and then compute the mean over 59 MiniWob++ tasks.

4.2 WebShop

WebShop (Yao et al., 2022) is a web-based shopping environment with over 1.1 million products from Amazon. The task is to find and purchase a product based on a human-authored text instruction. Finding a suitable product requires entering search queries, clicking on results, and determining the relevance of various products to the instruction. An automatic reward is computed based on similarity between the purchased product and the gold target product.

Human Demonstrations We use the 1,566 human demonstrations (with a train/development/test split of 1012/54/500) collected in Yao et al. (2022). As with the MiniWob++ demonstrations, we need to map between the observation and action sequences used in their setup to our framework. Yao et al. (2022) used high-level actions ( e.g . ' search ' or ' click[item] '), each of which could map to multiple lower-level actions in our environment. Specifically, for all actions involving a mouse click, we determine the coordinates of the center of the corresponding HTML element. For WebShop, the entire screen content is not always visible due to page heights exceeding the viewport dimensions. If the clicked element lies outside the visible area, we add scroll actions until the element is visible. Finally, we map search actions to two actions in our environment: clicking on the center of the search box and entering the search query followed by the enter key. We render the HTML inputs in the human demonstrations using our browser to obtain screenshots. Additionally we found that rendering the last 5 actions (separated by 
 ) on top of the screenshot to be helpful.

Evaluation Consistent with previous work, we report Task Score, which is the average reward across 500 test instructions.

5 Experiments and Analysis

5.1 Training Details

We updated all model parameters during fine-tuning, including both the image encoder and text decoder. We used the Adafactor optimizer (Shazeer and Stern, 2018) with a learning rate of 0.01.

MiniWoB++ We finetuned a single model jointly on episodes from all tasks for a total of 26K steps using a batch size of 512, input/output sequence lengths of 512/16. We also evaluated using the tree search procedure described in §3.1 to improve our agent. We performed 2 iterations of policy

Crowdworkers are better

PIX2ACT is better

choose-date

click-checkboxes-soft

count-shape

number-checkboxes

use-slider

click-collapsible

click-link

click-tab-2-easy

click-test

circle-center

drag-box

click-dialog

click-collapsible-2

click-pie

navigate-tree

click-test-transfer

click-test-2

click-option

click-button

unicode-test

use-autocomplete

click-tab-2-hard

email-inbox-delete

click-tab

click-dialog-2

email-inbox-important

click-tab-2

use-colorwheel-2

click-color

drag-items-grid

identify-shape

click-checkboxes-transfer

drag-item

drag-shapes

find-midpoint

count-sides

enter-time

drag-sort-numbers

enter-date

use-slider-2

click-checkboxes

visual-addition

click-tab-2-medium

bisect-angle

simple-arithmetic

grid-coordinate

resize-textarea

click-button-sequence

click-shape

enter-text-2

text-transform

use-colorwheel

drag-items

click-shades

right-angle

tic-tac-toe

click-checkboxes-large

simple-algebra

click-widget

0

100

improvement with tree search, collecting a total of 826K episodes across all tasks, and tuning for a further 26K steps.

WebShop We used only the provided human demonstrations to train our model. 6 Due to its larger resolution and text-heavy data, we used a higher input sequence length of 4096. We also found it useful to perform intermediate finetuning on MiniWoB++, followed by 10K steps of further finetuning on WebShop using a batch size of 256 (see §5.3 for details).

5.2 Main Results

We report the results of our models on MiniWob++ and WebShop in Figure 3. For MiniWob++, we also provide task-level comparisons between PIX2ACT and human crowdworkers in Figure 4. There is limited prior work studying these tasks without access to DOM and HTML information. For MiniWob++, the only comparable baselines are from the CC-Net model of Humphreys et al. (2022), which mentions an ablation experiment where performance dropped by 75% from their primary results when the models conditioned on only screenshots without DOM information. As they did not provide per-task numbers for this ablation, we estimate the performance of CC-Net without DOM information by assuming that the drop in performance on the subset of tasks we study was also 75%. Regardless, it is clear that PIX2ACT significantly outperforms CC-Net on this setting. The difference in performance can be largely attributed to the screenshot parsing pre-training of Lee et al. (2022). For WebShop, there is no prior work exploring such a setting, so we establish the first baseline.

5.3 Ablations and Analysis

Pre-training ablations To study the impact of the pre-training on our model's ability to effectively learn to follow instructions via GUIs, we evaluate model performance without the pre-training procedure. For these experiments, we only compared performance of models trained using behavioral cloning. The results are shown in Figure 3, and demonstrate that pre-training is critical for our model's performance.

Comparison with models that use DOM or HTML as input We can also compare our results without access to DOM or HTML to previous methods that utilized these resources, including those which also leverage DOM information to construct specialized action spaces. The performance of the best model from prior work leveraging DOM or HTML information is shown in Figure 3.

For MiniWob++, the best model on this setting is CC-Net (Humphreys et al., 2022) trained with BC and RL and with access to both DOM and pixel-based observations. 7 PIX2ACT achieves comparable performance to their best model, while relying on only a subset of the information used by CC-Net, and using a comparable number of human demonstrations for training. PIX2ACT also outperforms

CC-Net when each model is trained only with behavioral cloning, as CC-Net performance on this setting drops to 38.7 (results not shown in the Figure). Notably, CC-Net scores also drop by approximately 10% when the model is not given access to a dictionary of input strings provided by the environment. As shown in Figure 3, the key to our model's ability to achieve comparable performance without relying on DOM-based inputs is pixel-based pre-training. Another difference is that CC-Net uses a real time setting, which enables some forms of interaction not supported by our environment, and therefore can support a larger set of MiniWob++ tasks. On the other hand, for BC, CC-Net does not need to handle the shift in rendering format and potentially noisy action space conversion.

For WebShop, the best model on this setting is WebGUM (Furuta et al., 2023a), which leverages the HTML source, a custom action space for the shopping domain, and a Flan-T5-XL (Chung et al., 2022) backbone. WebGUM outperforms PIX2ACT when compared on this setting. Some of this gap can be attributed to their simplified high-level action space, direct access to the relevant text on the page, and ability to transfer from Flan-T5's pretraining scale and instruction finetuning. Comparable improvements to the scale and pretraining of pixel-based models could reduce this gap.

We discuss other approaches that leverage DOM or HTML information further in §6. We also offer a complete comparison across all MiniWob++ tasks in Appendix C.

Evaluating transfer across tasks Training a pretrained, pixel-based model to interact with a GUI can intuitively lead to better generalization to new tasks that use common GUI design principles. To study this, we evaluate the ability of PIX2ACT (without RL) to generalize to tasks unseen during training. Specifically, we hold out 9 out of 59 tasks and train on the remaining 50. 8 We then evaluate performance on the held-out tasks, comparing initializing with PIX2STRUCT to random initialization. Table 1 illustrates that PIX2ACT can reach a mean score of 28.3 on held out tasks compared to 65.5 when training on those tasks. Conversely, mean score is 7.6 when PIX2STRUCT initialization is not used. This shows that combining pretraining with a general GUI interface can lead to non-trivial generalization to held out tasks.

For WebShop, we find that finetuning directly on WebShop (without intermediate finetuning on MiniWoB++ as mentioned in 5.1) results in a drop of 4.0 in Task Score, demonstrating transfer learning benefits across these datasets.

Tree search analysis Table 2 shows the improvement in MiniWob++ scores by training on episodes generated by tree search. After an initial round of training on episodes generated by tree search, the effectiveness of tree search also improves due to improvements in the underlying model used to guide the search. The best greedy policy achieves performance close to the best tree search policy, but does not require access to reward signals or additional exploration at inference time. Our results indicate that we could further improve performance with more iterations of policy improvement via tree search.

6 Related Work

We focus on agents that interact with GUIs, such as operating system dialogs or web pages, to accomplish a given task. Many early approaches relied on the structured information from the GUIs (Zettlemoyer and St. Amant, 1999; Allen et al., 2007; Branavan et al., 2010). This information could range from a flat list of GUI components and their properties, to the full hierarchical structure of the components ( e.g . the DOM tree). The output space also depends on this structured information, often using GUI components as action targets ( e.g . clicking button #7). As discussed in §1, such structured information might not always be available, or might not align with what visually appears to the users.

When Shi et al. (2017) introduced the World of Bits tasks, which was the precursor to MiniWob++ (Liu et al., 2018), they proposed a model based on a convolutional neural network that takes both visual and structured inputs and then performs generic low-level computer actions ( e.g . clicking at a coordinate or pressing a key), similarly to PIX2ACT. However, the model performed poorly compared to humans. Follow-up work studied specialized architectures for incorporating structured DOMinformation and restricted the action space to clicking and typing predetermined texts on DOM elements (Liu et al., 2018; Gur et al., 2018; Jia et al., 2019). Humphreys et al. (2022) reconsidered incorporating both visual and structured information as well as a low-level action space that aligns better to the human demonstrations. We discussed their approach, CC-Net, in §5.3. Humphreys et al. (2022) also explored the benefits of large-scale human demonstrations, and we build on their work to utilize a large number of human demonstrations to train PIX2ACT. This paper shows that PIX2ACT, a model with pixel-only inputs, can outperform humans on MiniWob++ and match the state-of-the-art approaches that rely on DOM information.

Automating web-based tasks using large language models (LLMs) has also been broadly explored. For instance, WebGPT uses a text-based web browsing environment to search and navigate the web (Nakano et al., 2021). More relatedly, recent work has investigated prompting LLMs to produce agents that can generalize to tasks based on a small number of in-context examples. Yao et al. (2023) proposed ReAct, a few-shot prompted LLM, which uses observations derived from HTML and a custom action space to make predictions based on explicit reasoning steps. Similarly, Kim et al. (2023) proposed RCI, a prompted LLM that iteratively critiques and refines its outputs, also using HTML inputs and custom action spaces. These approaches achieve competitive performance on WebShop and MiniWob++, respectively, and are extremely sample-efficient, relying on just a handful of demonstrations per task. Gur et al. (2022) treated raw HTML as a string and fed it to LLMs pretrained on natural language. After fine-tuning them on demonstrations, the models improved MiniWob++ task success rate and sample efficiency compared to models that take DOM-based inputs and specialized architectures. Finally, WebGUM (Furuta et al., 2023b), discussed in §5.3, extends HTML-based models to integrate a vision encoder pretrained on ImageNet-21K.

Other work has focused on tasks related to mobile apps. Li and Li (2022) considered a model with pixel-based inputs similar to that of Lee et al. (2022), and included evaluations on tasks related to grounding instructions to screenshots, but did not consider interactive environments. Some work has considered instruction following tasks in mobile app environments (Li et al., 2020a; Burns et al., 2022), but has generally not studied observation and action formats similar to ours, instead relying on inputs based on the Android view hierarchy. We focused on web-based GUIs so that we could use a consistent environment framework for simplicity. Besides GUIs, several works on video game agents also considered visual-only input and low-level actions. For example, most works on Atari games used the screenshot as visual input and predicted the controller buttons to press (Mnih et al., 2015). More recently, Baker et al. (2022), which focuses on learning from unlabeled videos, proposes an agent for Minecraft that uses pixel-based inputs paired with keyboard and mouse actions, similarly to PIX2ACT.

7 Limitations and Discussion

Pixel-based vs. text-based representations Text-based representations may be practically useful when available, especially since they enable transferring knowledge from LLMs, demonstrating impressive few-shot learning with LLMs for MiniWob++ (Kim et al., 2023) and WebShop (Yao et al., 2023). When structured source is not available, OCR systems and models trained to predict the location and function of UI elements may also help connect models with the power of LLMs. On the other hand, similar advances in scaling and pre-training of vision or multimodal models could

potentially enable similar capabilities in a pixel-based setting in the future, as we have shown the effectiveness of pixel-based pre-training (albeit at a smaller scale) for GUI-based tasks. Nevertheless, beyond addressing the case where HTML or DOM information is unavailable, we hope our study contributes towards a better understanding of the potential of pixel-based representations for instruction following via GUIs.

Tree Search Our approach to policy improvement with tree search for MiniWob++ relied on the ability to procedurally generate new MiniWob++ environment and instruction variations and receive reward signals for task completion. Both aspects are unlikely to be available for some real world environments, and such an approach might need to rely on generative models of potential instructions and approximate reward models for task completion ( e.g . Bahdanau et al. (2018); Du et al. (2023)). Our implementation also relied on the ability to reset the environment to an initial state, a useful feature for environments being used for exploration. Additionally, while we show that tree search can be sufficient to reach high performance on MiniWob++, we did not perform a detailed comparison relative to other search and RL algorithms in this study, which would be useful to better understand the most efficient approaches for learning from GUI-based environments.

Broader Impact In this paper we have trained and evaluated models only in offline environments. Responsibly deploying models in an environment where they can interact with online services would require additional considerations. Prior to enabling a model to access a new service, it would be important to sufficiently verify and/or constrain the behavior of the model to ensure that it is consistent with the terms-of-service for that service and does not otherwise cause harm. Ensuring sufficient data privacy could also be an important consideration for deploying models such as PIX2ACT that rely on capturing screenshots from browsers.

There would be many potential risks associated with deploying models that could interact with services in violation of their terms-of-service or otherwise engage in various forms of spam, fraud, or abuse. Examples of such behavior could include impersonating human users, generating harmful content or spam, or engaging in denial-of-service attacks. Models that use the same conceptual interface humans use could potentially be more capable of breaking security defenses (e.g. solving CAPTCHAs) or engaging in forms of spam, fraud, or abuse that are more difficult to detect. It is therefore important for research related to security and techniques for detecting spam, fraud, and abuse to take such potential uses into account.

Acknowledgments

We would like to thank Peter Humphreys, Toby Pohlen, and Gregory Thornton for their assistance with the MiniWob++ demonstraions. We also thank Ming-Wei Chang, Austin Huang, Luheng He, Tianze Shi, David Gaddy, Jacob Eisenstein, and Yi Luan for useful discussions and comments.

References

James F. Allen, Nathanael Chambers, George Ferguson, Lucian Galescu, Hyuckchul Jung, Mary D. Swift, and William Taysom. Plow: A collaborative task learning agent. In AAAI Conference on Artificial Intelligence , 2007.

Thomas Anthony, Zheng Tian, and David Barber. Thinking fast and slow with deep learning and tree search. Advances in neural information processing systems , 30, 2017.

Dzmitry Bahdanau, Felix Hill, Jan Leike, Edward Hughes, Arian Hosseini, Pushmeet Kohli, and Edward Grefenstette. Learning to understand goal specifications by modelling reward. In International Conference on Learning Representations , 2018.

Bowen Baker, Ilge Akkaya, Peter Zhokov, Joost Huizinga, Jie Tang, Adrien Ecoffet, Brandon Houghton, Raul Sampedro, and Jeff Clune. Video pretraining (VPT): Learning to act by watching unlabeled online videos. Advances in Neural Information Processing Systems , 35:24639-24654, 2022.

R. K. Branavan, Luke Zettlemoyer, and Regina Barzilay. Reading between the lines: Learning to map high-level instructions to commands. In Annual Meeting of the Association for Computational Linguistics , 2010.

Andrea Burns, Deniz Arsan, Sanjna Agrawal, Ranjitha Kumar, Kate Saenko, and Bryan A Plummer. Interactive mobile app navigation with uncertain or under-specified natural language commands. arXiv preprint arXiv:2202.02312 , 2022.

Hyung Won Chung, Le Hou, Shayne Longpre, Barret Zoph, Yi Tay, William Fedus, Eric Li, Xuezhi Wang, Mostafa Dehghani, Siddhartha Brahma, et al. Scaling instruction-finetuned language models. arXiv preprint arXiv:2210.11416 , 2022.

Rémi Coulom. Efficient selectivity and backup operators in monte-carlo tree search. In Computers and Games , 2006.

Alexey Dosovitskiy, Lucas Beyer, Alexander Kolesnikov, Dirk Weissenborn, Xiaohua Zhai, Thomas Unterthiner, Mostafa Dehghani, Matthias Minderer, Georg Heigold, Sylvain Gelly, Jakob Uszkoreit, and Neil Houlsby. An image is worth 16x16 words: Transformers for image recognition at scale. In ICLR , 2021.

Yuqing Du, Ksenia Konyushkova, Misha Denil, Akhil Raju, Jessica Landon, Felix Hill, Nando de Freitas, and Serkan Cabi. Vision-language models as success detectors. arXiv preprint arXiv:2303.07280 , 2023.

Hiroki Furuta, Ofir Nachum, Kuang-Huei Lee, Yutaka Matsuo, Shixiang Shane Gu, and Izzeddin Gur. Instruction-finetuned foundation models for multimodal web navigation. In Workshop on Reincarnating Reinforcement Learning at ICLR 2023 , 2023a.

Hiroki Furuta, Ofir Nachum, Kuang-Huei Lee, Yutaka Matsuo, Shixiang Shane Gu, and Izzeddin Gurt. Instruction-finetuned foundation models for multimodal web navigation. In First Workshop on Multimodal Representation Learning at ICLR , 2023b.

Izzeddin Gur, Ulrich Rueckert, Aleksandra Faust, and Dilek Hakkani-Tur. Learning to navigate the web. arXiv preprint arXiv:1812.09195 , 2018.

Izzeddin Gur, Ofir Nachum, Yingjie Miao, Mustafa Safdari, Austin Huang, Aakanksha Chowdhery, Sharan Narang, Noah Fiedel, and Aleksandra Faust. Understanding HTML with large language models. arXiv preprint 2210.03945 , 2022.

Peter C Humphreys, David Raposo, Tobias Pohlen, Gregory Thornton, Rachita Chhaparia, Alistair Muldal, Josh Abramson, Petko Georgiev, Adam Santoro, and Timothy Lillicrap. A data-driven approach for learning to control computers. In International Conference on Machine Learning , pages 9466-9482. PMLR, 2022.

Sheng Jia, Jamie Ryan Kiros, and Jimmy Ba. Dom-q-net: Grounded rl on structured language. In International Conference on Learning Representations , 2019.

Geunwoo Kim, Pierre Baldi, and Stephen McAleer. Language models can solve computer tasks. arXiv preprint arXiv:2303.17491 , 2023.

Kenton Lee, Mandar Joshi, Iulia Turc, Hexiang Hu, Fangyu Liu, Julian Eisenschlos, Urvashi Khandelwal, Peter Shaw, Ming-Wei Chang, and Kristina Toutanova. Pix2Struct: Screenshot parsing as pretraining for visual language understanding. arXiv preprint arXiv:2210.03347 , 2022.

Gang Li and Yang Li. Spotlight: Mobile ui understanding using vision-language models with a focus. In The Eleventh International Conference on Learning Representations , 2022.

Yang Li, Jiacong He, Xin Zhou, Yuan Zhang, and Jason Baldridge. Mapping natural language instructions to mobile ui action sequences. arXiv preprint arXiv:2005.03776 , 2020a.

Yang Li, Gang Li, Luheng He, Jingjie Zheng, Hong Li, and Zhiwei Guan. Widget captioning: Generating natural language description for mobile user interface elements. In Proceedings of the 2020 Conference on Empirical Methods in Natural Language Processing (EMNLP) , pages 5495-5510, Online, November 2020b. Association for Computational Linguistics. doi: 10.18653/ v1/2020.emnlp-main.443. URL https://aclanthology.org/2020.emnlp-main.443 .

Evan Zheran Liu, Kelvin Guu, Panupong Pasupat, Tianlin Shi, and Percy Liang. Reinforcement learning on web interfaces using workflow-guided exploration. In International Conference on Learning Representations (ICLR) , 2018. URL https://arxiv.org/abs/1802.08802 .

Volodymyr Mnih, Koray Kavukcuoglu, David Silver, Andrei A. Rusu, Joel Veness, Marc G. Bellemare, Alex Graves, Martin A. Riedmiller, Andreas Kirkeby Fidjeland, Georg Ostrovski, Stig Petersen, Charlie Beattie, Amir Sadik, Ioannis Antonoglou, Helen King, Dharshan Kumaran, Daan Wierstra, Shane Legg, and Demis Hassabis. Human-level control through deep reinforcement learning. Nature , 518:529-533, 2015.

Reiichiro Nakano, Jacob Hilton, Suchir Balaji, Jeff Wu, Long Ouyang, Christina Kim, Christopher Hesse, Shantanu Jain, Vineet Kosaraju, William Saunders, et al. Webgpt: Browser-assisted question-answering with human feedback. arXiv preprint arXiv:2112.09332 , 2021.

Colin Raffel, Noam Shazeer, Adam Roberts, Katherine Lee, Sharan Narang, Michael Matena, Yanqi Zhou, Wei Li, and Peter J Liu. Exploring the limits of transfer learning with a unified text-to-text transformer. Journal of Machine Learning Research , 21:1-67, 2020. URL https: //arxiv.org/abs/1910.10683 .

John Schulman, Filip Wolski, Prafulla Dhariwal, Alec Radford, and Oleg Klimov. Proximal policy optimization algorithms. ArXiv , abs/1707.06347, 2017.

Noam Shazeer and Mitchell Stern. Adafactor: Adaptive learning rates with sublinear memory cost. In International Conference on Machine Learning , pages 4596-4604. PMLR, 2018.

Tianlin Shi, Andrej Karpathy, Linxi Fan, Jonathan Hernandez, and Percy Liang. World of bits: An open-domain platform for web-based agents. In Doina Precup and Yee Whye Teh, editors, Proceedings of the 34th International Conference on Machine Learning , volume 70 of Proceedings of Machine Learning Research , pages 3135-3144. PMLR, 06-11 Aug 2017. URL https:// proceedings.mlr.press/v70/shi17a.html .

David Silver, Julian Schrittwieser, Karen Simonyan, Ioannis Antonoglou, Aja Huang, Arthur Guez, Thomas Hubert, Lucas Baker, Matthew Lai, Adrian Bolton, et al. Mastering the game of go without human knowledge. nature , 550(7676):354-359, 2017.

Ashish Vaswani, Noam Shazeer, Niki Parmar, Jakob Uszkoreit, Llion Jones, Aidan N Gomez, Łukasz Kaiser, and Illia Polosukhin. Attention is all you need. Advances in neural information processing systems , 30, 2017.

Bryan Wang, Gang Li, Xin Zhou, Zhourong Chen, Tovi Grossman, and Yang Li. Screen2Words: Automatic mobile UI summarization with multimodal learning. In The 34th Annual ACM Symposium on User Interface Software and Technology , UIST '21, page 498-510, New York, NY, USA, 2021. Association for Computing Machinery. ISBN 9781450386357. doi: 10.1145/3472749.3474765. URL https://doi.org/10.1145/3472749.3474765 .

Ronald J. Williams. Simple statistical gradient-following algorithms for connectionist reinforcement learning. Machine Learning , 8:229-256, 2004.

Yonghui Wu, Mike Schuster, Zhifeng Chen, Quoc V Le, Mohammad Norouzi, Wolfgang Macherey, Maxim Krikun, Yuan Cao, Qin Gao, Klaus Macherey, et al. Google's neural machine translation system: Bridging the gap between human and machine translation. arXiv preprint arXiv:1609.08144 , 2016.

Shunyu Yao, Howard Chen, John Yang, and Karthik R Narasimhan. Webshop: Towards scalable real-world web interaction with grounded language agents. In Alice H. Oh, Alekh Agarwal, Danielle Belgrave, and Kyunghyun Cho, editors, Advances in Neural Information Processing Systems , 2022. URL https://openreview.net/forum?id=R9KnuFlvnU .

Shunyu Yao, Jeffrey Zhao, Dian Yu, Nan Du, Izhak Shafran, Karthik Narasimhan, and Yuan Cao. ReAct: Synergizing reasoning and acting in language models. In International Conference on Learning Representations (ICLR) , 2023.

Luke S Zettlemoyer and Robert St. Amant. A visual medium for programmatic control of interactive applications. In Proceedings of the SIGCHI conference on Human Factors in Computing Systems , pages 199-206, 1999.

A Additional Dataset Details

A.1 MiniWob++ Supported Tasks

MiniWob++ consists of 104 tasks. Most prior work (Shi et al., 2017; Liu et al., 2018; Gur et al., 2018; Jia et al., 2019) has evaluated performance on only a subset of these tasks, with the notable exception of Humphreys et al. (2022), which evaluated on all 104 tasks. We evaluated on 59 of these 104 tasks, based on our best effort attempt to (1) design a general purpose set of actions that could be implemented using Selenium and (2) convert the demonstrations collected by Humphreys et al. (2022) to our observation and action format. While further development of the conversion process and Selenium-based actions could potentially support more tasks, the 59 tasks we support still include a wide range of instructions and interactions. Note that determining the set of 59 tasks was based solely on the feasibility of conversion to our observation and action format, and not based on model performance. Below we offer further details.

Several tasks in MiniWob++ feature animated elements. These tasks can require sampling observations in a real-time manner in order to capture the information needed to select the correct action. Also, the effects of an action may be delayed and therefore not captured by an observation sampled immediately after the action has executed. MiniWob++ provides a -nodelay version for several tasks which removes such animations. We train and evaluate on the -nodelay version of these tasks ( choose-date , click-collapsible-2 , click-collapsible , click-pie , use-autocomplete ). We exclude choose-date-easy and choose-date-medium which offer simpler versions of choose-date but do not have a corresponding -nodelay version. Additionally, we exclude chase-circle , drag-cube , moving-items , and simon-says , which feature animation without a -nodelay version.

Many MiniWob++ tasks also involve vertical scrolling. In the human demonstrations, this can be implemented using a scroll wheel, or various clicking or dragging interactions with a vertical scroll bar rendered on the right side of a scrollable element. Mapping such interactions to actions that lead to equivalent scrolling in our Seleniumbased environment is non-trivial. Therefore, for simplicity, we excluded tasks that involve scrolling: book-flight , click-scroll-list , email-inbox , email-inbox-nl-turk , read-table , read-table-2 , scroll-text , scroll-text-2 , search-engine , social-media , social-media-all , social-media-some , terminal .

Demonstrations for many MiniWob++ tasks also include copying and pasting text. In many cases, this was executed in the human demonstrations by double clicking a text string and then clicking and dragging it into an input field. Such an interaction is not supported in Selenium, which made it challenging to support these tasks. This led us to exclude the following tasks: login-user-popup , copy-paste , copy-paste-2 , email-inbox-forward , email-inbox-forward-nl , email-inbox-forward-nl-turk , email-inbox-noscroll , email-inbox-reply , email-inbox-star-reply , enter-password , enter-text , enter-text-dynamic , find-word , login-user , multi-layouts , multi-orderings .

Finally, we excluded several other tasks for various other reasons. The choose-list task uses the HTML <select> tag to implement a drop-down menu, which is not supported properly by our Selenium-based environment. The click-menu and click-menu-2 tasks require unsupported mouseover effects. Demonstrations for the text-editor task features click and drag interactions to highlight text which do not have the same effect when executed in Selenium. There also appeared to be differences in how Selenium implemented the number input field for guess-number . Finally, we excluded several tasks due to low demonstration conversion success rates ( focus-text , focus-text-2 , use-spinner ). Upon further investigation, this was due to episodes completing immediately after a 'pointer down' event without a complete click for focus-text and focus-text-2 , and due to frequent double clicking for use-spinner .

A.2 MiniWob++ Rendering Differences

There are differences between the rendering of observations in the human demonstrations from Humphreys et al. (2022) and the rendering of environment state in our Selenium-based environment. We show an example in Figure 5, which shows subtle differences, e.g . in font style and in element sizes and positions.

Select yy.Mf6RM0, DJVKFV and

clickSubmit.

DJVKFV

Mf6RMO

人

Submit

Selectyy,Mf6RMO,DJVKFVand

clickSubmit.

DJVKFV

Mf6RMO

yy

Submit

B Additional Technical Details

B.1 Beam Search

As mentioned in §3, we use beam search over tokens in the text decoder to produce a set of topk actions for a given state, along with their approximate probabilities. We refer to these as approximate probabilities because they are subject to a length normalization factor (Wu et al., 2016) of 0 . 6 during beam search, following Raffel et al. (2020). For MiniWob and WebShop, our experiments used k = 8 and k = 10 , respectively.

B.2 Tree Search

Here we describe the details of the tree search approach described in §3.1. We adopt Monte Carlo Tree Search (MCTS) (Coulom, 2006), and follow prior work which has integrated MCTS with neural networks (Silver et al., 2017; Anthony et al., 2017), which we apply to MiniWob++ environments. We performed a minimal amount of tuning to determine an approach that yielded improvements in mean score over the greedy policy, even for the most challenging tasks.

Problem Setting We consider an environment with states S and actions A . The reward function, r ( s ) , returns a scalar corresponding to the reward given for transitioning to state s ∈ S , and is described below. MiniWob++ environments are randomly generated, but transitions are deterministic within an environment generated by a particular random seed. The transition function, f ( s, a ) , returns the state resulting from taking action a ∈ A in state s ∈ S .

Surrogate reward Rather than using the raw reward directly provided by the MiniWob++ environment, we consider a surrogate reward: r ( s ) = α s + r t ( s ) , where α s provides a small negative reward that encourages shorter trajectories without unnecessary actions. r t ( s ) is the raw reward from the MiniWob++ environment if s is a terminal state and the raw reward is > 0 . 8 , or 0 otherwise. We use α S = -1 30 . As all tasks can be completed within 30 steps, this is small enough to ensure a positive reward is possible for all tasks. Additionally, the penalty is small enough such that in practice the agent should not be incentivized to sacrifice raw reward to reduce the number of steps taken.

Value network The value function v π ( s ) for a given policy π is the expected future rewards from state s if actions are selected according to policy π . The optimal value function, v ∗ ( s ) , is the expected future rewards if optimal actions are chosen. We attempt to learn an approximation of this function, ˆ v φ ( s ) ≈ v ∗ ( s ) , parameterized as a PIX2STRUCT-initialized model with parameters φ , which we refer to as the value network . The model is trained on transitions from the human demonstrations, which demonstrate close to optimal behavior in many cases. For every state in the human demonstrations, we compute the actual future rewards for the given episode, according to the surrogate reward. We map these future rewards to discrete bins and represent them as integers in the PIX2STRUCT decoder. At inference time, we approximate the mean of the distribution over these discrete bins by considering the topn predictions from the model using beam search (with n = 3 ), weighted proportional to their respective probabilities.

Policy network For consistency with prior work, we will refer to the PIX2STRUCT model tuned to generate actions ( i.e . PIX2ACT) as the policy network , with parameters θ . The greedy policy π θ ( s ) selects the action a with the highest approximate probability p θ ( a | s ) in the topk beam (see §B.1), subject to the conditions described in § 3 .

Search policy We can use lookahead search to implement a policy, π ∗ θ ( s ) , which leverages interactions with the environment ( f ( s, a ) and r ( s ) ) to select actions in a more optimal way than the greedy policy π θ ( s ) . Both the policy network and value network are used to constrain and prioritize the search.

MCTS performs K rounds of traversing a search tree with nodes corresponding to states, and edges corresponding to actions. Due to the computational cost of the policy and value networks, we use a modest number of rounds, K = 16 , for our experiments. The search tree is initialized with a single root node for state s . Each round starts at s and traverses the tree. At each step t of a given round, an action a t is selected for state s t , where a t = max a Q ( s t , a ) + U ( s t , a ) . Q ( s t , a ) is an average reward over all rounds that have traversed the associated edge. It is based on actual accumulated rewards during tree traversal and the value estimates of leaf states (described below). √

U ( s t , a ) = c ∗ p θ ( a | s ) ∗ N ( s t ) 1+ n ( s t ,a ) is a term that encourages exploration, where n ( s t , a ) is the number of times action a has been selected from state s t , N ( s t ) is the total number of times state s t has been visited, and c is a scalar hyperparameter that we set to 0 . 1 . Following Silver et al. (2017), we use the policy network to bias this exploration term. To constrain the search, we only consider the topk actions according to the policy network, where k = 8 in our experiments.

If we select an action a t for state s t which has never been previously selected from s t , then the simulation ends and we add a new leaf state, s L = f ( s t , a ) , to the search tree. If s L is not a terminal state, then we estimate its value ( i.e . future returns) using both the value network and a rollout with the greedy policy. Specifically, following Silver et al. (2017), we estimate its value as λ ∗ ˆ v φ ( s L ) + (1 -λ ) ∗ v π θ ( s L ) where v π θ ( s L ) is equal to the actual returns from following the policy π θ starting at s L for a maximum of 20 steps, with actual returns clipped to a minimum value of 0 . Is there λ is a mixing parameter that we set to 0 . 1 . For challenging environments, rollouts may be unlikely to find a terminal state with positive reward, and in such cases rollouts may not be very informative. On the other hand, the value network can provide poor value estimates for certain states, especially if they are not well represented in the human demonstrations. By combining both methods we aim to provide a better approximation of the value of leaf states. Returns are propagated up the tree to each parent s ′ to update Q ( s ′ , a ) . As Q ( s L , a ) is undefined prior to selecting a from s L for the first time, we initialize Q ( s L , a ) for each action to be equal to the initial value estimate of s L plus α s .

To understand the impact of rollouts and value estimates using the value network, in Table 3 we compare mean scores over 12 challenging MiniWob++ tasks for different values of λ : 0 (rollout only), 0.1 (both rollout and value network), and 1 (value network only). We also include the mean score using the greedy policy for reference. These results use the policy network and value network trained on the human demonstrations. The results show that using a combination of rollouts and the value network gives the best results. The value network is primarily useful for challenging tasks that require longer trajectories, such as number-checkboxes , relative to using rollouts only.

Once we have completed K rounds, π ∗ θ ( s ) selects the most visited action a for state s , and we begin the process again at the subsequent state. We reuse the search tree for subsequent time steps for efficiency, so we require only K -n ( s, a ) additional rounds for the subsequent state.

Policy improvement We can sample trajectories with π ∗ θ , then update θ by training π θ ( s ) to approximate π ∗ θ ( s ) for each s in the sampled trajectories. This then also improves π ∗ θ ( s ) , as θ informs how the search space is constrained and prioritized. Therefore, we can continue to iteratively improve π θ ( s ) . To produce these trajectories, we randomly sample MiniWob++ tasks and seeds, and select

actions according to π ∗ θ . We then filter trajectories where the raw reward is < 0 . 8 . We then tune θ on these new trajectories. For simplicity, we keep the value network ( i.e . φ ) fixed.

We initially found that tuning on trajectories from MCTS could be unstable, leading to an early loss spike. To resolve this, we slightly decreased the learning rate (from 1 e -3 to 5 e -4 ) and increased the number of warmup steps (from 1000 to 4000 ) relative to the hyperparameters used for behavioral cloning.

B.3 Compute Details

We fine-tuned models using 64 Google Cloud TPU v3 cores.

C Additional Results

C.1 Variance Estimates

We evaluated results for MiniWob++ based on 100 randomly selected seeds for each of the 59 tasks. To understand how results vary depending on which 100 seeds per task are used for evaluation, we ran 3 trials with different evaluation seeds for the strongest PIX2ACT model reported in Table 3, yielding mean scores of 96 . 2 , 96 . 4 , and 96 . 1 ; the standard deviation across these trials was 0 . 15 . For WebShop, there is a standard test set consisting of 500 instances, so selecting seeds for evaluation is not necessary.

C.2 MiniWob++ Results Per Task

We show the performance of PIX2ACT (ours) on each of the 59 MiniWob++ tasks we study, compared to other approaches, in Table 4. We compare with human crowdworker performance reported by Humphreys et al. (2022), CC-Net (Humphreys et al., 2022), DOM-Q-Net (Jia et al., 2019), DOMNET with workflow-guided execution (Liu et al., 2018), QWeb (Gur et al., 2018), RCI (Kim et al., 2023), WebN-T5-3B (Gur et al., 2022), and WebGUM (Furuta et al., 2023a). We also report scores for PIX2ACT and CC-Net with behavioral cloning (BC) only. We do not include scores for GlobalCNN (Shi et al., 2017), which reported only human normalized success rates. Other than Humphreys et al. (2022), prior work has primarily reported success rate ( i.e . the percentage of episodes with positive rewards), which can be equivalently mapped to the scores we report for tasks without partial rewards.

Peer reviews:
{"IS_META_REVIEW": false, "comments": "SUMMARY:\nThe paper describes a system to perform pixel-based interaction with human-taylored screens based on GUIs. The proposed system relies on a transformation to a text-based, structured representation of the screen and the screen tasks performed using a larger, pre-trained model of almost 300M parameters. The proposed system achieves performance similar to human beings performing those tasks and SOTA using text representations of the screen.\n\nSTRENGTHS:\nThe main strengths of the paper are:\n\n1. It demonstrates a successful system able to interact with the screen (web-based applications)  in a similar way that humans do.\n2. It does so by fine-tuning a pre-trained system with a small number of demonstrations. \n3. It performs an ablation test where it shows that the pre-training information is critical.\n\nWEAKNESSES:\nThe main weaknesses of the paper are:\n\n1. The evaluation procedure is so succinctly described that it is hard to understand what was exactly done.\n2. The granularity of number of tests according to test type seems to be too coarse (500 tests for 59 task is less than 10 per task).\n3. Results are report with little extra information such as mean scores.\n4. No statistical significance analysis is performed justifying the claims that the scores are better than SOTA or human (therefore here I assume they are similar, not better).\n5. The discussion section is very limited and not clear about the actual contributions.\n6. The authors fail to discuss important ethical issues concerning their research. In particular, if AI systems can interact using GUI interfaces, they can produce content and affect web-based systems in very impactful ways, including altering political and social discourses, create hard-to-identify denial attacks, fool \"I am not a robot\" tests, and pose as people while interacting with real people.\n\nQUESTIONS:\n1. Are the results of the proposed system statistically better than the other systems? With what level of confidence?\n2. Can you provide the standard deviations of the tests performed? Are they adequate?\n\nLIMITATIONS:\nThe limitation section is very brief and not adequate for a paper with the ethical impacts that such technologies can create.", "TITLE": "Official Review", "RECOMMENDATION": 6, "REVIEWER_CONFIDENCE": 3, "SOUNDNESS_CORRECTNESS": 3, "DATE": "2023-06-30"}

{"IS_META_REVIEW": false, "comments": "SUMMARY:\nThis paper investigates to create agents that interact with the UIs based on pixel-based screenshots and a generic action space corresponding to keyboard and mouse actions. Extensive experiments on simulated environments, i.e., MiniWob++, WebShop, evaluate the effectiveness of proposed agents, show the benefits of pretraining, and demonstrate the successful application of tree search.\n\nSTRENGTHS:\nThis paper builds a pixel-based agent for instruction-following interaction with UIs. This is an interesting and valuable research topic of high bussiness impacts as well for building AI-based UI interaction assistants. The proposed methods are reasonable, and the conducted experiments are clearly presented. The corresponding analysis and discussion for limitations are comprehensive and sufficient.\n\nWEAKNESSES:\nThe biggest problem of this work is about the novelty of its proposed method. It's kind of difficult for readers to catch up with the core differences compared with the existing pixel-based agents, e.g., Pix2Act, and understraning its advantages. The authors should provide mode insight by highlighting the novelty of proposed methods and clarifying the rationale behind its advantages. \n\nAs for the application of tree search, it is widely seen in RL-based agent building. The corresponding introduction in this works, including its motivation, formulation and benefits, lacks of a detailed and clear statement.\n\nQUESTIONS:\n1. What are the key differences between this work and the existing pixel-only UI agents and tree search based RL agents in terms of the methodology?\n2. What are the motivations, detailed formulation and beneits of applying tree search here? Are there anything special for completing UI tasks?\n\nLIMITATIONS:\nPls kindly see the contents of weakness and questions as above.", "TITLE": "Official Review", "RECOMMENDATION": 4, "REVIEWER_CONFIDENCE": 4, "SOUNDNESS_CORRECTNESS": 3, "DATE": "2023-07-03"}

{"IS_META_REVIEW": false, "comments": "SUMMARY:\nOne of the main goals of intelligent agents is to interact with the internet in the same way that humans do. Prior state-of-the-art models relied on both the DOM structure and the graphical user interface (GUI) to achieve good performance on web browsing tasks that involve following instructions. In this paper, the authors propose a model that relies solely on pixel-based screenshots as input. The model then selects actions that correspond to basic mouse and keyboard operations. \n\nThe authors use a pre-trained model called Pix2Struct and a standard behavior cloning and reinforcement learning (RL) framework to improve the model's ability to follow instructions and perform various web tasks. The authors demonstrate that their pixel-based approach, which does not require access to DOM-related information, can achieve competitive performance on the MiniWob++ benchmark. Additionally, they show through ablation studies that the performance gains can be primarily attributed to the pre-training of Pix2Struct.\n\n**Update after rebuttal**\nI confirm that I have read the rebuttal and other reviewers feedback. Authors have addressed my concerns.\n\nSTRENGTHS:\n- Teaching agents to interact with the web is an important research direction. The experiments presented in this paper provide valuable insights to the community. In particular, the authors demonstrate that large-scale pre-training on image-to-text tasks, such as Pix2Struct, can reduce the need to explicitly provide DOM information to the model. Additionally, the task transfer ablation study shows that the model is able to generalize to other tasks. These findings will help to shape the future research direction on this topic.\n\n- Paper is well written and is easy to follow. For MiniWob++ benchmark, authors performed sufficient ablation studies to quantify gains due to individual components i.e. pre-training, behavior cloning.\n\nWEAKNESSES:\n- It's not clear if authors have tuned all baselines correctly. In particular, authors show that their proposed RL data bootstrapping method leads to better performance than Behavior Cloning. If I understood the experiments correctly, they have not trained the model for the same number of steps so it's very likely that behavior cloning baseline is potentially under-trained. \n- There are multiple important design discussions that are added to the text without any particular justification. For example, authors fine-tune model on MiniWoB++ data for WebShop experiment but don't use WebShop data for MiniWob++ experiment. There is no justification for such design choices. Further, authors provide different learning rate and number of steps for various experiments without justifying these design choices.   \n- Authors present very sparse results and discussion on WebShop dataset and all experiments are mainly conducted on MiniWoB++ dataset. As a reader, it feels like that authors added that dataset mainly for the sake of adding it.\n\nQUESTIONS:\n- Hyperparameter on Webshop data: learning rate and optimizer details. If these are similar to MiniWoB++, please add a note \n- Choose-date: what kind of mistakes model make ? How does RL sampled data help with poor performing tasks? \n- Fig 3: for how many steps did you train Ours (BC only) vs Ours ? In particular, does BC training for additional steps help in improving BC baseline? \n- Authors suggest that intermediate fine-tuning on MiniWoB++ helps for WebShop tasks. Do they observe similar gains while using WebShop data for MiniWoB++ task?\n\nLIMITATIONS:\nAuthors have addressed major limitations of their work and have also provided potential misuse of their research.", "TITLE": "Official Review", "RECOMMENDATION": 6, "REVIEWER_CONFIDENCE": 4, "SOUNDNESS_CORRECTNESS": 3, "DATE": "2023-07-04"}

{"IS_META_REVIEW": false, "comments": "SUMMARY:\nThis paper presents PIX2ACT, a method that interacts with GUIs using pixel-level visual representations and generic low-level actions, emulating how humans interact with these interfaces. Unlike previous approaches, it doesn't rely on structured text-based data sources, but rather processes pixel-based screenshots, circumventing issues associated with obfuscation or misalignment in structured application data. The model showed improved performance compared to human crowdworkers on the MiniWob++ benchmark for GUI-based instruction following tasks.\n\nSTRENGTHS:\nThe paper introduces a novel approach to the challenge of automated GUI interactions, using pixel-based screenshots as opposed to relying on text-based representations.\nThe PIX2ACT model was tested on two benchmark datasets, MiniWob++ and WebShop, which ensured a robust and varied testing process.\nThe paper is clearly written and offers substantial and important contributions, namely the ability to build an agent that can outperform humans in task completion using pixel-based inputs and a generic action space.\nTheir findings indicate that the pre-training of PIX2STRUCT via screenshot parsing is effective for GUI-based instruction following with pixel-based inputs.\n\nWEAKNESSES:\nThe performance on the WebShop benchmark is still significantly below larger language models using HTML-based inputs and task-specific actions.\nSince the method uses tree search, it seems to rely on offline environments. It’s not clear if this approach will be useful in real-world online environments, although I could see this working with perhaps re-settable virtual environments.\n\nQUESTIONS:\nWould the model maintain high performance in real-time environments, given the training was conducted in offline environments?\nHow much of the performance difference is due to the size of the model? Can this method be easily adapted to take advantage of existing larger pretrained models?\n\nLIMITATIONS:\nThe tree search approach relies on the ability to generate new environment and instruction variations and receive reward signals, which might not be feasible in some real-world applications.\nThe paper does not provide a clear solution for the performance gap on complex tasks and environments, like the WebShop benchmark.", "TITLE": "Official Review", "RECOMMENDATION": 7, "REVIEWER_CONFIDENCE": 4, "SOUNDNESS_CORRECTNESS": 3, "DATE": "2023-07-21"}

{"IS_META_REVIEW": false, "comments": "SUMMARY:\nThe paper introduces PIX2ACT, a model designed to interact with GUIs using only pixel-level visual representations. Unlike most prior works that depend on structured interfaces (like HTML or DOM trees), PIX2ACT relies solely on what it visually sees. This approach is motivated by the way humans interact with interfaces without necessarily knowing the underlying code , which is interesting. The study reveals that such a model can efficiently operate in GUI tasks with generic mouse and keyboard actions, and even outperforms humans in specific benchmarks. For this they use PIX2STRUCT pre-trained on mapping screenshots to structured representation.\n\nMajor Contributions:\n\n1. Demonstrated that an agent with pixel-only inputs and generic actions can outperform humans on the MiniWob++ benchmark, achieving performance similar to top-tier agents with access to DOM information.\n\n2. Adapted the WebShop benchmark to function with pixel-based observations and generic actions, establishing the initial baseline performance in this setting.\n\n3. Highlighted the efficiency of PIX2STRUCT’s pre-training through screenshot parsing for GUI-based instruction with pixel-only inputs, leading to notable performance improvements on benchmarks like MiniWob++ and WebShop.\n\n4. Successfully applied tree search for policy improvement in the MiniWob++ environment.\n\nSTRENGTHS:\nThe paper studies a new approach to controlling GUIs using just pixel level information unlike alternate approach which uses DOM tree or HTML. While both has its pros and cons , this is one of the new papers ton explore the pixel based controls in this setting.\n\nAdaptable Benchmarking Capabilities: Authors successfully adapt and operate on the WebShop benchmark using solely pixel-based observations and broad actions, laying down a foundational performance standard.\n\nRobust Pre-training Mechanism: The strength of PIX2STRUCT’s screenshot parsing pre-training is evident. This pre-training strategy remarkably boosts the model's efficiency in GUI-based instruction tasks using pixel-based inputs, as observed in dramatic performance jumps on benchmarks. The authors have performed and demonstrated the importance of this through the ablation studies. This is very interesting.\n\nEffective Policy Enhancement with Tree Search: The model employs tree search effectively, which proves to be a straightforward yet potent method for enhancing its policy in environments like MiniWob++.\n\nA very good point that the paper has made - \" Finally, aligning human demonstrations with task-dependent actions is often challenging.\"\n\nWEAKNESSES:\nThe paper talks about generalization but slightly fails to highlight the same- for example : The chosen datasets ( miniwob++ and webshop(which is a great start) ) are relatively simple datasets and does not capture not capture the full complexity of real-world GUIs. \nAddtionally , to perform well reasonable well on the webshop dataset the model seems to require a fine tuning on miniwob++ which is still unclear as to why it is needed. Thus does it actually generalize to webshop? Furthermore, the authors perform hold out set evaluation on miniwob++. However, the tasks are \"click-checkboxes-large, click-color, click-tab-2, click-tab-2-hard, count-shape, drag-shapes, use-color-wheel-2, use-slider-2\" which could be similar to tasks in the miniwob++ benchmark train set , for instance \"click-tab-2-hard\" etc. It would be great to test on more diverse and complicated datasets.\n\nOver-reliance on Pre-training: While the pre-training from PIX2STRUCT seems beneficial, depending heavily on pre-trained models can sometimes limit the adaptability and flexibility of the system.\n\nHuman Demonstrations: The model seems to be heavily reliant on human demonstrations for training. Gathering these demonstrations can be time-consuming and might not be feasible for all applications. Plus, the quality of these demonstrations can significantly affect the model's performance.\n\nScalability: The approach described in the paper works well for small, simple web applications. It is not clear how the approach would scale to larger, more complex web applications.\n\nThe limitation I see in the environment setup is the use of discrete bins for mouse coordinates and scroll amounts rather than allowing continuous values. A few potential issues with the discrete coordinate binning:\nIt may make precise clicking and dragging actions more difficult if the bin sizes are too coarse. The agent would need to learn to chain multiple discrete drag actions to move long distances. The optimal binning resolution may vary across different interfaces and tasks. Finding the right granularity could require environment-specific tuning. Discretization can potentially lead to suboptimal policies compared to allowing continuous coordinates.\n\nSimilarly for scroll amounts: Scrolling by discrete bins could be inefficient for long pages compared to direct scrolling by pixel amounts. The scroll bin size would again need tuning based on typical page lengths. Overall, the discrete coordinate and scrolling simplification may be necessary to limit the action space size. But it could negatively impact performance on some tasks compared to an environment with continuous values.\n\nNo fine-tuning of visual features. The Vision Transformer weights are frozen, so the model may not learn visual representations best suited for this task.\n\nMinor comments: Despite the visual nature of GUIs, prior work has primarily focused on utilizing structured representations of the user interfaces (such as HTML sources, Document Object Model (DOM) trees, and Android view hierarchies) as well as custom, task-specific representations of high-level actions based on these structured representations. ---- > Missing citations.\n\nQUESTIONS:\nLatency and Computation: Transforming screenshots into structured representations can be computationally intensive, which may lead to latency in real-time applications. So what is the typical time taken? Wanted to hear authors thoughts since , the system is based on interpreting pixel-level information from screenshots. This can be computationally intensive?? and may struggle with dynamic content that changes frequently, or with GUI elements that look visually similar but have different functionalities.\n\nHuman Demonstrations: The model seems to be heavily reliant on human demonstrations for training. Gathering these demonstrations can be time-consuming and might not be feasible for all applications. Plus, the quality of these demonstrations can significantly affect the model's performance. Does authors have any thoughts one this?\n\nScalability: How does PIX2ACT scale with different screen resolutions, GUI complexities, or dynamic content on the screens? The performance of models designed for specific benchmarks might not generalize well to real-world applications. Since even the original PIX2STRUCT paper talks about the impact of resolution.\n\nis there a scope for more sophisticated reward function, such as one that provides the agent with feedback about the progress of the task? also is the cumulative reward that accumulates in the miniwob++ benchmark based on the time taken to solve the task in one episode considered? since the agent receives a reward only at a terminal state. Sparse rewards can make learning more challenging, especially if the agent needs to take a long sequence of actions before receiving feedback?\n\nCursor Representation: The system manually draws a cursor on the screenshot to indicate the mouse pointer position. This might not capture the full nuance of the cursor's state or type (e.g., a hand cursor vs. an arrow cursor), which could provide additional context to the agent.\n\nGreedy Action Selection: The agent follows a greedy policy by selecting the highest scoring action. Does this kind of approach can be short-sighted and might not always result in the optimal long-term strategy??\n\nBeam Search Limitations: While beam search helps in narrowing down the set of most probable sequences, does it always yield the optimal sequence in practice. in the context longer sequences, where the algorithm might not explore sufficiently outside of its \"beam\" to find a better solution. It can sometimes prefer shorter sequences over more accurate longer ones??\n\nFuture work: An additional pre-training task that could potentially improve the model is predicting affordances of UI elements from screenshots. Affordances refer to the possible interactions that an element supports, like if a button can be clicked, a text box can be typed in, a menu can be opened, etc.\n\nLIMITATIONS:\nThe authors have mentioned the limitations ( It was thoughtful to anticipate about CAPTCHA) .\nHowever:\nData Privacy: Since they are using screenshots to interpret and interact with GUIs, there might be concerns about data privacy, especially if sensitive information is displayed on the screen. This should be highlighted.\nI understand this would be issue while using DOM/HTML as well but highlighted might lead to responsible adoption.", "TITLE": "Official Review", "RECOMMENDATION": 6, "REVIEWER_CONFIDENCE": 4, "SOUNDNESS_CORRECTNESS": 3, "DATE": "2023-07-27"}

{"IS_META_REVIEW": false, "comments": "SUMMARY:\nThis work explores the possibility of building an agent that can complete tasks for users solely based on pixel-level visual representations of the GUI state and generic low-level actions, without relying on structured or task-specific representations. The authors demonstrate the effectiveness of their approach on two benchmarks, MiniWob++ and WebShop, adapted to their general Chrome-based environment framework.\n\nSTRENGTHS:\nThis submission demonstrates significant strength in several key areas. Firstly, it presents a clear and compelling motivation for developing an agent capable of directly interacting with pixel-level GUI interfaces, aligning with a more natural and human-like manner. This aspect alone distinguishes the submission and makes it particularly appealing to the readers.\n\nMoreover, the use of well-crafted illustrations enhances the understanding of the content and effectively communicates the main ideas, further solidifying its strength. Readers can easily grasp the concepts presented, adding to the submission's overall impact.\n\nFurthermore, the proposed framework is rigorously validated on two datasets, and the comparison with baselines using DOM/HTML as input is conducted in great detail. This thorough analysis strengthens the submission's credibility and highlights its robustness.\n\nWEAKNESSES:\nI have several concerns regarding the experimental design in the submission:\n\n(1) One notable issue is the absence of certain ablations that would provide valuable insights into the effectiveness of each component in the proposed agent framework. Specifically, the authors should address the bottleneck of the framework, whether it lies in the visual encoder or the text decoder. Additionally, it would be beneficial to explore the potential performance improvements by utilizing alternative variants of visual encoders or text decoders. Furthermore, since the visual encoder plays a crucial role in extracting and understanding instructions embedded in the interface, it is essential to verify the effectiveness of using ViT in OCR accuracy.\n\n(2) The rationale behind embedding instructions into the UI screenshot instead of providing them directly as input remains unclear. The submission should elaborate on this choice and justify its benefits. Moreover, considering the availability of open-source Visual Language Models (VLMs) like mPLUG-Owl [1], OpenFlamingo [2], and Otter [3], which natively support instruction following and accept multimodal input, it would be valuable to include these VLMs as baselines in the revised version. This addition would offer a more comprehensive evaluation and better contextualize the proposed method's unique contribution in comparison to existing approaches.\n\n[1] https://github.com/X-PLUG/mPLUG-Owl\n[2] https://github.com/mlfoundations/open_flamingo\n[3] https://github.com/Luodian/Otter\n\nAddressing these concerns would significantly enhance the experimental design and strengthen the submission's overall validity and contribution to the field.\n\nQUESTIONS:\n(1) In Figure 2, step 3, why are there two overlapping ‘+’? Does it mean the mouse is pressed down?\n\n(2) This is not a serious issue, but some Figures are quite blurry (e.g., MiniWob++ examples in Figure1 and top row in Figure 2), is this due to the way the dataset is constructed?\n\nLIMITATIONS:\nThe authors have included discussions of the limitations.", "TITLE": "Official Review", "RECOMMENDATION": 5, "REVIEWER_CONFIDENCE": 2, "SOUNDNESS_CORRECTNESS": 3, "DATE": "2023-08-01"}