Feature Integration Spaces: Joint Training Reveals Dual Encoding in Neural Network Representations
DOI:
https://doi.org/10.1609/aaai.v40i42.40847Abstract
Current sparse autoencoder (SAE) approaches to neural network interpretability assume that activations can be decomposed through linear superposition into sparse, interpret-able features. Despite high reconstruction fidelity, SAEs consistently fail to eliminate polysemanticity and exhibit pathological behavioral errors. We propose that neural networks encode information in two complementary spaces com-pressed into the same substrate: feature identity and feature integration. To test this dual encoding hypothesis, we develop sequential and joint-training architectures to capture identity and integration patterns simultaneously. Joint training achieves 41.3% reconstruction improvement and 51.6% reduction in KL divergence errors. This architecture spontaneously develops bimodal feature organization: low squared norm features contributing to integration pathways and the rest contributing directly to the residual. Small nonlinear components (3% of parameters) achieve 16.5% standalone improvements, demonstrating parameter-efficient capture of computational relationships crucial for behavior. Additionally, intervention experiments using 2×2 factorial stimulus designs demonstrated that integration features exhibit selective sensitivity to experimental manipulations and produce systematic behavioral effects on model outputs, including significant interaction effects across semantic dimensions. This work provides systematic evidence for (1) dual encoding in neural representations, (2) meaningful nonlinearly encoded feature interactions, and (3) introduces an architectural paradigm shift from post-hoc feature analysis to integrated computational design, establishing foundations for next-generation SAEs.Downloads
Published
2026-03-14
How to Cite
Claflin, O. (2026). Feature Integration Spaces: Joint Training Reveals Dual Encoding in Neural Network Representations. Proceedings of the AAAI Conference on Artificial Intelligence, 40(42), 35384–35392. https://doi.org/10.1609/aaai.v40i42.40847
Issue
Section
AAAI Technical Track on Philosophy and Ethics of AI