Practical guidelines for intent recognition: BERT with minimal training data evaluated in real-world HRI application

Intent recognition models, which match a written or spoken input's class in order to guide an interaction, are an essential part of modern voice user interfaces, chatbots, and social robots. However, getting enough data to train these models can be very expensive and challenging, especially whe...

Full description

Bibliographic Details
Main Authors: Huggins, M, Alghowinem, S, Jeong, S, Colon-Hernandez, P, Breazeal, C, Park, HW
Other Authors: Massachusetts Institute of Technology. Media Laboratory
Format: Article
Language:English
Published: ACM 2021
Online Access:https://hdl.handle.net/1721.1/137130
_version_ 1826197514069475328
author Huggins, M
Alghowinem, S
Jeong, S
Colon-Hernandez, P
Breazeal, C
Park, HW
author2 Massachusetts Institute of Technology. Media Laboratory
author_facet Massachusetts Institute of Technology. Media Laboratory
Huggins, M
Alghowinem, S
Jeong, S
Colon-Hernandez, P
Breazeal, C
Park, HW
author_sort Huggins, M
collection MIT
description Intent recognition models, which match a written or spoken input's class in order to guide an interaction, are an essential part of modern voice user interfaces, chatbots, and social robots. However, getting enough data to train these models can be very expensive and challenging, especially when designing novel applications such as real-world human-robot interactions. In this work, wefi rst investigate how much training data is needed for high performance in an intent classification task. We train and evaluate BiLSTM and BERT models on various subsets of the ATIS and Snips datasets. Wefi nd that only 25 training examples per intent are required for our BERT model to achieve 94% intent accuracy compared to 98% with the entire datasets, challenging the belief that large amounts of labeled data are required for high performance in intent recognition. We apply this knowledge to train models for a real-world HRI application, character strength recognition during a positive psychology interaction with a social robot, and evaluate against the Character Strength dataset collected in our previous HRI study. Our real-world HRI application results also confirm that our model can produce 76% intent accuracy with 25 examples per intent compared to 80% with 100 examples. In a real-world scenario, the difference is only one additional error per 25 classifications. Finally, we investigate the limitations of our minimal data models and offer suggestions on developing high quality datasets. We conclude with practical guidelines for training BERT intent recognition models with minimal training data and make our code and evaluation framework available for others to replicate our results and easily develop models for their own applications.
first_indexed 2024-09-23T10:48:50Z
format Article
id mit-1721.1/137130
institution Massachusetts Institute of Technology
language English
last_indexed 2024-09-23T10:48:50Z
publishDate 2021
publisher ACM
record_format dspace
spelling mit-1721.1/1371302023-04-11T19:55:46Z Practical guidelines for intent recognition: BERT with minimal training data evaluated in real-world HRI application Huggins, M Alghowinem, S Jeong, S Colon-Hernandez, P Breazeal, C Park, HW Massachusetts Institute of Technology. Media Laboratory Intent recognition models, which match a written or spoken input's class in order to guide an interaction, are an essential part of modern voice user interfaces, chatbots, and social robots. However, getting enough data to train these models can be very expensive and challenging, especially when designing novel applications such as real-world human-robot interactions. In this work, wefi rst investigate how much training data is needed for high performance in an intent classification task. We train and evaluate BiLSTM and BERT models on various subsets of the ATIS and Snips datasets. Wefi nd that only 25 training examples per intent are required for our BERT model to achieve 94% intent accuracy compared to 98% with the entire datasets, challenging the belief that large amounts of labeled data are required for high performance in intent recognition. We apply this knowledge to train models for a real-world HRI application, character strength recognition during a positive psychology interaction with a social robot, and evaluate against the Character Strength dataset collected in our previous HRI study. Our real-world HRI application results also confirm that our model can produce 76% intent accuracy with 25 examples per intent compared to 80% with 100 examples. In a real-world scenario, the difference is only one additional error per 25 classifications. Finally, we investigate the limitations of our minimal data models and offer suggestions on developing high quality datasets. We conclude with practical guidelines for training BERT intent recognition models with minimal training data and make our code and evaluation framework available for others to replicate our results and easily develop models for their own applications. 2021-11-02T17:26:47Z 2021-11-02T17:26:47Z 2021 2021-06-24T16:30:12Z Article http://purl.org/eprint/type/ConferencePaper https://hdl.handle.net/1721.1/137130 Huggins, M, Alghowinem, S, Jeong, S, Colon-Hernandez, P, Breazeal, C et al. 2021. "Practical guidelines for intent recognition: BERT with minimal training data evaluated in real-world HRI application." ACM/IEEE International Conference on Human-Robot Interaction. en 10.1145/3434073.3444671 ACM/IEEE International Conference on Human-Robot Interaction Creative Commons Attribution NonCommercial License 4.0 https://creativecommons.org/licenses/by-nc/4.0/ application/pdf ACM ACM
spellingShingle Huggins, M
Alghowinem, S
Jeong, S
Colon-Hernandez, P
Breazeal, C
Park, HW
Practical guidelines for intent recognition: BERT with minimal training data evaluated in real-world HRI application
title Practical guidelines for intent recognition: BERT with minimal training data evaluated in real-world HRI application
title_full Practical guidelines for intent recognition: BERT with minimal training data evaluated in real-world HRI application
title_fullStr Practical guidelines for intent recognition: BERT with minimal training data evaluated in real-world HRI application
title_full_unstemmed Practical guidelines for intent recognition: BERT with minimal training data evaluated in real-world HRI application
title_short Practical guidelines for intent recognition: BERT with minimal training data evaluated in real-world HRI application
title_sort practical guidelines for intent recognition bert with minimal training data evaluated in real world hri application
url https://hdl.handle.net/1721.1/137130
work_keys_str_mv AT hugginsm practicalguidelinesforintentrecognitionbertwithminimaltrainingdataevaluatedinrealworldhriapplication
AT alghowinems practicalguidelinesforintentrecognitionbertwithminimaltrainingdataevaluatedinrealworldhriapplication
AT jeongs practicalguidelinesforintentrecognitionbertwithminimaltrainingdataevaluatedinrealworldhriapplication
AT colonhernandezp practicalguidelinesforintentrecognitionbertwithminimaltrainingdataevaluatedinrealworldhriapplication
AT breazealc practicalguidelinesforintentrecognitionbertwithminimaltrainingdataevaluatedinrealworldhriapplication
AT parkhw practicalguidelinesforintentrecognitionbertwithminimaltrainingdataevaluatedinrealworldhriapplication