A Framework for Interpretability in Machine Learning for Medical Imaging

Abstract

Interpretability for machine learning models in medical imaging (MLMI) is an important direction of research. However, there is a general sense of murkiness in what interpretability means. Why does the need for interpretability in MLMI arise? What goals does one actually seek to address when interpretability is needed? To answer these questions, we identify a need to formalize the goals and elements of interpretability in MLMI. By reasoning about real-world tasks and goals common in both medical image analysis and its intersection with machine learning, we identify five core elements of interpretability: localization, visual recognizability, physical attribution, model transparency, and actionability. From this, we arrive at a framework for interpretability in MLMI, which serves as a step-by-step guide to approaching interpretability in this context. Overall, this paper formalizes interpretability needs in the context of medical imaging, and our applied perspective clarifies concrete MLMI-specific goals and considerations in order to guide method design and improve real-world usage. Our goal is to provide practical and didactic information for model designers and practitioners, inspire developers of models in the medical imaging field to reason more deeply about what interpretability is achieving, and suggest future directions of interpretability research.

Keywords: Interpretability; explainability; machine learning; medical imaging.

FIGURE 1.

A framework for interpretability in MLMI.

FIGURE 2.

Common tasks in MLMI. Tasks are primarily characterized by the structures of their input features and output predictions. In MLMI, inputs are images or features derived from images, sometimes combined with meta-data such as patient information. The structure of output predictions are determined by the task.

FIGURE 3.

Graphical overview of the elements of interpretable MLMI.