English

Editing a classifier by rewriting its prediction rules

Machine Learning 2021-12-03 v1 Computer Vision and Pattern Recognition

Abstract

We present a methodology for modifying the behavior of a classifier by directly rewriting its prediction rules. Our approach requires virtually no additional data collection and can be applied to a variety of settings, including adapting a model to new environments, and modifying it to ignore spurious features. Our code is available at https://github.com/MadryLab/EditingClassifiers .

Keywords

Cite

@article{arxiv.2112.01008,
  title  = {Editing a classifier by rewriting its prediction rules},
  author = {Shibani Santurkar and Dimitris Tsipras and Mahalaxmi Elango and David Bau and Antonio Torralba and Aleksander Madry},
  journal= {arXiv preprint arXiv:2112.01008},
  year   = {2021}
}
R2 v1 2026-06-24T08:00:58.373Z