论文标题
基于皮质表面深度学习的解释性框架
An explainability framework for cortical surface-based deep learning
论文作者
论文摘要
解释性方法的出现使人们可以通过易于理解和实施的概念来更好地理解深层神经网络的运作方式。尽管大多数解释性方法是为传统深度学习设计的,但有些方法是为几何深度学习而进一步开发的,其中数据主要表示为图。这些表示定期从医学成像数据中得出,尤其是在神经影像领域,其中使用图来表示大脑结构和功能接线模式(脑连接组),并使用皮质表面模型来表示大脑的解剖结构。尽管已经开发出用于识别重要顶点(大脑区域)和图形分类的特征的解释性技术,但对于更复杂的任务,例如基于表面的模态转移(或Vertex-Wise回归)仍然缺乏这些方法。在这里,我们通过为基于皮质表面的深度学习的框架开发一个基于表面的可解释性方法的需求,为模态转移任务提供了透明的系统。首先,我们改编了一种基于扰动的方法,以与表面数据一起使用。然后,我们应用了基于扰动的方法来研究为直接在皮质表面模型上直接从解剖学预测脑功能的几何深度学习模型所使用的关键特征和顶点。我们表明,我们的解释性框架不仅能够识别重要的特征及其空间位置,而且还可以可靠且有效。
The emergence of explainability methods has enabled a better comprehension of how deep neural networks operate through concepts that are easily understood and implemented by the end user. While most explainability methods have been designed for traditional deep learning, some have been further developed for geometric deep learning, in which data are predominantly represented as graphs. These representations are regularly derived from medical imaging data, particularly in the field of neuroimaging, in which graphs are used to represent brain structural and functional wiring patterns (brain connectomes) and cortical surface models are used to represent the anatomical structure of the brain. Although explainability techniques have been developed for identifying important vertices (brain areas) and features for graph classification, these methods are still lacking for more complex tasks, such as surface-based modality transfer (or vertex-wise regression). Here, we address the need for surface-based explainability approaches by developing a framework for cortical surface-based deep learning, providing a transparent system for modality transfer tasks. First, we adapted a perturbation-based approach for use with surface data. Then, we applied our perturbation-based method to investigate the key features and vertices used by a geometric deep learning model developed to predict brain function from anatomy directly on a cortical surface model. We show that our explainability framework is not only able to identify important features and their spatial location but that it is also reliable and valid.