Using linear initialisation to improve speed of convergence and fully-trained error in Autoencoders

Marcel Marais, Mate Hartstein, George Cevora
11/17/2023
cs.LGcs.AIcs.CV

Official code: artefactory-uk/autoencoder-paper (3 stars, last commit 2023-11-17).

Abstract

Good weight initialisation is an important step in successful training of Artificial Neural Networks. Over time a number of improvements have been proposed to this process. In this paper we introduce a novel weight initialisation technique called the Straddled Matrix Initialiser. This initialisation technique is motivated by our assumption that major, global-scale relationships in data are linear with only smaller effects requiring complex non-linearities. Combination of Straddled Matrix and ReLU activation function initialises a Neural Network as a de facto linear model, which we postulate should be a better starting point for optimisation given our assumptions. We test this by training autoencoders on three datasets using Straddled Matrix and seven other state-of-the-art weight initialisation techniques. In all our experiments the Straddeled Matrix Initialiser clearly outperforms all other methods.

View on arXivView PDF

Code Implementations(1)

31Shell, PythonJun 28, 20222 years ago

Similar Papers

Cite this paper

@article{marais2023using,
  title  = {Using linear initialisation to improve speed of convergence and fully-trained error in Autoencoders},
  author = {Marcel Marais and Mate Hartstein and George Cevora},
  year   = {2023},
  eprint = {2311.10699},
  archivePrefix = {arXiv},
  url    = {http://arxiv.org/abs/2311.10699v1}
}

Add a badge to your README

Shows whether official code is linked for this paper.

Code of Paper badge
[![Code of Paper](https://codeofpaper.com/badge/2311.10699.svg)](https://codeofpaper.com/paper/2311.10699)

Discussion