Classifying the classifier: dissecting the weight space of neural networksVise andre og tillknytning
2020 (engelsk)Inngår i: Proceedings of the 24th European Conference on Artificial Intelligence (ECAI 2020) / [ed] Giuseppe De Giacomo, Alejandro Catala, Bistra Dilkina, Michela Milano, Senén Barro, Alberto Bugarín, Jérôme Lang, IOS PRESS , 2020, Vol. 325, s. 8s. 1119-1126, artikkel-id FAIA200209Konferansepaper, Publicerat paper (Fagfellevurdert)
Abstract [en]
This paper presents an empirical study on the weights of neural networks, where we interpret each model as a point in a high-dimensional space – the neural weight space. To explore the complex structure of this space, we sample from a diverse selection of training variations (dataset, optimization procedure, architecture,etc.) of neural network classifiers, and train a large number of models to represent the weight space. Then, we use a machine learning approach for analyzing and extracting information from this space. Most centrally, we train a number of novel deep meta-classifiers withthe objective of classifying different properties of the training setup by identifying their footprints in the weight space. Thus, the meta-classifiers probe for patterns induced by hyper-parameters, so that we can quantify how much, where, and when these are encoded through the optimization process. This provides a novel and complementary view for explainable AI, and we show how meta-classifiers can reveal a great deal of information about the training setup and optimization, by only considering a small subset of randomly selected consecutive weights. To promote further research on the weight space, we release the neural weight space (NWS) dataset – a collection of 320K weightsnapshots from 16K individually trained deep neural networks.
sted, utgiver, år, opplag, sider
IOS PRESS , 2020. Vol. 325, s. 8s. 1119-1126, artikkel-id FAIA200209
Serie
Frontiers in Artificial Intelligence and Applications, ISSN 1879-8314 ; 325
Emneord [en]
machine learning, deep learning, ai, computer vision
HSV kategori
Identifikatorer
URN: urn:nbn:se:liu:diva-169431DOI: 10.3233/FAIA200209ISI: 000650971301047ISBN: 9781643681016 (digital)OAI: oai:DiVA.org:liu-169431DiVA, id: diva2:1467249
Konferanse
European Conference on Artificial Intelligence (ECAI 2020)
Merknad
Funding: Wallenberg Autonomous Systems and Software Program (WASP); strategic research environment ELLIIT
2020-09-152020-09-152025-02-01