中文

ProtoVAE:一种可信的自解释原型变分模型

机器学习 2022-10-18 v1

摘要

对可解释模型的需求促进了自解释分类器的发展。现有方法要么基于多阶段优化方案,影响了模型的预测性能,要么产生的解释不透明、不可信或未能捕捉数据的多样性。为解决这些不足,我们提出 ProtoVAE,一种基于变分自编码器的框架,它以端到端方式学习类特定原型,并通过正则化表示空间与引入正交归一约束来增强可信性与多样性。最后,该模型通过将原型直接纳入决策过程来实现透明性。与以往自解释方法的广泛比较证明了 ProtoVAE 的优越性,凸显了其在生成可信且多样的解释的同时不降低预测性能的能力。

关键词

引用

@article{arxiv.2210.08151,
  title  = {ProtoVAE: A Trustworthy Self-Explainable Prototypical Variational Model},
  author = {Srishti Gautam and Ahcene Boubekki and Stine Hansen and Suaiba Amina Salahuddin and Robert Jenssen and Marina MC Höhne and Michael Kampffmeyer},
  journal= {arXiv preprint arXiv:2210.08151},
  year   = {2022}
}