title: Balancing Robustness and Sensitivity using Feature Contrastive Learning
abstract: It is generally believed that robust training of extremely large networks is critical to their success in real-world applications. However, when taken to the extreme, methods that promote robustness can hurt the model’s sensitivity to rare or underrepresented patterns. In this paper, we discuss this trade-off between robustness and sensitivity by introducing two notions: contextual feature utility and contextual feature sensitivity. We propose Feature Contrastive Learning (FCL) that encourages the model to be more sensitive to the features that have higher contextual utility. Empirical results demonstrate that models trained with FCL achieve a better balance of robustness and sensitivity, leading to improved generalization in the presence of noise.
authors: Seungyeon Kim Daniel Glasner Srikumar Ramalingam Cho-Jui Hsieh Kishore Papineni Sanjiv Kumar
hIndex: Cho-Jui Hsieh 58