🤖 AI Summary
This study addresses the limited interpretability of deep neural networks and the lack of structural fidelity in post-hoc analyses by proposing a polyhedral neural network framework. This framework pioneers the embedding of polyhedral geometric constraints directly into the network inference process, integrating amortized simplex inference with learned corpus representations to manipulate information through inter-layer polyhedral constraints. Its core contribution lies in establishing a novel paradigm for vector quantization training that generates semantically meaningful latent space structures with minimal performance degradation, thereby achieving intrinsic model transparency. Experimental results demonstrate that the proposed method outperforms conventional vector quantization approaches on unsupervised compression tasks, effectively reconciling high interpretability with competitive model performance.
📝 Abstract
Understanding how deep neural networks process information remains a central challenge. Existing interpretability methods often compromise structural fidelity, rely on prespecified corpora, or explain models post-hoc. We propose Polytopal Neural Networks (PNNs), a framework that extracts distinct layer-wise aspects by enforcing a polytope-based structure that is used directly in subsequent information processing. We scale our approach using learned corpus representations and an amortized simplex inference procedure and highlight how the framework also gives a direct route to vector quantized (VQ) training. In PNNs, observations are explicitly described by their alignment with layer-specific aspects. Empirical results show that imposing polytopal constraints on neural network representations preserves meaningful structures in the latent space with minimal degradation in performance, favorable compressed representations when compared to VQ representations in unsupervised learning, while also providing a performant new approach to VQ deep learning training. Our findings suggest that deep networks can enforce interpretable polytope-based representations, offering a principled path toward more transparent AI systems with minimal performance compromise.