Clarify: Improving Model Robustness With Natural Language Corrections

要旨

The standard way to teach models is by feeding them lots of data. However, this approach often teaches models incorrect ideas because they pick up on misleading signals in the data. To prevent such misconceptions, we must necessarily provide additional information beyond the training data. Prior methods incorporate additional instance-level supervision, such as labels for misleading features or additional labels for debiased data. However, such strategies require a large amount of labeler effort. We hypothesize that people are good at providing textual feedback at the concept level, a capability that existing teaching frameworks do not leverage. We propose Clarify, a novel interface and method for interactively correcting model misconceptions. Through Clarify, users need only provide a short text description of a model's consistent failure patterns. Then, in an entirely automated way, we use such descriptions to improve the training process. Clarify is the first end-to-end system for user model correction. Our user studies show that non-expert users can successfully describe model misconceptions via Clarify, leading to increased worst-case performance in two datasets. We additionally conduct a case study on a large-scale image dataset, ImageNet, using Clarify to find and rectify 31 novel hard subpopulations.

著者
Yoonho Lee
Stanford University, Stanford, California, United States
Michelle S.. Lam
Stanford University, Stanford, California, United States
Helena Vasconcelos
Stanford University, Stanford, California, United States
Michael S.. Bernstein
Stanford University, Stanford, California, United States
Chelsea Finn
Stanford University, Stanford, California, United States
論文URL

https://doi.org/10.1145/3654777.3676362

動画

会議: UIST 2024

ACM Symposium on User Interface Software and Technology

セッション: 3. Validation in AI/ML

Westin: Allegheny 3
5 件の発表
2024-10-16 23:00:00
2024-10-17 00:15:00