Collaborative Research: SCH: Geometry and Topology for Interpretable and Reliable Deep Learning in Medical Imaging

合作研究：SCH：医学成像中可解释且可靠的深度学习的几何和拓扑

• 批准号: 2205417
• 负责人: Preston Fletcher
• 金额: $ 62.3万
• 依托单位: University of Virginia Main Campus
• 依托单位国家: 美国
• 项目类别: Standard Grant
• 财政年份: 2022
• 资助国家: 美国
• 起止时间: 2022-09-01 至 2026-08-31
• 项目状态: 未结题
• 来源: https://www.nsf.gov/awardsearch/showAward?AWD_ID=2205417&HistoricalAwards=false
• 关键词: Collaborative; Research; SCH; Geometry; Topology

Deep learning models are being developed for safety-critical applications, such as health care, autonomous vehicles, and security. Their impressive performance has the potential to make profound impacts on human lives. For example, deep neural networks (DNNs) in medical imaging have been shown to have impressive diagnostic capabilities, often near that of expert radiologists. However, deep learning has not made it into standard clinical care, primarily due to a lack of understanding of why a model works and why it fails. The goal of this project is to develop methods for making machine learning models interpretable and reliable, and thus bridge the trust gap to make machine learning translatable to the clinic. This project achieves this goal through investigation of the mathematical foundations -- specifically the geometry and topology -- of DNNs. Based on these mathematical foundations, this project will develop computational tools that will improve the interpretability and reliability of DNNs. The methods developed in this project will be broadly applicable wherever deep learning is used, including health care, security, computer vision, natural language processing, etc.The power of a deep neural network lies in its hidden layers, where the network learns internal representations of input data. This research project centers around the hypothesis that geometry and topology provide critical tools for analyzing the internal representations of DNNs. The first goal of this project is to develop a rigorous mathematical and algorithmic foundation for describing the geometry and topology of a neural network's internal representations and then design efficient algorithms for geometric and topological computations necessary to explore these spaces. The next aim of this project is to apply these tools to improve the interpretability of deep learning. This will be done by linking a model's internal representation with interpretable and trusted features and by interactive visualization that explores the landscape of a model's internal representation. The next goal of this project focuses on model reliability, where geometry and topology will be used for failure identification, mitigation, and prevention. Finally, this project will test the developed techniques for reliable and interpretable neural networks in a real-world setting to aid expert oncologists in predicting patient outcomes in head and neck cancers, e.g., whether a tumor will metastasize.This award reflects NSF's statutory mission and has been deemed worthy of support through evaluation using the Foundation's intellectual merit and broader impacts review criteria.

深度学习模型正在开发用于安全关键型应用，如医疗保健、自动驾驶汽车和安全。它们令人印象深刻的表现有可能对人类生活产生深远的影响。例如，医学成像中的深度神经网络（DNN）已被证明具有令人印象深刻的诊断能力，通常接近放射科专家的诊断能力。然而，深度学习还没有成为标准的临床护理，主要是由于缺乏对模型为什么有效以及为什么失败的理解。该项目的目标是开发使机器学习模型可解释和可靠的方法，从而弥合信任差距，使机器学习可转化为临床。该项目通过研究DNN的数学基础（特别是几何和拓扑）来实现这一目标。基于这些数学基础，该项目将开发计算工具，以提高DNN的可解释性和可靠性。该项目开发的方法将广泛适用于任何使用深度学习的地方，包括医疗保健，安全，计算机视觉，自然语言处理等。深度神经网络的强大之处在于其隐藏层，网络在其中学习输入数据的内部表示。这个研究项目围绕着这样一个假设，即几何和拓扑为分析DNN的内部表示提供了关键工具。该项目的第一个目标是开发一个严格的数学和算法基础，用于描述神经网络内部表示的几何和拓扑结构，然后设计有效的算法，用于探索这些空间所需的几何和拓扑计算。该项目的下一个目标是应用这些工具来提高深度学习的可解释性。这将通过将模型的内部表示与可解释和可信的功能联系起来，以及通过探索模型内部表示的交互式可视化来实现。该项目的下一个目标集中在模型的可靠性，几何和拓扑结构将用于故障识别，缓解和预防。最后，该项目将在现实世界中测试开发的可靠和可解释的神经网络技术，以帮助肿瘤专家预测头颈癌患者的预后，例如，该奖项反映了NSF的法定使命，并通过使用基金会的知识价值和更广泛的影响审查标准进行评估，被认为值得支持。