English

Association Restoration Test: Revealing Restorable Shortcuts after Unlearning

Computer Vision and Pattern Recognition 2026-07-07 v1 Machine Learning

Abstract

Association unlearning aims to disable learned label-attribute shortcuts while preserving task performance. Existing evaluations mainly measure output-level robustness or probe whether shortcut attributes remain readable in frozen features, but neither test determines whether a retained association remains functionally usable by the original classifier. We propose the Association Restoration Test (ART), a post-hoc diagnostic for functional shortcut restorability. ART estimates class-conditional association directions, amplifies residual components, and evaluates the modified features with the original classifier head. Across Waterbirds, CelebA, SpuCoDogs, and an ISIC timestamp-artifact extension, we show that output metrics, representation probes, and ART characterize distinct aspects of shortcut mitigation. These findings motivate restoration-aware evaluation for unlearning and shortcut-mitigation methods that target learned associations rather than individual classes or concepts.

Keywords

Cite

@article{arxiv.2607.05726,
  title  = {Association Restoration Test: Revealing Restorable Shortcuts after Unlearning},
  author = {Amy Lu and Changxiu Ji},
  journal= {arXiv preprint arXiv:2607.05726},
  year   = {2026}
}

Comments

Preprint. 16 pages