arXivDaily arXiv每日学术速递 周一至周五更新
arXiv周末暂无论文更新,休息一下吧,周末愉快~~
arXiv 2602.22968cs.AIcs.CVcs.CY
收藏

认证电路:机械论电路的稳定性保证

Certified Circuits: Stability Guarantees for Mechanistic Circuits

Max Planck Institute for Informatics(马克斯·普朗克研究所信息学研究所);CISPA Helmholtz Center for Information Security(信息安全赫尔姆霍茨中心)

Alaa Anani, Tobias Lorenz, Bernt Schiele, Mario Fritz, Jonas Fischer

AI总结:

提出Certified Circuits框架,通过随机数据子采样认证电路组件(神经元或边)对概念数据集编辑距离扰动的稳定性,生成更紧凑、更准确的电路。

摘要:

理解神经网络如何得出其预测对于调试、审计和部署至关重要。机械论可解释性通过识别电路——负责特定行为的最小子网络——来追求这一目标。然而,现有的电路发现方法脆弱:电路强烈依赖于所选的概念数据集,并且常常无法迁移到分布外,引发对其是否捕捉概念或仅仅是数据集特定伪影的怀疑。我们引入了Certified Circuits,它为电路发现提供了可证明的稳定性保证。我们的框架用随机数据子采样包装任何黑盒发现算法,以认证电路组件——根据基础算法,模型图的神经元或边——的包含决策对概念数据集的有界编辑距离扰动是不变的。不稳定的组件被弃用,从而产生更紧凑、更准确的电路。我们在三个架构(ResNet、ViT、GPT-2)上,针对视觉(ImageNet和四个OOD数据集)和语言(IOI、IOI-Hard、Greater-Than)任务进行了验证。认证电路实现了高达56%的更高准确率和高达80%的更少组件,并且在基线退化时保持可靠。Certified Circuits通过产生可证明稳定且与目标概念更好对齐的机械论解释,将电路发现置于形式化的基础上。代码:https://github.com/AlaaAnani/certified-circuits。

英文摘要:

Understanding how neural networks arrive at their predictions is essential for debugging, auditing, and deployment. Mechanistic interpretability pursues this goal by identifying circuits--minimal subnetworks responsible for specific behaviors. However, existing circuit discovery methods are brittle: circuits depend strongly on the chosen concept dataset and often fail to transfer out-of-distribution, raising doubts whether they capture the concept or merely dataset-specific artifacts. We introduce Certified Circuits, which provide provable stability guarantees for circuit discovery. Our framework wraps any black-box discovery algorithm with randomized data subsampling to certify that inclusion decisions over circuit components--neurons or edges of the model graph, depending on the base algorithm--are invariant to bounded edit-distance perturbations of the concept dataset. Unstable components are abstained from, yielding circuits that are more compact and more accurate. We validate across three architectures (ResNet, ViT, GPT-2) on vision (ImageNet and four OOD datasets) and language (IOI, IOI-Hard, Greater-Than) tasks. Certified circuits achieve up to 56% higher accuracy and up to 80% fewer components, and remain reliable where baselines degrade. Certified Circuits puts circuit discovery on formal ground by producing mechanistic explanations that are provably stable and better aligned with the target concept. Code: https://github.com/AlaaAnani/certified-circuits.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑