Skip to main content
Article
Robustar: Interactive Toolbox Supporting Precise Data Annotation for Robust Vision Learning
arXiv
  • Chonghan Chen, School of Computer Science, Carnegie Mellon University, United States
  • Haohan Wang, School of Computer Science, Carnegie Mellon University, United States & School of Information Science, University of Illinois Urbana-Champaign, United States
  • Leyang Hu, School of Computer Science, University of Nottingham Ningbo, China
  • Yuhao Zhang, School of Computer Science, University of Nottingham, United Kingdom
  • Shuguang Lyu, Bren School of Information and Computer Sciences University of California, Irvine, United States
  • Jingcheng Wu, School of Computer Science, Carnegie Mellon University, United States
  • Xinnuo Li, College of Literature, Science and the Arts, University of Michigan, United States
  • Linjing Sun, School of Computer Science, University of Nottingham Ningbo, China
  • Eric Xing, School of Computer Science, Carnegie Mellon University, United States & Mohamed bin Zayed University of Artificial Intelligence
Document Type
Article
Abstract

We introduce the initial release of our software Robustar, which aims to improve the robustness of vision classification machine learning models through a data-driven perspective. Building upon the recent understanding that the lack of machine learning model’s robustness is the tendency of the model’s learning of spurious features, we aim to solve this problem from its root at the data perspective by removing the spurious features from the data before training. In particular, we introduce a software that helps the users to better prepare the data for training image classification models by allowing the users to annotate the spurious features at the pixel level of images. To facilitate this process, our software also leverages recent advances to help identify potential images and pixels worthy of attention and to continue the training with newly annotated data. Our software is hosted at the GitHub Repository https://github.com/HaohanWang/Robustar. © 2022, CC BY.

DOI
10.48550/arXiv.2207.08944
Publication Date
7-18-2022
Keywords
  • Classification (of information),
  • Computer vision,
  • Machine learning
Comments

Preprint: arXiv

Archived with thanks to arXiv

Preprint License: CC by 4.0

Uploaded 25 August 2022

Citation Information
C. Chen et al, "Robustar: Interactive Toolbox Supporting Precise Data Annotation for Robust Vision Learning", 2022, arXiv:2207.08944