Using Mechanical Turk¶
Authors: Will Feng, Jack Urbanek, Emily Dinan
In ParlAI, you can use Amazon Mechanical Turk for data collection, training, or evaluation of your dialog model.
Human Turkers are viewed as just another type of agent in ParlAI; hence, agents in a group chat consisting of any number of humans and/or bots can communicate with each other within the same framework.
The human Turkers communicate in observation/action dict format, the same as all other agents in ParlAI. During the conversation, human Turkers receive a message that is rendered on the live chat webpage, such as the following:
Each MTurk task has at least one human Turker that connects to ParlAI via the Mechanical Turk Live Chat interface, encapsulated as an
Each MTurk task also consists of a
World where all agents live and interact within.
We provide a few examples of using Mechanical Turk with ParlAI:
- QA Data Collection: collect questions and answers from Turkers, given a random Wikipedia paragraph from SQuAD.
- Model Evaluator: ask Turkers to evaluate the information retrieval baseline model on the Reddit movie dialog dataset.
- Multi-Agent Dialog: round-robin chat between a local human agent and two Turkers.
- Deal or No Deal: negotiation chat between two agents over how to fairly divide a fixed set of items when each agent values the items differently.
- Qualification Flow Example: filter out workers from working on more instances of your task if they fail to complete a test instance properly.
Task 1: Collecting Data¶
One of the biggest use cases of Mechanical Turk is to collect natural language data from human Turkers.
As an example, the QA Data Collection task does the following:
- Pick a random Wikipedia paragraph from SQuAD dataset.
- Ask a Turker to provide a question given the paragraph.
- Ask the same Turker to provide an answer to their question.
QADataCollectionWorld, there are two agents: one is the human Turker (
MTurkAgent), the other is the task agent (
DefaultTeacher from SQuAD) that provides the Wikipedia paragraph.
turn_index to denote what stage the conversation is at. One turn means that
world.parley() has been called once.
After two turns, the task is finished, and the Turker’s work is submitted for review.
Task 2: Evaluating a Dialog Model¶
You can easily evaluate your dialog model’s performance with human Turkers using ParlAI. As an example, the Model Evaluator task does the following:
- Initialize a task world with a dialog model agent (ir_baseline) and a dataset (MovieDD-Reddit).
- Let all the agents in the task world
act()once, by calling
parley()on the world.
- Ask the human Turker to rate the dialog model agent’s response on a scale of 0-10.
ModelEvaluatorWorld, there are two main components: one is the
task_world that contains the task and the dialog model we are evaluating, the other is the
MTurkAgent which is an interface to the human Turker.
Note that since the human Turker speaks only once to provide the rating, the
ModelEvaluatorWorld doesn’t need to use
turn_index to keep track of the turns.
After one turn, the task is finished, and the Turker’s work is submitted for review.
Task 3: Multi-Agent Dialog¶
ParlAI supports dialogs between multiple agents, whether they are local ParlAI agents or human Turkers. In the Multi-Agent Dialog task, one local human agents and two Turkers engage in a round-robin chat, until the first local human agent sends a message ending with
[DONE], after which other agents will send a final message and the task is concluded.
This task uses the
MultiAgentDialogWorld which is already implemented in
Task 4: Advanced Functionality - Deal or No Deal¶
ParlAI is able to support more than just generic chat. The Deal or No Deal task provides additional functionality over the regular chat window to allow users to view the items they are dividing, select an allocation, and then submit a deal.
This task leverages the ability to override base functionality of the core.html page using
In general it is easier/preferred to use a custom webpage as described in step 4 of “Creating Your Own Task”, though this is an alternate that can be used if you specifically only want to show additional components in the task description pane of the chat window.
Task 5: Advanced Functionality - MTurk Qualification Flow¶
ParlAI MTurk is able to support filtering users through a form of qualification system. The Qualification Flow task demos this functionality using a simple “addition” task.
In this task, all users see a test version of the task on the first time they enter it and a real version every subsequent time, however users that fail to pass the test version are assigned a qualification that prevents them from working on the task again. Thus ParlAI users are able to filter out workers from the very beginning who don’t necessarily meet the specifications you are going for. This is preferred to filtering out workers using the onboarding world for tasks that require a full instance’s worth of work to verify a worker’s readiness.
Creating Your Own Task¶
ParlAI provides a generic MTurk dialog interface that one can use to implement any kind of dialog tasks. To create your own task, start with reading the tutorials on the provided examples, and then copy and modify the example
task_config.py files to create your task.
A few things to keep in mind:
- To end a conversation, you should check to see if an action has
True, as this signals that the world should start returning
run.py, You can use
assignment_indexto differentiate between different HITs and assignments, and change the content of the task accordingly.
- Make sure to test your dialog task using MTurk’s sandbox mode before pushing it live, by using the
--sandboxflag (enabled by default) when running
- [Optional] If you want to show a custom webpage (instead of the default one) for any of your MTurk agents, you can create an
htmlfolder within your task directory, and then create the
<mturk_agent_id>_index.htmlfiles within the
htmldirectory. In those files, you can extend from
core.htmland override any code blocks that you want to change. (Please look at parlai/mturk/core/html/mturk_index.html as an example.) These agent-specific templates will automatically be shown to the Turkers in the next run.
Running a Task¶
If you have not used Mechanical Turk before, you will need an MTurk Requester Account and an AWS account (these are two separate accounts). Follow the steps below:
- Sign up for an AWS account at aws.amazon.com
- Sign up for an MTurk account at requester.mturk.com
- Go to the developer tab (https://requester.mturk.com/developer) and link your AWS account to your MTurk account (Step 2 on that screen)
- MTurk also has a “Sandbox” which is a test version of the MTurk marketplace. You can use it to test publishing and completing tasks without paying any money. ParlAI supports the Sandbox. To use the Sandbox, you need to sign up for a Sandbox account. You will then also need to link your AWS account to your Sandbox account. In order to test faster, you will also want to create a Sandbox Worker account. You can then view tasks your publish from ParlAI and complete them yourself.
- ParlAI’s MTurk default functionality requires a free heroku account which can be obtained here. Running any ParlAI MTurk operation will walk you through linking the two. If, instead, you wish to run ParlAI MTurk’s node server on the same machine you are running ParlAI from, use the flag
--local. Note that if you specify this flag, you will need to set up SSL for your server.
Then, to run an MTurk task, first ensure that the task directory is in parlai/mturk/tasks/. Then, run its
run.py file with proper flags:
python run.py -nc <num_conversations> -r <reward> [--sandbox]/[--live]
E.g. to create 2 conversations for the QA Data Collection example with a reward of $0.05 per assignment in sandbox mode, first go into the task directory and then run:
python run.py -nc 2 -r 0.05 --sandbox
Please make sure to test your task in MTurk sandbox mode first (
--sandbox) before pushing it live (
Additional flags can be used for more specific purposes.
--uniqueensures that an Turker is only able to complete one assignment, thus ensuring each assignment is completed by a unique person.
--unique-qual-name <name>lets you use the above functionality across more than one task. Workers will only be able to complete a task launched with this flag for a given <name> once.
--allowed-conversations <num>prevents a Turker from entering more than <num> conversations at once (by using multiple windows/tabs). This defaults to 0, which is unlimited.
--count-completeonly counts completed assignments towards the num_conversations requested. This may lead to more conversations being had than requested (and thus higher costs for instances where one Turker disconnects and we pay the other) but it ensures that if you request 1,000 conversations you end up with at least 1,000 completed data points.
--max-connectionscontrols the number of HITs can be launched at the same time. If not specified, it defaults to 30; 0 is unlimited.
Handling Turker Disconnects¶
Sometimes you may find that a task you have created is leading to a lot of workers disconnecting in the middle of a conversation, or that a few people are disconnecting repeatedly. ParlAI MTurk offers two kinds of blocks to stop these workers from doing your hits.
- soft blocks can be created by using the
--block-qualification <name>flag with a name that you want to associate to your ParlAI tasks. Any user that hits the disconnect cap for a HIT with this flag active will not be able to participate in any HITs using this flag.
- hard blocks can be used by setting the
--hard-blockflag. Soft blocks in general are preferred, as Turkers can be block-averse (as it may affect their reputation) and sometimes the disconnects are out of their control. This will prevent any Turkers that hit the disconnect cap with this flag active from participating in any of your future HITs of any type.
Reviewing Turker’s Work¶
You can programmatically review work using the commands available in the MTurkManager class. See, for example, the review_work function in the
personachat_collect_personas task. In this task, HITs are automatically approved if they are deemed completed by the world.
If you don’t take any action in 4 weeks, all HITs will be auto-approved and Turkers will be paid.