Mar, 2024

pyvene:通过干预理解和改进 PyTorch 模型的开源库

TL;DR通过在模型内部状态进行干预,我们介绍了一个开源的 Python 库 pyvene,它支持对不同的 PyTorch 模块进行可定制化的干预。该库提供了一个统一且可扩展的框架,可以在神经模型上执行干预操作,并与他人共享这些修改过的模型。我们通过因果抽象和知识定位的解释性分析展示了这个库的功能。