BrainSurgery: 可重现且可靠的声明式权重操作,用于模型编辑与升级
摘要
BrainSurgery 是一个用于对神经网络检查点进行可重现且声明式权重操作的工具,通过内置验证的 YAML 计划实现模型编辑与升级。
查看缓存全文
缓存时间: 2026/06/10 09:44
论文页面 - BrainSurgery:用于模型编辑和升级的可重复、可靠的声明式权重操作
来源:https://huggingface.co/papers/2606.09707
摘要
BrainSurgery 是一个通过带有内置验证的声明式 YAML 计划,对神经网络检查点进行鲁棒且可重复的张量操作的工具。
随着深度学习模型规模的扩大,管理和修改大型检查点变得越来越具有挑战性。研究人员经常需要修改模型权重以进行层重组、精度转换、低秩分解和架构调试,但这些工作流通常依赖于脆弱的临时 Python 脚本。在此,我们介绍 BrainSurgery,这是一个用于对神经网络检查点进行鲁棒且可重复的“张量手术”的工具,并通过四个示例和三个案例研究(涵盖从模型升级到 LoRA 提取)提供了系统演示。通过抽象存储格式和内存管理,BrainSurgery 通过声明式 YAML 计划执行复杂的转换。它支持结构修改、数学变换和张量重塑,通过表达性的正则表达式和结构定位实现,同时内置的断言可验证张量形状、数据类型和值,以防止静默错误。我们期望 BrainSurgery 通过其可重复且经过验证的操作,为未来的研究提供坚实的基础。
查看 arXiv 页面 (https://arxiv.org/abs/2606.09707)查看 PDF (https://arxiv.org/pdf/2606.09707)GitHub3 (https://github.com/schneiderkamplab/brainsurgery)添加到收藏 (https://huggingface.co/login?next=%2Fpapers%2F2606.09707)
在您的代理中获取此论文:
hf papers read 2606\.09707
没有最新的 CLI?curl \-LsSf https://hf\.co/cli/install\.sh \| bash
引用此论文的模型0
没有模型链接此论文
请在模型 README.md 中引用 arxiv.org/abs/2606.09707 以从此页面链接。
引用此论文的数据集0
没有数据集链接此论文
请在数据集 README.md 中引用 arxiv.org/abs/2606.09707 以从此页面链接。
引用此论文的空间0
没有空间链接此论文
请在空间 README.md 中引用 arxiv.org/abs/2606.09707 以从此页面链接。
包含此论文的收藏集0
没有收藏集包含此论文
请将此论文添加到收藏集 (https://huggingface.co/new-collection) 以从此页面链接。
相似文章
从神经网络中定向恢复权重空间机制
本文提出Targeted Parameter Decomposition (tPD)方法,可针对特定输入选择性地从神经网络中恢复可解释的权重空间机制,相比全分解大幅降低计算需求。通过在玩具模型和Transformer语言模型上的验证,tPD能够忠实地恢复电路并进行手术级消融,且副作用极小。
@ClementDelangue: 不是你的权重,就不是你的大脑!
Clement Delangue 发推文强调拥有AI模型权重的重要性,指出如果你不拥有它们,你就没有拥有你的大脑。
微调是否会撤销激活引导?行为恢复而无权重编辑逆转
本文研究微调是否会撤销语言模型中的激活引导,发现尽管在优化压力下,拒绝抑制等行为效应可能退化,但底层的权重修改在机制上保持稳定。
BrainG3N: 一种用于可控3D脑部MRI生成的双用途分词器
介绍了BrainG3N,一种用于3D脑部MRI潜在扩散的双用途分词器,它使用冻结的掩码自编码器(MAE)编码器生成临床信息丰富的嵌入表示,并使用CNN解码器进行重建,在23个任务的基准测试中达到了最先进性能,并实现了可控生成和纵向预测。
将分析严谨性引入科学中的代理AI:Brain Researcher神经影像数据分析平台
Brain Researcher是一个代理平台,通过强制执行分析严谨性、提高工具选择准确性和确保可重复性来增强AI驱动的神经影像数据分析。该研究展示了显著的性能提升,并将方法论判断集成到工作流程中。