用合理分子片段替换被掩蔽结构,让化学解释更可信。
Enhancing Chemical Explainability Through Counterfactual Masking
- 用生成模型采样合理片段替代被遮蔽的分子部分。
- 解释结果与真实数据分布一致,避免不合理假设。
- 适合药物和材料设计中的可解释性分析。
分子性质预测是指导新化合物(包括药物和材料)设计的关键任务。现有可解释人工智能方法通常通过遮蔽原子或原子特征来评估重要性,但这些方法常违背分子分布规律,导致解释不直观。本文提出反事实遮蔽框架:将被遮蔽的分子结构替换为从训练好的图生成模型中采样的化学合理片段。不与虚构的零值基线比较,而是与来自数据分布的反事实分子对比。该方法带来两大优势:(1) 保持分子真实性,实现稳定且分布一致的解释;(2) 提供有意义的反事实,直接揭示结构修改对性质的影响。实验表明,反事实遮蔽适用于基准测试模型解释器,并在多个数据集和性质预测任务中提供更具行动价值的洞见。该方法弥合了可解释性与分子设计之间的鸿沟,为化学领域可解释机器学习提供了原则性且生成式的新路径。
原文摘要 · Abstract (English)
Molecular property prediction is a crucial task that guides the design of new compounds, including drugs and materials. While explainable artificial intelligence methods aim to scrutinize model predictions by identifying influential molecular substructures, many existing approaches rely on masking strategies that remove either atoms or atom-level features to assess importance via fidelity metrics. These methods, however, often fail to adhere to the underlying molecular distribution and thus yield unintuitive explanations. In this work, we propose counterfactual masking, a novel framework that replaces masked substructures with chemically reasonable fragments sampled from generative models trained to complete molecular graphs. Rather than evaluating masked predictions against implausible zeroed-out baselines, we assess them relative to counterfactual molecules drawn from the data distribution. Our method offers two key benefits: (1) molecular realism underpinning robust and distribution-consistent explanations, and (2) meaningful counterfactuals that directly indicate how structural modifications may affect predicted properties. We demonstrate that counterfactual masking is well-suited for benchmarking model explainers and yields more actionable insights across multiple datasets and property prediction tasks. Our approach bridges the gap between explainability and molecular design, offering a principled and generative path toward explainable machine learning in chemistry.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。