用可调控的肺结节数据评估医学AI的推理能力
FunnyNodules: A Customizable Medical Dataset Tailored for Evaluating Explainable AI
- 生成可控形状的合成肺结节,定义属性与诊断的关联规则
- 验证模型是否因正确原因做出判断,识别错误归因
- 适合研究可解释AI的开发者和医疗影像领域研究人员
现有密集标注的医学图像数据集缺乏诊断背后的推理信息,难以支撑可解释AI(xAI)模型的开发与评估。为此,我们提出FunnyNodules,一个完全参数化的合成数据集,用于系统分析基于属性的医学AI推理。该数据集生成具有可调视觉属性(如圆形度、边缘锐度、毛刺性)的抽象肺结节样形状,目标类别由预设属性组合决定,实现对诊断规则的完全控制。通过该数据集,可进行模型无关评估,检验模型是否学习到正确的属性-目标关系,揭示属性预测中的过/欠拟合问题,并分析注意力机制与特定属性区域的一致性。框架支持数据复杂度、目标定义、类别平衡等多维度自定义。完备的真值信息为可解释AI方法在医学图像分析中的开发、基准测试与深度分析提供了通用基础。
原文摘要 · Abstract (English)
Densely annotated medical image datasets that capture not only diagnostic labels but also the underlying reasoning behind these diagnoses are scarce. Such reasoning-related annotations are essential for developing and evaluating explainable AI (xAI) models that reason similarly to radiologists: making correct predictions for the right reasons. To address this gap, we introduce FunnyNodules, a fully parameterized synthetic dataset designed for systematic analysis of attribute-based reasoning in medical AI models. The dataset generates abstract, lung nodule-like shapes with controllable visual attributes such as roundness, margin sharpness, and spiculation. The target class is derived from a predefined attribute combination, allowing full control over the decision rule that links attributes to the diagnostic class. We demonstrate how FunnyNodules can be used in model-agnostic evaluations to assess whether models learn correct attribute-target relations, to interpret over- or underperformance in attribute prediction, and to analyze attention alignment with attribute-specific regions of interest. The framework is fully customizable, supporting variations in dataset complexity, target definitions, class balance, and beyond. With complete ground truth information, FunnyNodules provides a versatile foundation for developing, benchmarking, and conducting in-depth analyses of explainable AI methods in medical image analysis.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。