Back to Search Start Over

FAMLINN: Representation for Storing Neural Network Architecture.

Authors :
Maslov, Ivan
Bessonnitsyn, Evgeny
Efmova, Valeria
Shalamov, Viacheslav
Source :
Procedia Computer Science; 2023, Vol. 229, p55-61, 7p
Publication Year :
2023

Abstract

To date, there are several ways to store neural networks, however, each of them has drawbacks. Sometimes it becomes necessary to store not only the weights of a particular model but also its architecture. The ONNX and TorchScript representations manage this task but have a problem that after saving they do not allow changing the network architecture. We propose a more flexible neural network saving method that solves this problem by allowing automated architecture generation and the network architecture manual tuning. For the convenience of working with existing formats, the ability to convert our format to the PyTorch format was implemented. According to the comparison results on several well-known architectures, the new format surpasses existing solutions. The method implementation is publicly available at https://github.com/IvanMaslov/famlinn. [ABSTRACT FROM AUTHOR]

Subjects

Subjects :
PROBLEM solving

Details

Language :
English
ISSN :
18770509
Volume :
229
Database :
Supplemental Index
Journal :
Procedia Computer Science
Publication Type :
Academic Journal
Accession number :
174470540
Full Text :
https://doi.org/10.1016/j.procs.2023.12.007