用压缩思想优化神经网络解释,让特征更简洁准确
Interpretability as Compression: Reconsidering SAE Explanations of Neural Activations with MDL-SAEs
- 将SAE视为信息压缩工具,用最小描述长度原则选择解释
- 实验显示线段特征比碎片化特征更优,避免过度拆分
- 适合研究模型可解释性与结构化特征提取的学者
稀疏自编码器(SAEs)已成为解释神经网络内部表示的有效工具。然而,单纯优化重建损失和稀疏性会导致SAE极度宽且稀疏。本文提出一种基于信息论的框架,将SAE视为神经激活解释的有损压缩算法。依据最小描述长度(MDL)原则,追求既准确又简洁的解释。进一步指出,可解释的SAE需具备“独立可加性”:各特征应能独立理解。在手写数字MNIST数据集上训练SAE的实例表明,代表显著线段的特征最优,而非记忆数据集中的完整数字或微小片段。相比单纯最大化稀疏性,采用MDL可避免特征过度分裂等陷阱,并自然引导出更简洁的层次化SAE架构。
原文摘要 · Abstract (English)
Sparse Autoencoders (SAEs) have emerged as a useful tool for interpreting the internal representations of neural networks. However, naively optimising SAEs for reconstruction loss and sparsity results in a preference for SAEs that are extremely wide and sparse. We present an information-theoretic framework for interpreting SAEs as lossy compression algorithms for communicating explanations of neural activations. We appeal to the Minimal Description Length (MDL) principle to motivate explanations of activations which are both accurate and concise. We further argue that interpretable SAEs require an additional property, "independent additivity": features should be able to be understood separately. We demonstrate an example of applying our MDL-inspired framework by training SAEs on MNIST handwritten digits and find that SAE features representing significant line segments are optimal, as opposed to SAEs with features for memorised digits from the dataset or small digit fragments. We argue that using MDL rather than sparsity may avoid potential pitfalls with naively maximising sparsity such as undesirable feature splitting and that this framework naturally suggests new hierarchical SAE architectures which provide more concise explanations.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。