ExplainabilityTransparencyAccountabilitycs.CV
Classifying the classifier: dissecting the weight space of neural networks
Gabriel Eilertsen, Daniel Jönsson, Timo Ropinski, Jonas Unger, Anders Ynnerman
February 13, 2020
European Conference on Artificial Intelligence
79 citations
7 influential
Key Takeaways
-
Neural networks encode training information in their weights: The research demonstrates that machine learning models can identify what dataset was used, which optimization method was applied, and other training details simply by analyzing the pattern of weights in a trained network—suggesting that the training process leaves distinctive "fingerprints" in model parameters.
-
Small samples of weights reveal significant information: Meta-classifiers achieved meaningful results by examining only a small, randomly selected subset of consecutive weights rather than analyzing entire networks, indicating that training information is distributed throughout the weight space rather than concentrated in specific locations.
-
This work advances explainability of AI systems: By mapping how hyperparameters and training choices manifest in neural network weights, this research provides a new avenue for understanding what happens inside "black box" AI models—complementing existing explainability approaches and potentially helping practitioners understand why their models behave as they do.
-
A new public dataset enables future research: The release of 320,000 weight snapshots from 16,000 trained neural networks provides researchers with a resource to further investigate neural network behavior, potentially accelerating development of more interpretable and trustworthy AI systems.
-
Training choices have measurable, detectable effects: The ability to classify training setups from weights alone suggests that decisions about datasets, optimization procedures, and architecture are not equally important or equally encoded—some choices leave stronger signatures than others, which could inform best practices for responsible model development.