Machine Learning Robustness: A Primer

TL;DR

This paper provides a comprehensive overview of robustness in Machine Learning, covering its definitions, importance for trustworthy AI, assessment techniques, and strategies for improvement, highlighting ongoing challenges and future directions.

Contribution

It offers a detailed survey of robustness concepts, metrics, assessment methods, and mitigation strategies, integrating both adversarial and non-adversarial perspectives in ML.

Findings

Robustness is essential for trustworthy AI systems.

Multiple techniques exist for assessing and improving ML robustness.

Challenges remain in accurately estimating and enhancing robustness.

Abstract

This chapter explores the foundational concept of robustness in Machine Learning (ML) and its integral role in establishing trustworthiness in Artificial Intelligence (AI) systems. The discussion begins with a detailed definition of robustness, portraying it as the ability of ML models to maintain stable performance across varied and unexpected environmental conditions. ML robustness is dissected through several lenses: its complementarity with generalizability; its status as a requirement for trustworthy AI; its adversarial vs non-adversarial aspects; its quantitative metrics; and its indicators such as reproducibility and explainability. The chapter delves into the factors that impede robustness, such as data bias, model complexity, and the pitfalls of underspecified ML pipelines. It surveys key techniques for robustness assessment from a broad perspective, including adversarial attacks, encompassing both digital and physical realms. It covers non-adversarial data shifts and nuances of Deep Learning (DL) software testing methodologies. The discussion progresses to explore amelioration strategies for bolstering robustness, starting with data-centric approaches like debiasing and augmentation. Further examination includes a variety of model-centric methods such as transfer learning, adversarial training, and randomized smoothing. Lastly, post-training methods are discussed, including ensemble techniques, pruning, and model repairs, emerging as cost-effective strategies to make models more resilient against the unpredictable. This chapter underscores the ongoing challenges and limitations in estimating and achieving ML robustness by existing approaches. It offers insights and directions for future research on this crucial concept, as a prerequisite for trustworthy AI systems.